panlei
2019-07-11 95652c49ee7c0a4a9b167c6bcbe585ebd58ab2e5
insertdata/insertDataToEs.go
@@ -1,13 +1,13 @@
package insertdata
import (
   "ruleprocess/logger"
   "encoding/json"
   "errors"
   "fmt"
   "io/ioutil"
   "net"
   "ruleprocess/cache"
   "ruleprocess/logger"
   "strings"
   "time"
@@ -20,20 +20,18 @@
)
var weedfsUrl string
type conf struct {
   PhotoUrl string
   PhotoUrl string `yaml:"photoUrl"`
}
func init() {
   logger.Info("_________________________________________老子想看你执行了没")
   fmt.Println("_________________________________________老子想看你执行了没")
   data, _ := ioutil.ReadFile("conf.yml")
   fmt.Println("从配置文件中读取的值为:",string(data))
   data, err := ioutil.ReadFile("./config/conf.yml")
   if err != nil {
      fmt.Println("读取配置文件出错--",err)
      logger.Error("读取配置文件出错--",err)
   }
   c := conf{}
   //把yaml形式的字符串解析成struct类型
   yaml.Unmarshal(data, &c)
   fmt.Println("c.PhotoUrl的值为:",c.PhotoUrl)
   weedfsUrl = c.PhotoUrl
   fmt.Println("weedfsUrl的值为:",weedfsUrl)
   logger.Info("weedfsUrl的值为:",weedfsUrl)
}
// 人脸的数据结构
@@ -61,6 +59,7 @@
   AnalyServerName string   `json:"analyServerName"`
   AnalyServerIp   string   `json:"analyServerIp"`
   ClusterId       string   `json:"clusterId"`
   DetectScore      float64    `json:"detectScore"`
   IsAlarm         int      `json:"isAlarm"`
   IsAckAlarm      int      `json:"isAckAlarm"`
   IsCollect       int      `json:"isCollect"`
@@ -102,7 +101,7 @@
func InsertToEs(msg ruleserver.ResultMsg) {
   // 直接从规则的标签数据里拿符合规则的人脸结果
   logger.Info("------------------------------------从配置文件读取的值:",weedfsUrl)
   //logger.Info("------------------------------------从配置文件读取的值:",weedfsUrl)
   if msg.RuleResult["face"] != nil && len(msg.RuleResult["face"].([]ruleserver.Arg)) > 0 {
      logger.Info("往ES插人脸数据")
      for _, face := range msg.RuleResult["face"].([]ruleserver.Arg) {
@@ -136,7 +135,7 @@
            }
            i := protomsg.Image{}
            err = proto.Unmarshal(bdata, &i)
            //logger.Info("-------------------------------------------看下宽和高", i.Width, i.Height)
            logger.Info("-------------------------------------------看下宽和高", int(face.Location.X),int(face.Location.Y),int(face.Location.X+face.Location.Width),int(face.Location.Y+face.Location.Height))
            bytes := util.SubImg(i, int(face.Location.X), int(face.Location.Y), int(face.Location.X+face.Location.Width), int(face.Location.Y+face.Location.Height))
            resp, err := util.PostFormBufferData1(weedfsUrl, bytes, uuid.NewV4().String())
            if err != nil {
@@ -176,6 +175,7 @@
               localConfig.ServerName,
               serverIp,
               "",
               face.Score,
               1,
               0,
               0,
@@ -189,10 +189,14 @@
               return
            }
            err = EsReq("POST", "http://192.168.1.182:9200/videopersons/perVideoPicture", requstbody)
            err1 := EsReq("POST", "http://192.168.1.182:9200/videopersons/perVideoPicture", requstbody)
            if err1 != nil {
               logger.Error("上传ES出错!---",err1)
            }
            //logger.Info(err.Error())
         } else {
            // 人脸提取
            // 人脸比对
            logger.Warn("___________________________________________这是有baseinfo的")
            localConfig, err := cache.GetServerInfo()
            if err != nil {
               logger.Info("查询本机信息失败!")
@@ -249,6 +253,7 @@
               localConfig.ServerName,
               serverIp,
               "",
               face.Score,
               1,
               0,
               0,
@@ -347,7 +352,8 @@
      if err != nil {
         logger.Error("往ES插入数据失败", err)
      } else {
         logger.Info("__________________________________________往ES插入yolo数据成功")
         logger.Warn("__________________________________________往ES插入yolo数据成功")
         //os.Exit(1)
      }
   }
}