---

panlei

2019-07-02 aafaf19184a4d3ef5bb88a8613d9bc9d7c1f27e5

 main.go

@@ -1,25 +1,39 @@
package main

import (
   "basic.com/dbapi.git"
   "basic.com/pubsub/protomsg.git"
   "basic.com/valib/deliver.git"
   "bufio"
   "bytes"
   "flag"
   "fmt"
   "github.com/golang/protobuf/proto"
   "gocv.io/x/gocv"
   "image"
   "image/color"
   "log"
   "os"
   "ruleprocess/cache"
   "ruleprocess/insertdata"
   "ruleprocess/ruleserver"
   "ruleprocess/util"
   "sync"
)

var dbIp = flag.String("dbIp","127.0.0.1","dbserver ip")
var dbPort = flag.Int("dbPort",8001,"default dbPort=8001")
var surveyPort = flag.Int("surveyPort",40007,"survey port") //心跳
var pubPort = flag.Int("pubPort",50007,"pubsub port") //数据更新
var initchan = make(chan bool)
func main() {
   flag.Parse()
   wg := sync.WaitGroup{}
   wg.Add(1)

   dbapi.Init(*dbIp,*dbPort)
   go cache.Init(initchan,*dbIp,*surveyPort,*pubPort)
   fmt.Println("cache init completed!!!",<- initchan)//dbserver初始化完毕

   go ruleserver.TimeTicker()
   nReciever("ipc:///tmp/sdk-2-rules-process.ipc", deliver.PushPull, 1)
   wg.Wait()
@@ -96,7 +110,7 @@
   }
   i := protomsg.Image{}
   err = proto.Unmarshal(bdata, &i)

   log.Println("----------------看看有几个算法：",len(m.Tasklab.Sdkinfos))
   for _, sdkinfo := range m.Tasklab.Sdkinfos { // yolo算法
      if sdkinfo.Sdktype == "Yolo" {
         arg := ruleserver.SdkData{}
@@ -123,6 +137,7 @@
               fmt.Println("解析YOLO sdk数据时出现错误", err)
               continue
            }
            log.Println("------当前算法name",sdkinfo.SdkName,"--------------看看yolo中有几个目标：",len(yoloParam.Infos))
            for _, info := range yoloParam.Infos {
               if info.Typ == 0 {
                  photoMap := ruleserver.PhotoMap{Rects: rectFormat(info.RcObj), Score: float64(info.Prob)*100,IsYolo:true}