调整配置文件的引用位置

panlei

2019-07-09 b2f549332f6f890b252c3cc6abcbd35fdc6ae2e5

 insertdata/insertDataToEs.go

@@ -1,12 +1,13 @@
package insertdata

import (
   "ruleprocess/logger"
   "encoding/json"
   "errors"
   "fmt"
   "io/ioutil"
   "net"
   "ruleprocess/cache"
   "ruleprocess/logger"
   "strings"
   "time"

@@ -15,9 +16,30 @@
   "github.com/satori/go.uuid"
   "ruleprocess/ruleserver"
   "ruleprocess/util"
   "github.com/go-yaml/yaml"
)
var weedfsUrl string
type conf struct {
   PhotoUrl string
}
func init() {
   logger.Info("_________________________________________老子想看你执行了没")
   fmt.Println("_________________________________________老子想看你执行了没")
   data, err := ioutil.ReadFile("./config/conf.yml")
   if err != nil {
      fmt.Println("读取配置文件出错--",err)
      logger.Error("读取配置文件出错--",err)
   }
   fmt.Println("从配置文件中读取的值为：",string(data))
   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)
}

var weedfsUrl = "http://192.168.1.182:6333/submit"
// 人脸的数据结构
type PerVideoPicture struct {
   Id              string   `json:"id"`
@@ -84,6 +106,7 @@
func InsertToEs(msg ruleserver.ResultMsg) {

   // 直接从规则的标签数据里拿符合规则的人脸结果
   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) {
@@ -102,7 +125,7 @@
            // 人脸检测，没有相似的底库人员
            localConfig, err := cache.GetServerInfo()
            if err != nil {
               logger.Info("查询本机信息失败！")
               logger.Error("查询本机信息失败！")
            }
            serverIp, err := GetLocalIP()
            // 解压缩并上传图片
@@ -113,16 +136,17 @@
            // 查询cameraName
            camera, err := cache.GetCameraById(msg.Cid)
            if err != nil {
               logger.Info("查询摄像机信息失败")
               logger.Error("查询摄像机信息失败")
            }
            i := protomsg.Image{}
            err = proto.Unmarshal(bdata, &i)
            logger.Info("-------------------------------------------看下宽和高", i.Width, i.Height)
            //logger.Info("-------------------------------------------看下宽和高", i.Width, i.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 {
               logger.Info("上传小图出错")
               logger.Error("上传小图出错")
            }
            logger.Info("================小图地址:",resp["fileUrl"].(string))
            sex := ""
            if face.ThftRes.Gender == 1 {
               sex = "男"
@@ -149,7 +173,7 @@
               race,
               face.ThftRes.Smile,
               face.ThftRes.Beauty,
               "",
               string(face.Feature),
               []string{strings.Split(resp["fileUrl"].(string), "/")[1]},
               "暂无集群",
               localConfig.ServerId,
@@ -186,16 +210,17 @@
            // 查询cameraName
            camera, err := cache.GetCameraById(msg.Cid)
            if err != nil {
               logger.Info("查询摄像机信息失败")
               logger.Error("查询摄像机信息失败")
            }
            i := protomsg.Image{}
            err = proto.Unmarshal(bdata, &i)
            logger.Info("-------------------------------------------看下宽和高", i.Width, i.Height)
            //logger.Info("-------------------------------------------看下宽和高", i.Width, i.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 {
               logger.Info("上传小图出错")
               logger.Error("上传小图出错")
            }
            logger.Info("================小图地址:",resp["fileUrl"].(string))
            sex := ""
            if face.ThftRes.Gender == 1 {
               sex = "男"
@@ -237,7 +262,7 @@
            requstbody, err := json.Marshal(pervideo)

            if err != nil {
               logger.Info("json parse error ", err)
               logger.Error("json parse error ", err)
               return

            }
@@ -272,7 +297,7 @@
         //resp, err = util.PostFormBufferData(weedfsUrl, i, uuid.NewV4().String())
         resp, err = util.DrawPolygonOnImage(msg.Cid, i,msg.RuleResult["yolo"].([]ruleserver.Result))
         if err != nil {
            logger.Info("画框或上传图片服务器出错", err)
            logger.Error("画框或上传图片服务器出错", err)
         } else {
            logger.Info("已报警并上传改帧图片到服务器")
         }
@@ -285,12 +310,12 @@
      // 查询本机信息
      localConfig, err := cache.GetServerInfo()
      if err != nil {
         logger.Info("查询本机信息失败！")
         logger.Error("查询本机信息失败！")
      }
      // 查询cameraName
      camera, err := cache.GetCameraById(msg.Cid)
      if err != nil {
         logger.Info("查询摄像机信息失败")
         logger.Error("查询摄像机信息失败")
      }
      serverIp, err := GetLocalIP()
      peraction := Personaction{
@@ -324,7 +349,7 @@
      }
      err = EsReq("POST", "http://192.168.1.182:9200/personaction/perVideoAction", requstbody)
      if err != nil {
         logger.Info("往ES插入数据失败", err)
         logger.Error("往ES插入数据失败", err)
      } else {
         logger.Info("__________________________________________往ES插入yolo数据成功")
      }