bug fixed memvirt leak

zhangmeng

2019-10-30 467d125ef6ac819cc1e8bd1a8b5dcd00d491f624

 camera/camera.go

@@ -1,19 +1,13 @@
package camera

import (
   "basic.com/valib/deliver.git"
   //"sync"
   "taskpubsub/logger"

   "basic.com/pubsub/protomsg.git"
   "github.com/gogo/protobuf/proto"
   "taskpubsub/sdk"
   "taskpubsub/util"

   // "golang.org/x/sys/unix"

   "fmt"
   //"time"
   "basic.com/pubsub/protomsg.git"
   "basic.com/valib/deliver.git"
   "basic.com/valib/logger.git"
   "github.com/gogo/protobuf/proto"
)

const (
@@ -30,9 +24,9 @@
}

func initInnerTopic() {
   for _, recvTopic := range innerRecvTopic {
      createCameraRecvServerAndListen(recvTopic)
   }
   // for _, recvTopic := range innerRecvTopic {
   //    createCameraRecvServerAndListen(recvTopic)
   // }
}

func Init(useShm bool) {
@@ -59,16 +53,15 @@
func createCameraRecvServerAndListen(id string) {
   if _, isExist := SocketManage[id]; !isExist { //不存在

      url := fmt.Sprintf("ipc:///tmp/%s.ipc", id)
      url := "ipc:///tmp/" + id + ".ipc"
      m := deliver.PushPull
      if shm{
      if shm {
         m = deliver.Shm
         url = id

         // unix.Unlink("/dev/shm/" + url)
      }
		
      fmt.Println("ipc url: ", url)

      logger.Info("CAMERA URL : ", url)
      socket, err := util.NewSocketListen(int(m), url, shm)
      if err != nil {
         logger.Error("create socket error")
@@ -124,6 +117,8 @@

func Recv(socket util.SocketContext) {

   tryCount := 0

   var recvmessage []byte
   var imagemsg protomsg.Image
   var err error
@@ -135,6 +130,19 @@
      default:
         if recvmessage, err = socket.Sock.Recv(); err != nil {
            //logger.Error("[camera] err is: ", err)

            if socket.UseSHM {
               if tryCount > util.SHMMaxTryCount {
                  socket.Sock.Close()
                  socket, err = util.NewSocketListen(socket.Mode, socket.URL, socket.UseSHM)
                  logger.Info("CAMERA SHM TRY :", tryCount, " RESTART IT")

                  tryCount = 0

                  continue
               }
               tryCount++
            }
            continue
         }

@@ -149,6 +157,7 @@
            continue
         }
         if faceExtractWebCID == imagemsg.Cid { //以图搜图
            logger.Info("id: ", faceExtractWebTaskID)
            doTaskList(imagemsg.Cid, "", faceExtractWebTaskID, recvmessage)
         } else {
            taskIDs := GetAllTaskByID(imagemsg.Cid)
@@ -166,6 +175,9 @@
   for _, camsingle := range util.CameraTasks {
      if cid == camsingle.Camera.Id {
         for _, tasksingle := range camsingle.Tasks {
            if !tasksingle.Enable {
               continue
            }
            tasks = append(tasks, tasksingle.Taskid)
         }
         return
@@ -177,7 +189,7 @@
func doTaskList(cid string, caddr string, taskid string, data []byte) {

   //  数据加工(打标签)
   logger.Debug("taskid: ", taskid, "has ", len(data), "data[]byte")
   //logger.Info("taskid: ", taskid, "has ", len(data), "data[]byte")
   sdkmsg := sdk.ToSdkMsg(cid, caddr, taskid, data)
   if sdkmsg.Tasklab == nil {
      logger.Error(cid, " not have taskid: ", taskid)
@@ -186,11 +198,19 @@

   //  计算分发的主题
   SendTopic := sdk.GetSdkSendTopic(sdkmsg)
   logger.Debug(SendTopic)
   //logger.Info(SendTopic)
   if "facedetect-sdk-no-track" == SendTopic || "virtual-faceextract-sdk-pull" == SendTopic {
      logger.Info(SendTopic)
   }
   if _, ok := sdk.SdkMap[SendTopic]; ok {
      if "facedetect-sdk-no-track" == SendTopic || "virtual-faceextract-sdk-pull" == SendTopic {
         logger.Info(SendTopic)
      }
      sdk.SdkMap[SendTopic] <- sdkmsg
      logger.Debug("dispute sendtopic success", SendTopic)
      if "facedetect-sdk-no-track" == SendTopic || "virtual-faceextract-sdk-pull" == SendTopic {
         logger.Info("dispute sendtopic success", SendTopic)
      }
   } else {
      logger.Debug("分发的主题不存在")
      logger.Info("分发的主题不存在")
   }
}