554325746@qq.com
2019-05-28 70df7b912014201e271a2966599b84d77addd0f9
sdk/sdk.go
@@ -5,17 +5,25 @@
   "errors"
   "fmt"
   "github.com/long/test/httpclient"
   "github.com/long/test/protomsg"
//   "github.com/long/test/httpclient"
   "github.com/long/test/tasktag"
   "github.com/long/test/util"
   "github.com/golang/protobuf/proto"
   "github.com/long/test/deliver"
   "basic.com/valib/deliver.git"
   "basic.com/pubsub/protomsg.git"
    "basic.com/dbapi.git"
)
//var doOnce sync.Once
const (
    postPull="_1.ipc"
    postPush="_2.ipc"
)
var SocketManage = make(map[string]SocketContext)
var sdkapi dbapi.SdkApi
type SocketContext struct {
   Sock    deliver.Deliver
@@ -25,32 +33,34 @@
func Init() {
    fmt.Println("============= init sdk info =====================")
   sdklist := SdkAll() //获取所有sdk
   fmt.Println("sdk list have: ", sdklist)
   SdkCreateTopic(sdklist) // 创建主题
   for _, sdkid := range sdklist { // 创建sdk server
      url := fmt.Sprintf("%s%d", "tcp://192.168.1.124:", UrlPort)
      sdkid, socket, err := NewSdkSocketListen(deliver.PushPull, sdkid, url)
        url := fmt.Sprintf("ipc:///tmp/%s%s",sdkid,postPull)
      sdkidser, socketser, err := NewSdkSocketListen(deliver.PushPull, sdkid, url)
      if err != nil {
         continue
      }
      go Send(sdkid, socket, SdkMap[sdkid])
      //break
      //   Recv(socket)
      go Send(sdkidser, socketser, SdkMap[sdkid])
        url = fmt.Sprintf("ipc:///tmp/%s%s",sdkid,postPush)
      _, socketdial, err := NewSdkSocketListen(deliver.PushPull, sdkid, url)
      if err != nil {
         continue
      }
      go Recv(socketdial)
   }
   go es(SdkMap["es"])
}
//单独处理   es 主题的情况
func es(sdkmsgchan chan *protomsg.SdkMessage) {
   for data := range sdkmsgchan {
      fmt.Println("this data is finish all sdk! ", data)
   for _ = range sdkmsgchan {
      fmt.Println("this data is finish all sdk! ")
   }
}
//动态处理
@@ -84,7 +94,6 @@
      sdkmsg.Tasklab = nil
      return sdkmsg
   }
   sdkmsg.Tasklab = tasktag.TaskMapLab[taskid]
   sdkmsg.Data = data
   return sdkmsg
@@ -97,15 +106,14 @@
   } else {
      sdksend = "es"
   }
   fmt.Println()
   fmt.Println("分发的主题是: ", sdksend)
   fmt.Println()
   return
}
// 调用  http 借口获取摄像机信息
func SdkAll() (sdklist []string) {
   sdklist = httpclient.GetSdk("http://127.0.0.1:8000/data/api-v/sdk/findskdid")
   //sdklist = httpclient.GetSdk("http://192.168.1.124:8000/data/api-v/sdk/findskdid")
    sdklist = sdkapi.GetAllSdkIds()
   return
}
@@ -122,7 +130,6 @@
   return nil
}
var UrlPort = 9000
// create server
func NewSdkSocketListen(mode int, sdkid string, url string) (sid string, socket SocketContext, err error) {
@@ -140,7 +147,24 @@
   }
   SocketManage[sdkid] = socket
   UrlPort++
   return sdkid, socket, nil
}
func NewSdkSocketDial(mode int, sdkid string, url string) (sid 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.NewClient(deliver.Mode(mode), url)
   fmt.Println(sdkid, socket.Sock)
   if socket.Sock == nil {
      return sdkid, socket, errors.New("create listen error")
   }
   SocketManage[sdkid] = socket
   return sdkid, socket, nil
}
@@ -159,6 +183,7 @@
            continue
         } else {
            err = proto.Unmarshal(msg, repsdkmsg)
            fmt.Println("receive len: ", len(msg))
            if err != nil {
               fmt.Println("unmarshal error: ", err)
               continue
@@ -182,19 +207,18 @@
         data, err := proto.Marshal(v)
         if err != nil {
            fmt.Println("proto marshal error ", err)
            continue
         }
         fmt.Printf("从管道sdkid=%s 接受数据 %d\n", sdkid, len(data))
         fmt.Println()
         fmt.Println("send len of data: ", len(data))
         if err := socket.Sock.Send(data); err != nil {
                fmt.Println(socket.Sock)
            fmt.Println("failed send")
            continue
         }
         fmt.Printf("sdkid = %s ,send success:%d \n", sdkid, len(data))
         //   go func(sock SocketContext) {
         //      doOnce.Do(func() { Recv(sock) })
         //   }(socket)
      }
   }
}