debug

zhangmeng

2020-01-11 c46b023d7a165d4b1a32f2b6fdc00d66334597f1

 main.go

@@ -3,9 +3,13 @@
import (
   "analysis/logo"
   "analysis/util"
   "fmt"
   "os/exec"
   "os/signal"
   "runtime/debug"
   "strconv"
   "syscall"
   "time"

   "analysis/app/master"
   "analysis/app/slave"
@@ -14,8 +18,6 @@
   "flag"

   "os"

   "basic.com/valib/gogpu.git"
   // "net/http"
   // _ "net/http/pprof"
)
@@ -38,8 +40,8 @@
)

func init() {
   flag.StringVar(&role, "role", roleMaster, "run process role master/slave, default single")
   flag.StringVar(&runType, "sdk", util.FeatAll, "run sdk type, face/facedetect/facecompare/yolo,etc.")
   flag.StringVar(&role, "role", roleMaster, "run process role master/slave, default master")
   flag.StringVar(&runType, "sdk", util.FeatAll, "run sdk type, facedetect/yolo,etc.")
   flag.StringVar(&id, "id", util.FakeSdkID, "sdk id as ipc label")

   flag.StringVar(&configPath, util.ConfigPath, "", "sdk config file path")
@@ -90,12 +92,6 @@

func main() {

   indices := gogpu.RankGPU()
   if len(indices) == 0 {
      logo.Errorln("there is no gpu for valid")
      return
   }

   util.InitConfig()

   flag.Parse()
@@ -115,11 +111,14 @@

   ret := false
   if role == roleMaster {
      defer dump(util.LogFile + "_dump_" + time.Now().Format("20060102150405"))
      setParamters()
      ret = master.Run(ctx, "./libcomm.so", configPath)

   } else if role == roleSlave {
      ret = slave.Run(ctx, configPath, runType, id, gpu, shm)
   } else if role == "test" {
      ret = slave.TwoPluginConflict("./libcomm.so", "./zconf/Yolo.json")
   }

   if ret {
@@ -144,3 +143,26 @@
// viper.AddConfigPath("./")

// envString := ""

func dump(dumpFile string) {
   dumpErr := recover()
   if dumpErr == nil {
      return
   }
   f, err := os.Create(dumpFile)
   if err != nil {
      return
   }
   f.WriteString(fmt.Sprintf("panic:\r\n%v", dumpErr))
   f.WriteString("\r\n\nstack:\r\n")
   f.WriteString(string(debug.Stack()))
   f.Close()

   //////////////////////////////////////////////
   // restart master

   cmd := exec.Command(os.Args[0], os.Args[1:]...)
   cmd.Stdout = os.Stdout
   cmd.Stderr = os.Stderr
   cmd.Start()
}