add common socket protocol

龙赣华

2019-05-20 c7c180fcbe012173e56bde6141ae4b2ed3827ab4

 camera/camera.go

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

import (
   "errors"

   "github.com/long/test/deliver"
   "github.com/long/test/httpclient"
   "github.com/long/test/sdk"

   "nanomsg.org/go-mangos"
   "nanomsg.org/go-mangos/protocol/pair"
   "nanomsg.org/go-mangos/transport/ipc"
   "nanomsg.org/go-mangos/transport/tcp"

   "context"
   "encoding/json"
@@ -30,7 +28,7 @@
}

type SocketContext struct {
   Sock    mangos.Socket
   Sock    deliver.Deliver
   Context context.Context
   Cancel  context.CancelFunc
}
@@ -38,7 +36,6 @@
func Taskdolist(cid string, taskid string, data []byte) {

   fmt.Println("======================================")
   fmt.Println()
   //  数据加工(打标签)
   sdkmsg := sdk.SdkData(cid, taskid, data)
   if sdkmsg.Tasklab == nil {
@@ -64,10 +61,12 @@
   fmt.Println()
   fmt.Println("cid,taskid , sid: ", CtsId)

   _, socket, err := NewCamerSocketListen("init", "tcp", "192.168.1.124", 0)
   url := fmt.Sprintf("%s%d", "tcp://192.168.1.124:", UrlPort)
   _, socket, err := NewCamerSocketListen(deliver.Pair, "init", url)
   if err != nil {
      return
   }
   UrlPort++

   go SendRecv(socket, Initchannel)

@@ -87,96 +86,37 @@
   }
}

func Recv(cameraid string, socket SocketContext) {
   socket.Sock.SetOption(mangos.OptionRecvDeadline, 5*time.Second)
   var msg []byte
   var err error
   for {
      select {
      case <-socket.Context.Done():
         fmt.Println("listen recv quit")
         return
      default:
         if msg, err = socket.Sock.Recv(); err != nil {
            //fmt.Printf("%s ", err)
            continue
         } else {
            for _, taskid := range GetAlltask(cameraid) {
               go Taskdolist(cameraid, taskid, msg)
               fmt.Println("receive: ", len(msg), "cameraid: ", cameraid, "taskid: ", taskid)
            }
         }
      }
   }
}

func send(socket SocketContext, cam Camerdata) {
   b, err := json.Marshal(cam)
   if err != nil {
      fmt.Println("can not json convert !", cam)
   }
   if err := socket.Sock.Send(b); err != nil {
      fmt.Println("failed send")
      fmt.Println("camera info: failed send")
   }
}

func SendRecv(socket SocketContext, camera chan Camerdata) {
   socket.Sock.SetOption(mangos.OptionRecvDeadline, 5*time.Second)
   //socket.Sock.SetOption(mangos.OptionRecvDeadline, 5*time.Second)
   for cam := range camera {
      send(socket, cam)
      if _, ok := SocketManage[cam.Cameraid]; !ok {
         go func(id string) {
         go func(id string, urlport int) {
            fmt.Println("create cid server: ", id)
            cid, socketlisten, _ := NewCamerSocketListen(id, "tcp", "192.168.1.124", 0)
            url := fmt.Sprintf("%s%d", "tcp://192.168.1.124:", urlport)
            cid, socketlisten, err := NewCamerSocketListen(deliver.PushPull, id, url)

            if err != nil {
               return
            }
            fmt.Println("input id: ", id, " output id :", cid)
            Recv(cid, socketlisten)

         }(cam.Cameraid)
      }
   }
}
         }(cam.Cameraid, UrlPort)

// new camera server
func NewCamerSocketListen(cameraid string, protocol string, ip string, port int) (cid string, socket SocketContext, err error) {

   ctx, cancel := context.WithCancel(context.Background())

   var url string

   socket.Context = ctx
   socket.Cancel = cancel

   switch protocol {
   case "tcp":
      if port == 0 {
         port = UrlPort
         UrlPort++
      }
      url = fmt.Sprintf("%s://%s:%d", protocol, ip, port)
   case "ipc":
      url = fmt.Sprintf("%s://%s", cameraid)
   }
   fmt.Println(url)

   if socket.Sock, err = pair.NewSocket(); err != nil {
      fmt.Println(cameraid, "can't get new pair socket: ", err.Error())
      return cameraid, socket, err
   }
   socket.Sock.AddTransport(tcp.NewTransport())
   socket.Sock.AddTransport(ipc.NewTransport())

   socket.Sock.SetOption(mangos.OptionMaxRecvSize, 32*1024*1024)
   socket.Sock.SetOption(mangos.OptionWriteQLen, 10)
   socket.Sock.SetOption(mangos.OptionReadQLen, 10)

   if err = socket.Sock.Listen(url); err != nil {
      fmt.Println("socket lisnte error ", cameraid)
   }
   fmt.Println(cameraid, "success")

   SocketManage[cameraid] = socket

   return cameraid, socket, err
}

// 获取 cid ， taskid, sdkid 关系
@@ -198,3 +138,46 @@
   }
   return
}

// create server
func NewCamerSocketListen(mode int, cameraid string, url string) (cid string, socket SocketContext, err error) {
   fmt.Println("url is: ", url)
   ctx, cancel := context.WithCancel(context.Background())

   socket.Context = ctx
   socket.Cancel = cancel

   socket.Sock = deliver.NewServer(deliver.Mode(mode), url)
   fmt.Println("new socket.Sock: ", socket.Sock)

   if socket.Sock == nil {
      return cameraid, socket, errors.New("create listen error")
   }

   SocketManage[cameraid] = socket
   return cameraid, socket, nil
}

func Recv(cameraid string, socket SocketContext) {
   //   socket.Sock.SetOption(mangos.OptionRecvDeadline, 5*time.Second)
   var msg []byte
   var err error
   for {
      select {
      case <-socket.Context.Done():
         fmt.Println("listen recv quit")
         return
      default:
         if msg, err = socket.Sock.Recv(); err != nil {
            //fmt.Printf("%s ", err)
            continue
         } else {
            fmt.Println("cameraid: ", len(msg))
            for _, taskid := range GetAlltask(cameraid) {
               Taskdolist(cameraid, taskid, msg)
               //   fmt.Println("receive: ", len(msg), "cameraid: ", cameraid, "taskid: ", taskid)
            }
         }
      }
   }
}