Merge pull request #63 from tbs60/dev_tming

Dev tming（支持controller的stderr重定向到文件）

src/backend/booster/bk_dist/common/syscall/syscall_unix.go

@@ -309,3 +309,18 @@ func AddPath2Env(p string) {
 	newpath := fmt.Sprintf("%s:%s", p, path)
 	os.Setenv("PATH", newpath)
 }
+
+func RedirectStderror(f string) error {
+	file, err := os.OpenFile(f, os.O_RDWR|os.O_CREATE|os.O_APPEND, 0666)
+	if err != nil {
+		fmt.Println(err)
+		return err
+	}
+
+	if err = syscall.Dup2(int(file.Fd()), int(os.Stderr.Fd())); err != nil {
+		fmt.Println(err)
+		return err
+	}
+
+	return nil
+}

src/backend/booster/bk_dist/common/syscall/syscall_windows.go

@@ -453,3 +453,43 @@ func AddPath2Env(p string) {
 	newpath := fmt.Sprintf("%s;%s", p, path)
 	os.Setenv("path", newpath)
 }
+
+func SetStdHandle(stdhandle int32, handle syscall.Handle) error {
+	kernel32, err := syscall.LoadLibrary("kernel32.dll")
+	if err != nil {
+		fmt.Println(err)
+		return err
+	}
+	defer syscall.FreeLibrary(kernel32)
+
+	api, err := syscall.GetProcAddress(kernel32, "SetStdHandle")
+	if err != nil {
+		fmt.Println(err)
+		return err
+	}
+
+	r0, _, e1 := syscall.Syscall(uintptr(api), 2, uintptr(stdhandle), uintptr(handle), 0)
+	if r0 == 0 {
+		if e1 != 0 {
+			return error(e1)
+		}
+		return syscall.EINVAL
+	}
+	return nil
+}
+
+func RedirectStderror(f string) error {
+	var err error
+	file, err := os.OpenFile(f, os.O_RDWR|os.O_CREATE|os.O_APPEND, 0666)
+	if err != nil {
+		fmt.Println(err)
+		return err
+	}
+	err = SetStdHandle(syscall.STD_ERROR_HANDLE, syscall.Handle(file.Fd()))
+	if err != nil {
+		fmt.Println(err)
+		return err
+	}
+
+	return nil
+}

src/backend/booster/bk_dist/controller/main.go

@@ -12,7 +12,9 @@ package main
 import (
 	"fmt"
 	"os"
+	"path/filepath"
 
+	"github.com/TencentBlueKing/bk-turbo/src/backend/booster/bk_dist/common/syscall"
 	"github.com/TencentBlueKing/bk-turbo/src/backend/booster/bk_dist/controller/config"
 	"github.com/TencentBlueKing/bk-turbo/src/backend/booster/bk_dist/controller/pkg"
 	"github.com/TencentBlueKing/bk-turbo/src/backend/booster/common/blog"
@@ -23,6 +25,9 @@ func main() {
 	c.Parse()
 	blog.InitLogs(c.LogConfig)
 
+	panicfile := filepath.Join(c.LogConfig.LogDir, "bk-dist-controller-panic.log")
+	syscall.RedirectStderror(panicfile)
+
 	if err := pkg.Run(c); err != nil {
 		_, _ = fmt.Fprintln(os.Stderr, err)
 		blog.Errorf("start controller failed: %v", err)