From 5459ba1d3f7f944aa97923ed9c09a5dbc7663928 Mon Sep 17 00:00:00 2001 From: zhangmeng <775834166@qq.com> Date: 星期二, 14 一月 2020 18:01:24 +0800 Subject: [PATCH] update --- run.go | 15 ++++++++++++--- 1 files changed, 12 insertions(+), 3 deletions(-) diff --git a/run.go b/run.go index dae12d4..fd70be9 100644 --- a/run.go +++ b/run.go @@ -6,10 +6,12 @@ "fmt" "io/ioutil" "os" + "strings" "time" "unsafe" - "basic.com/libgowrapper/sdkhelper.git" + "reid/common" + "basic.com/valib/gogpu.git" "basic.com/pubsub/protomsg.git" @@ -34,7 +36,7 @@ // Create Reid func Create(config string, typ, id string, gpu int, shm bool, ipc2Rule string, ruleMaxSize int, fn func(...interface{}), reserved map[string]interface{}) interface{} { - cfg, err := sdkhelper.ReadConfig(config) + cfg, err := common.ReadConfig(config) if err != nil { fn("Reid SDK Create Error When Read Config: ", err) return nil @@ -54,7 +56,7 @@ } } - gpuM := sdkhelper.Atoi(cfg.Param[sGPU]) + gpuM := common.Atoi(cfg.Param[sGPU]) rGPU := gpu @@ -99,6 +101,13 @@ for { if err = sock.Listen(s.ipc); err != nil { + suf := "ipc://" + p := strings.Index(s.ipc, suf) + if p >= 0 { + file := s.ipc[p+len(string(suf)):] + os.Remove(file) + s.fnLogger("remove:", file) + } s.fnLogger("can't listen on rep socket: ", err) time.Sleep(5 * time.Millisecond) } else { -- Gitblit v1.8.0