From 330a9e5cb636286cdb399cd305856e5f419c540f Mon Sep 17 00:00:00 2001
From: zhangmeng <775834166@qq.com>
Date: 星期五, 13 十二月 2019 17:22:33 +0800
Subject: [PATCH] remove no use code
---
sdk/sdk.go | 109 +++++++++++++++++++++++++++++-------------------------
1 files changed, 59 insertions(+), 50 deletions(-)
diff --git a/sdk/sdk.go b/sdk/sdk.go
index 42bff5d..0941757 100644
--- a/sdk/sdk.go
+++ b/sdk/sdk.go
@@ -1,18 +1,14 @@
package sdk
import (
- "fmt"
- // "golang.org/x/sys/unix"
-
"taskpubsub/tasktag"
"taskpubsub/util"
"github.com/gogo/protobuf/proto"
- "taskpubsub/logger"
-
"basic.com/pubsub/protomsg.git"
"basic.com/valib/deliver.git"
+ "basic.com/valib/logger.git"
)
const (
@@ -25,25 +21,6 @@
var shm bool = false
-var innerRecvTopic = []string{
- "facedetect-sdk-no-track", //to sdk-no-track 浠ュ浘鎼滃浘
-}
-
-var innerSendTopic = []string{
- "facedetect-sdk-no-track", //to sdk-no-track 浠ュ浘鎼滃浘
- "virtual-faceextract-sdk-pull", //to web 浠ュ浘鎼滃浘
-}
-
-func initInnerTopic() {
- for _, sendTopic := range innerSendTopic {
- createSdkSendServerAndListen(sendTopic)
- }
-
- for _, recvTopic := range innerRecvTopic {
- createSdkRecvServerAndListen(recvTopic)
- }
-}
-
func Init(useShm bool) {
shm = useShm
@@ -51,16 +28,7 @@
logger.Info("============= init sdk info =====================")
for _, sdkid := range util.Sdklist { // 鍒涘缓sdk server
createSdkTopicAndServer(sdkid)
- logger.Info()
}
-
- // 鎵嬪姩杈撳叆鐨勪富棰�
- initInnerTopic()
-
- // es
- SdkMap["es"] = make(chan protomsg.SdkMessage)
- logger.Info("create es channel: ")
- go DealEsTopic()
go autoUpdateSdk(util.Sdkflag)
}
@@ -87,17 +55,14 @@
func createSdkSendServerAndListen(id string) {
if _, isExist := SdkMap[id]; !isExist { //涓嶅瓨鍦�
SdkMap[id] = make(chan protomsg.SdkMessage)
- logger.Info("create", id)
}
- url := fmt.Sprintf("ipc:///tmp/%s%s.ipc", id, postPush)
+ url := "ipc:///tmp/" + id + postPush + ".ipc"
m := deliver.PushPull
if shm {
m = deliver.Shm
url = id + postPush
}
-
- fmt.Println("SDK URL: ", url)
socket, err := util.NewSocketListen(int(m), url, shm)
if err != nil {
@@ -105,7 +70,9 @@
logger.Error(id, "create socket error!")
return
}
+
SocketManage[id] = socket
+ logger.Info("SDK URL Send: ", url)
go Send(id, socket, SdkMap[id])
}
@@ -116,7 +83,7 @@
logger.Info("create", id)
}
- url := fmt.Sprintf("ipc:///tmp/%s%s.ipc", id, postPull)
+ url := "ipc:///tmp/" + id + postPull + ".ipc"
m := deliver.PushPull
if shm {
m = deliver.Shm
@@ -131,7 +98,9 @@
logger.Error(id, "create socket error!")
return
}
+
SocketManage[id] = socket
+ logger.Info("SDK URL Recv: ", url)
go Recv(socket)
}
@@ -157,14 +126,6 @@
}
newSdkList := util.Sdklist
-
- // 鎵嬪姩娣诲姞鐨勫叏閮ㄥ姞涓�
- for _, sendTopic := range innerSendTopic {
- newSdkList = append(newSdkList, sendTopic)
- }
- for _, recvTopic := range innerRecvTopic {
- newSdkList = append(newSdkList, recvTopic)
- }
sdkListUpdate := util.Difference(oldSdkList, newSdkList)
logger.Info(sdkListUpdate)
@@ -198,24 +159,51 @@
func GetSdkSendTopic(sdkmsg protomsg.SdkMessage) (sendTopic string) {
if int(sdkmsg.Tasklab.Index) < len(sdkmsg.Tasklab.Sdkinfos) {
sendTopic = sdkmsg.Tasklab.Sdkinfos[sdkmsg.Tasklab.Index].Ipcid
+ // if "Yolo" != sdkmsg.Tasklab.Sdkinfos[sdkmsg.Tasklab.Index].Sdktype &&
+ // "FaceDetect" != sdkmsg.Tasklab.Sdkinfos[sdkmsg.Tasklab.Index].Sdktype &&
+ // "FaceCompare" != sdkmsg.Tasklab.Sdkinfos[sdkmsg.Tasklab.Index].Sdktype {
+ // //if sdkmsg.Tasklab.Sdkinfos[sdkmsg.Tasklab.Index].Sdktype == "" {
+ // logger.Info("----------Sdktype:yitusoutu")
+ // logger.Info("鍒嗗彂鐨勪富棰�:", sendTopic, "!Sdktype锛�", sdkmsg.Tasklab.Sdkinfos[sdkmsg.Tasklab.Index].Sdktype)
+ // }
} else {
sendTopic = "es"
}
- logger.Debug("鍒嗗彂鐨勪富棰�: ", sendTopic, "浣嶇疆锛�", int(sdkmsg.Tasklab.Index)+1, "/", len(sdkmsg.Tasklab.Sdkinfos))
- return
+
+ return sendTopic
}
func Recv(socket util.SocketContext) {
+
+ tryCount := 0
+
var repsdkmsg = protomsg.SdkMessage{}
for {
select {
case <-socket.Context.Done():
+ socket.Sock.Close()
+
logger.Info("socket close")
return
default:
if msg, err := socket.Sock.Recv(); err != nil {
+
+ // tryCount++
+ // socket = util.MaybeRestartSocket(socket, &tryCount)
+
+ if socket.UseSHM {
+ if tryCount > util.ShmMaxTryCount {
+ socket.Sock.Close()
+ socket, err = util.NewSocketListen(socket.Mode, socket.URL, socket.UseSHM)
+ // logger.Info("SDK SEND SHM TRY :", util.ShmMaxTryCount, " RESTART IT")
+ tryCount = 0
+ continue
+ }
+ tryCount++
+ }
continue
} else {
+
err = proto.Unmarshal(msg, &repsdkmsg)
if err != nil {
logger.Error("unmarshal error: ", err)
@@ -225,6 +213,7 @@
//璋冪敤璁＄畻鍑芥暟锛� 鍒嗗彂缁欎笅涓�涓富棰�
nexttopic := GetSdkSendTopic(repsdkmsg)
SdkMap[nexttopic] <- repsdkmsg
+ logger.Info("recv from URL: ", socket.URL, " success: ", len(msg), " send to: ", repsdkmsg)
}
}
}
@@ -232,9 +221,13 @@
func Send(sdkid string, socket util.SocketContext, in chan protomsg.SdkMessage) {
+ tryCount := 0
+
for {
select {
case <-socket.Context.Done():
+ socket.Sock.Close()
+
logger.Info("socket is close")
return
case v, ok := <-in:
@@ -246,10 +239,26 @@
}
if err := socket.Sock.Send(data); err != nil {
- logger.Error("failed send:sdkid=", sdkid)
+
+ // tryCount++
+ // socket = util.MaybeRestartSocket(socket, &tryCount)
+
+ if socket.UseSHM {
+ if tryCount > util.ShmMaxTryCount {
+ socket.Sock.Close()
+ socket, err = util.NewSocketListen(socket.Mode, socket.URL, socket.UseSHM)
+ // logger.Info("SDK SEND SHM TRY :", util.ShmMaxTryCount, " RESTART IT")
+ tryCount = 0
+ continue
+ }
+ tryCount++
+ }
+
continue
}
- logger.Debug(sdkid, " send success: ", len(data))
+
+ logger.Info("send to sdk id: ", sdkid, " success: ", len(data))
+
} else {
logger.Debug(sdkid, " 涓婚鍏抽棴, 鍏抽棴send()")
return
--
Gitblit v1.8.0