polygon---

panlei

2019-09-20 fba572d554d849777ded0132a47dcf4c7e1950de

 insertdata/insertDataToEs.go

@@ -5,7 +5,6 @@
   "encoding/json"
   "errors"
   "fmt"
   "io/ioutil"
   "net"
   "ruleprocess/cache"
   "ruleprocess/logger"
@@ -13,7 +12,6 @@
   "time"

   "basic.com/pubsub/protomsg.git"
   "github.com/go-yaml/yaml"
   "github.com/golang/protobuf/proto"
   "github.com/satori/go.uuid"
   "ruleprocess/ruleserver"
@@ -31,18 +29,10 @@
   DbTablePersons string `yaml:"dbTablePersons"`
}

func init() {
   data, err := ioutil.ReadFile("./config/conf.yml")
   if err != nil {
      fmt.Println("读取配置文件出错--", err)
      logger.Error("读取配置文件出错--", err)
   }
   c := conf{}
   //把yaml形式的字符串解析成struct类型
   yaml.Unmarshal(data, &c)
   weedfsUrl = c.PhotoUrl
   videoPersonUrl = c.VideoPersons
   personAction = c.PersonAction
func InitInsertEs() {
   weedfsUrl = "http://"+WeedFs.Ip+":"+strconv.Itoa(WeedFs.UploadPort)+"/submit"
   videoPersonUrl = "http://"+EsInfo.Masterip+":"+EsInfo.Httpport+"/"+EsInfo.EsIndex.VideoPersons.IndexName+"/"+EsInfo.EsIndex.VideoPersons.IndexType
   personAction = "http://"+EsInfo.Masterip+":"+EsInfo.Httpport+"/"+EsInfo.EsIndex.Personaction.IndexName+"/"+EsInfo.EsIndex.Personaction.IndexType
}

// 人脸的数据结构
@@ -145,6 +135,11 @@
//   //}
//}
func InsertToEs(msg ruleserver.ResultMsg) {
   defer func() {
      if err := recover(); err != nil {
         logger.Error("es模块儿的异常捕获：",err)
      }
   }()
   InsertFace(msg, "")
   // 如果标签中含有持续时间首次报警的timeLabel的话则不需要过人体追踪，不然就没的插入了
   fk := ruleserver.TrackOrNot(msg.RuleResult)
@@ -170,6 +165,8 @@
      faces = PutFace(faces,msg)
      //logger.Info("整理后的数据：",faces)
      if faces != nil {
         var imgMaxUrl string = ""
         var picTime string = ""
         for _,face := range faces {
            // 上传大图
            // 解压缩并上传图片
@@ -186,9 +183,13 @@
            }
            i := protomsg.Image{}
            err = proto.Unmarshal(bdata, &i)
            bigPhotoUrl := make(map[string]interface{})
            bigPhotoUrl, err = util.PostFormBufferData(weedfsUrl, i, uuid.NewV4().String())
            logger.Debug("========大图路径：", bigPhotoUrl)
            if imgMaxUrl == "" {
               bigPhotoUrl := make(map[string]interface{})
               bigPhotoUrl, err = util.PostFormBufferData(weedfsUrl, i, uuid.NewV4().String())
               logger.Debug("========大图路径：", bigPhotoUrl)
               imgMaxUrl = bigPhotoUrl["fileUrl"].(string)
               picTime = i.Timestamp
            }
            // 人脸检测，没有相似的底库人员
            localConfig, err := cache.GetServerInfo()
            if err != nil {
@@ -227,8 +228,8 @@
               esDataId,
               msg.Cid,
               camera.Addr,
               i.Timestamp,
               bigPhotoUrl["fileUrl"].(string),
               picTime,
               imgMaxUrl,
               msg.Tasklab.Taskid,
               msg.Tasklab.Taskname,
               "人脸",
@@ -366,7 +367,9 @@
                     } else {
                        logger.Info("上传的图片信息：", resp1)
                     }
                     url = append(url, resp1["fileUrl"].(string))
                     if resp1["fileUrl"] != nil {
                        url = append(url, resp1["fileUrl"].(string))
                     }
                  }
               }
            }