From 53941272739616d56cf980afe81bd0f5e164e11d Mon Sep 17 00:00:00 2001 From: panlei <2799247126@qq.com> Date: 星期三, 30 十月 2019 20:03:48 +0800 Subject: [PATCH] 一个socket拨一个号 --- ruleserver/readyDataForRule.go | 445 +++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 files changed, 445 insertions(+), 0 deletions(-) diff --git a/ruleserver/readyDataForRule.go b/ruleserver/readyDataForRule.go new file mode 100644 index 0000000..15f781f --- /dev/null +++ b/ruleserver/readyDataForRule.go @@ -0,0 +1,445 @@ +package ruleserver + +import ( + "basic.com/dbapi.git" + "basic.com/pubsub/protomsg.git" + "encoding/json" + "errors" + "fmt" + "github.com/golang/protobuf/proto" + "math" + "nanomsg.org/go-mangos" + "nanomsg.org/go-mangos/protocol/req" + "nanomsg.org/go-mangos/transport/tcp" + "net" + "ruleprocess/cache" + "basic.com/valib/logger.git" + "ruleprocess/structure" + "strconv" + "time" +) + +var serverIp, serverPort, dbTablePersons string +var sock mangos.Socket +type conf struct { + PhotoUrl string `yaml:"photoUrl"` + VideoPersons string `yaml:"videoPersons"` + PersonAction string `yaml:"personAction"` + ServerIp string `yaml:"serverIp"` + ServerPort string `yaml:"serverPort"` + DbTablePersons string `yaml:"dbTablePersons"` +} + +func Init() { + //logger.Debug("鏈満淇℃伅鍜宻erver淇℃伅锛�", localConfig, serverIp, serverPort) + //bigCache.Init(dbTablePersons, serverIp, serverPort, localConfig.ServerId) + go Push1() +} + +var sender chan *protomsg.CompareArgs = make(chan *protomsg.CompareArgs) +var receiver chan []byte = make(chan []byte) +type BaseInfo struct { + TableId string `json:"tableId"` + TableName string `json:"tableName"` + BwType string `json:"bwType"` + CompareScore float64 `json:"compareScore"` + PersonId string `json:"personId"` + PersonName string `json:"personName"` + PersonPicUrl string `json:"personPicUrl"` + PhoneNum string `json:"phoneNum"` + Sex string `json:"sex"` + IdCard string `json:"idCard"` + MonitorLevel string `json:"monitorLevel"` + Content string `json:"content"` +} + +// 浠ユ憚鍍忔満id鏌ュ嚭璺熷叾鐩稿叧鐨勬墍鏈変换鍔′笅鐨勬墍鏈夎鍒欑粍 +func GetRuleGroup(cameraId string, taskId string) *protomsg.TaskGroupArgs { + all := cache.GetCameraTaskRulesByCameraId(cameraId) + //logger.Info("==================鎵�鏈夌殑浠诲姟鐨勮鍒欑粍锛�",all) + var taskGroup *protomsg.TaskGroupArgs + for _, task := range all { + if task.TaskId == taskId { + taskGroup = task + } + } + logger.Debug("褰撳墠鏁版嵁甯ц鍖归厤鐨勮鍒欑粍锛�-------------------------","鎽勫儚鏈篿d:",cameraId,"浠诲姟id",taskId) + if taskGroup == nil { + return nil + } else { + for _,ruleGroup := range taskGroup.GroupRules{ + logger.Info(ruleGroup.GroupText) + } + return taskGroup + } +} + +// 鏍规嵁鎽勫儚鏈篿d鎷垮埌鎽勫儚鏈烘墍鏈夊尯鍩� +func GetPolygons(cameraId string) []protomsg.CameraPolygon { + var cameraPolygons []protomsg.CameraPolygon + cameraPolygons = cache.GetPolygonsByCameraId(cameraId) + logger.Debug("------=======鏌ョ湅涓嬪叏閮ㄥ尯鍩燂細",cameraPolygons) + return cameraPolygons +} + +// 淇濈暀鍥涗綅灏忔暟 +func Retain(f float32) (float64) { + value1, _ := strconv.ParseFloat(fmt.Sprintf("%.2f", float64(f)), 64) + return value1 +} +func Decimal(value float32) float64 { + value1, _ := strconv.ParseFloat(fmt.Sprintf("%.2f", float64(value)), 64) + logger.Info("鍒濇淇濈暀涓や綅鎴恠tr:",value1) + n10 := math.Pow10(4) + value2 := math.Trunc((value1+0.5/n10)*n10) / n10 + logger.Info("鍒濇淇濈暀涓や綅鎴恠tr::::",value2) + return value2 +} +// 鍙栧嚭鏌愪釜鏃堕棿瑙勫垯鐨勭鍑犲ぉ鐨勮鍒欐闆嗗悎 +func GetTimeById(id string, index int) []TimeRange { + _, cameraTimeRule := cache.GetTimeRuleById(id) + var timeRangeList []day + err := json.Unmarshal([]byte(cameraTimeRule.TimeRule), &timeRangeList) + if err != nil { + logger.Error("鍙栨椂闂磋鍒欐椂鍙嶅簭鍒楀寲閿欒锛�") + } + for _, timerange := range timeRangeList { + if timerange.Day == index { + //logger.Println("鍙栧埌鐨勬椂闂磋鍒欙細", timerange.TimeRange) + return timerange.TimeRange + } + } + return nil +} + +// 鏍规嵁浼犲叆鐨勫瓧绗︿覆寰楀埌鍏跺湪涓�鍛ㄥ唴鐨勭储寮� 鍛ㄤ竴鍒板懆鏃ュ垎鍒搴�1鍒�7 +func getIndexOfWeek(weekday string) int { + var weekdays = [7]string{"Monday", "Tuesday", "Wednesday", "Thursday", "Friday", "Saturday", "Sunday"} + for k, value := range weekdays { + if value == weekday { + return k + 1 // 鍥犱负鏁版嵁搴撲腑瀛樼殑鏄�1-7浠h〃鐨勫懆涓�鍒板懆鏃� + } + } + return 0 +} + +type TimeRange struct { + Start string `json:"start"` + End string `json:"end"` +} +type day struct { + Day int `json:"day"` // 鏍囩ず褰撳墠鏄熸湡鍑� + TimeRange []TimeRange `json:"time_range"` // 褰撳ぉ鐨勫嚑涓椂闂存 +} + +// 灏嗗瓧绗︿覆鏍煎紡鐨勫潗鏍囧簭鍒楀寲涓篜oint鏍煎紡 +func Json2points(areaPoints string) []Point { + var pts []Point + if areaPoints == "[]" || areaPoints == "" { + logger.Error("=====================姝ゅ尯鍩熶负鍏ㄩ儴鍖哄煙") + pts = append(pts, Point{0, 0}) + pts = append(pts, Point{0, 540}) + pts = append(pts, Point{960, 540}) + pts = append(pts, Point{960, 0}) + } else { + err := json.Unmarshal([]byte(areaPoints), &pts) + if err != nil { + logger.Error("json.Unmarshal閿欒", err) + panic("搴忓垪鍖栧潗鏍囧紓甯革紝绋嬪簭閫�鍑�") + } + } + return pts +} + +// 缁欑洰鏍囧~鍏卨iker +func fillLiker(tableId []string, compareThreshold float32,arg *structure.Arg) { + //bytes := bigCache.GetComparePersonBaseInfo(tableId, arg.Feature, compareThreshold) + comArg := &protomsg.CompareArgs{ + TableIds:tableId, + FaceFeature:arg.Feature, + CompareThreshold:compareThreshold, + Source:false, + } + + bytes := getCompareMsg(comArg) + //bytes := []byte{} + var scResult protomsg.SdkCompareResult + err1 := proto.Unmarshal(bytes, &scResult) + if err1 != nil { + logger.Error("getBaseInfo瑙e帇閿欒", err1) + return + } + //logger.Info("----------------------------------------map鏄�", m) + ids := []string{} + m := make(map[string]*protomsg.SdkCompareEach) + for _,key := range scResult.CompareResult { + ids = append(ids, key.Id) + m[key.Id] = key + } + logger.Info("------------------------------------------------------------------------------------------------------------------------") + //logger.Info("=====================浜哄憳id鐨勯泦鍚堜负锛�",ids) + if len(ids) > 0 { + var dbApi dbapi.DbPersonApi + baseinfos, err1 := dbApi.Dbpersoninfosbyid(ids) + //baseinfos, err1 := esutil.Dbpersoninfosbyid(ids, dbTablePersons, serverIp, serverPort) + if err1 != nil { + logger.Error("鏌ヨ搴曞簱浜哄憳淇℃伅鍑洪敊", err1) + } + var dtapi dbapi.DbTableApi + for _, baseinfo := range baseinfos { + // 鏍规嵁tableId鏌ヨ搴曞簱淇℃伅缁檒iker璧嬪�� + //logger.Info("---------鐪嬬湅姣忎釜搴曞簱浜哄憳鐨勪俊鎭細",baseinfo.Id,baseinfo.PersonName) + tableIds := []string{} + tableIds = append(tableIds, baseinfo.TableId) // 铏界劧鏄紶鍏ユ暟缁勮繑鍥炴暟缁勭殑鎺ュ彛锛屼絾鎴戞寜鍗曚釜鐨勪娇鐢ㄤ簡 + table, err := dtapi.DbtablesById(tableIds) + //table, err := esutil.Dbtablefosbyid(tableIds, "dbtables", serverIp, serverPort) + if err != nil || len(table) == 0 { + logger.Error("鏍规嵁id鏌ヨ搴曞簱淇℃伅鍑洪敊锛�", err, "--杩斿洖鍊奸暱搴︿负锛�", len(table)) + } + logger.Debug("鐪嬬湅杩欎釜base鐨勫姣斿�兼槸澶氬皯锛�", Decimal(m[baseinfo.Id].CompareScore)) + base := structure.BaseInfo{TableId: baseinfo.TableId, TableName: table[0].TableName, BwType: table[0].BwType, CompareScore: Decimal(m[baseinfo.Id].CompareScore), PersonId: baseinfo.Id, PersonName: baseinfo.PersonName, PersonPicUrl: baseinfo.PersonPicUrl, PhoneNum: baseinfo.PhoneNum, Sex: baseinfo.Sex, IdCard: baseinfo.IdCard, MonitorLevel: baseinfo.MonitorLevel, Content: baseinfo.Reserved} + //os.Exit(1) + arg.Liker = append(arg.Liker, &base) + } + } + +} + +// 浜鸿劯姣斿 +func Compare(args *structure.SdkDatas, groupRule *protomsg.GroupRule) { + compareFlag := 0 + var tableIds []string + var threshold float32 = 50 // 榛樿闃堝�间负50 + // 鐪嬬湅鏄惁鏈夊彧閰嶄汉鑴告瘮瀵圭畻娉曚絾娌℃湁閰嶅姣斿簱鐨勮鍒欙紝濡傛灉鏈夛紝鍒欐瘮瀵瑰璞′负鍏ㄩ儴搴曞簱 + for j := 0; j < len(groupRule.Rules); j++ { + if groupRule.Rules[j].SdkId == "812b674b-2375-4589-919a-5c1c3278a972" { // 閰嶄簡浜鸿劯姣斿 + if groupRule.Rules[j].SdkArgAlias == "compareBase" && groupRule.Rules[j].SdkArgValue != "" { // 閰嶄簡姣斿搴曞簱鐨勫弬鏁颁絾涓嶆槸閰嶇殑鍏ㄩ儴搴曞簱 + compareFlag = 2 + tableIds = append(tableIds, groupRule.Rules[j].SdkArgValue) // 鏈�鍚庝細澶氫竴涓猼ableId锛屽垏鍑烘潵鐨勬暟缁勫彇len-1灏卞彲浠� + } + if groupRule.Rules[j].SdkArgAlias == "compareBase" && groupRule.Rules[j].SdkArgValue == "" { // 閰嶇殑鍙傛暟鏄瘮瀵瑰叏閮ㄥ簳搴� + compareFlag = 1 + } + if groupRule.Rules[j].SdkArgAlias == "cmpThreshold" { + v2, err := strconv.ParseFloat(groupRule.Rules[j].SdkArgValue, 32) + if err != nil { + logger.Error("string杞琭loat32澶辫触锛�") + } + threshold = float32(v2) + } + } + } + // 涓轰簡娴嬭瘯浣滀笅寮婏紝鐩存帴璁╁叾姣斿鍏ㄩ儴搴曞簱 + //compareFlag = 1 + if compareFlag == 0 { + logger.Info("娌℃湁閰嶇疆瀵规瘮搴曞簱鍙傛暟") + return + } + logger.Debug("=====閰嶇疆鐨勯槇鍊间负锛�", threshold) + for j := 0; j < len(groupRule.Rules); j++ { + if groupRule.Rules[j].SdkId == "812b674b-2375-4589-919a-5c1c3278a972" { + for _, sdkData := range args.Sdkdata { + if sdkData.IpcId == "A8B73405-373D-4F23-CED2-A617EBD7EC55" { // 鎼滅储鍒颁汉鑴告娴嬬殑sdkdata 鐜板湪鍏充簬浜鸿劯鐨勫彧鏈変粬涓�涓猻dk锛屽叏鎵句粬 + logger.Info("============================================杩涜浜鸿劯瀵规瘮") + for _, areaMap := range sdkData.AreaMapList { + // 鎷垮尯鍩熶腑姣忎釜浜鸿劯鐗瑰緛鍊煎幓瀵规瘮锛屽~鍏呭叾liker + if groupRule.Rules[j].PolygonId == areaMap.AreaId { + //logger.Info("--------------鐪嬬湅compareFlag鐨勫�煎拰tableId鍜宎reaMap.args鐨勯暱搴︼細",compareFlag,tableIds,len(areaMap.args)) + for _, arg := range areaMap.Args { + arg.Liker = arg.Liker[0:0] + //logger.Info("娓呯┖涔嬪悗鐪嬬湅涔嬪墠鎵撶殑浜鸿劯鏍囩鍙樹簡娌★細") + //if args.RuleResult["face"] != nil && len(args.RuleResult["face"].([]FaceResult)) > 0 { + // for _, faceResult := range args.RuleResult["face"].([]FaceResult) { + // for _,arg := range faceResult.Args { + // logger.Info("浜哄憳鍒嗗�兼槸锛�",arg.Score,"liker鐨勬暟閲忎负",arg.Liker) + // } + // } + //} + if compareFlag == 1 { + fillLiker(nil, threshold, arg) + } + if compareFlag == 2 { + fillLiker(tableIds, threshold, arg) + } + //logger.Info("-------------------鎴愬姛缁檒iker璧嬪��,闀垮害涓猴細", len(arg.Liker)) + } + areaMap.FilterData = areaMap.Args + //logger.Info("=======绗竴娆$湅args锛�",(areaMap.filterData)) + } + //logger.Info("-------------------------------浜鸿劯瀵规瘮涔嬪悗鐨勭洰鏍囨暟閲�",len(areaMap.args)) + } + } + } + return // 姣斿畬涓�娆″氨鍑哄幓锛岄槻姝㈠啀姣旂浜岄亶閫犳垚閲嶅锛屽洜涓虹‘瀹炲彲浠ユ湁涓や釜浜鸿劯姣斿鐨勫皬瑙勫垯锛屼竴涓簳搴擄紝涓�涓槇鍊� + } + } +} + +// 璁$畻鍖哄煙鍐呯殑鐩爣鏁伴噺浠ュ強灏嗙浉浼煎害銆佸崰姣斻�佸昂瀵哥瓑鎵撳寘 +func CountAreaObjs(a *structure.AreaMap,arg *structure.SdkData) { + + a.TargetNum = 0 + threshold := 80.0 // 鐩镐技搴� + intersectionper := 0.2 // 鍗犳瘮 + size := 0.0 // 灏哄 + + areaPoints := Json2points(a.AreaJson) + logger.Info("鐪嬬湅鍥剧墖鐨剋idth鍜宧eight:",arg.ImageWidth,arg.ImageHeight) + widthScale := float64(arg.ImageWidth / 960) + heigthScale := float64(arg.ImageHeight / 540) + for _, obj := range arg.Photo { + //logger.Info("------------------鐪嬬湅sdkData:", arg.SdkName, "鐨凱hoto鏁版嵁----------------", obj, "----椤轰究鐪嬬湅鍗犳瘮-----锛�", PgsInterPercent(areaPoints, obj.Rects, widthScale, heigthScale)) + if threshold <= obj.Score && size <= float64(obj.Rects.Width*obj.Rects.Height) && intersectionper <= PgsInterPercent(areaPoints, obj.Rects, widthScale, heigthScale) { + // 杩欐瑕佸榻愯〃杈惧紡閲屾墍闇�瑕佺殑鎵�鏈夊弬鏁� + a.TargetNum++ + arg1 := structure.Arg{obj.Id,obj.Score, PgsInterPercent(areaPoints, obj.Rects, widthScale, heigthScale), float64(obj.Rects.Width * obj.Rects.Height), obj.IsYolo, obj.Rects, obj.Feature, obj.ThftRes, []*structure.BaseInfo{}} + //logger.Println("鏀捐繘鍘荤殑arg锛�-------", arg1) + a.Args = append(a.Args, &arg1) + a.FilterData = append(a.FilterData, &arg1) + } + } + logger.Info("鍖哄煙鏄細",areaPoints,"鍖哄煙鍐呯洰鏍囨暟閲忎负锛�",a.TargetNum,"---",len(a.FilterData)) + for _,tar := range a.FilterData { + logger.Info("鍏蜂綋鐩爣鐨勫潗鏍囦负",tar.Id,tar.Location) + } + a.Time = time.Unix(time.Now().Unix(), 0).String()[11:16] + a.KeepRight = arg.KeepRight + a.IsStatic = arg.IsStatic + //logger.Println("--------------------鐪嬬湅鍖哄煙鏁版嵁锛�",*a) +} + +// 鎶妔dk浠庢暟鎹抚涓婃彁鍙栫殑鎸夌収鍖哄煙鍒嗙被褰掔疆 +func SdkDataFormat(cameraId string, arg *structure.SdkData, cameraPolygons []protomsg.CameraPolygon) { + logger.Info("==================================鏈瑂dkData涓В鍑烘潵鐨勭洰鏍囨暟鎹�=======================================") + for _, photo := range arg.Photo { + logger.Info("--------瑙f瀽鍑烘潵鐨勬暟鎹�---", cameraId, arg.IpcId, photo.Rects, photo.Score) + } + for _, polygon := range cameraPolygons { + //logger.Println("++++++鍦ㄨ繖鍎跨湅涓�涓嬪尯鍩熷晩:", polygon.Polygon) + areaMap := structure.AreaMap{CameraId: cameraId, AreaId: polygon.Id, AreaJson: polygon.Polygon, TriggerLine: polygon.TriggerLine, DirectionLine: polygon.DirectionLine} + // 涓烘瘡涓憚鍍忔満鍖哄煙濉厖鏁版嵁 + CountAreaObjs(&areaMap,arg) + arg.AreaMapList = append(arg.AreaMapList, &areaMap) + } +} + +func Push(data *protomsg.CompareArgs,sock mangos.Socket) []byte{ + //var sock mangos.Socket + var err error + var msg []byte + + //if sock, err = req.NewSocket(); err != nil { + // logger.Error("鍒涘缓璇锋眰socket澶辫触: %s", err.Error()) + //} + errSize := sock.SetOption(mangos.OptionMaxRecvSize,5*1024*1024) + if errSize != nil { + logger.Error("Failed set MaxRecvSize: %v", err) + return nil + } + serverIP, _ := GetLocalIP() + if err = sock.Dial("tcp://"+serverIP+":4010"); err != nil { + logger.Error("璇锋眰socket鎷ㄥ彿澶辫触: %s", err.Error()) + } + //sock.AddTransport(ipc.NewTransport()) + sock.AddTransport(tcp.NewTransport()) + + logger.Info("搴忓垪鍖栨暟鎹�") + bytes,err1 := proto.Marshal(data) + logger.Info("鏁版嵁闀垮害涓猴細",len(bytes)) + if err1 != nil { + logger.Info("搴忓垪鍖栧け璐ワ細",err1) + } + logger.Debug("鎺ㄩ�佹暟鎹�") + //bytes := []byte("ndfasojdfaidsos") + if err = sock.Send(bytes); err != nil { + logger.Error("鎺ㄩ�乻ocket鍙戦�佹暟鎹け璐�: %s", err.Error()) + //os.Exit(1) + } + if msg, err = sock.Recv(); err != nil { + logger.Error("鎺ユ敹鍝嶅簲澶辫触: %s", err.Error()) + //os.Exit(1) + } + logger.Debug("鏁版嵁鎺ㄩ�佹垚鍔燂紒鏀跺埌鍝嶅簲,鏁版嵁闀垮害涓猴細",len(msg)) + return msg +} + + +func Push1(){ + //var sock mangos.Socket + var err error + var msg []byte + + if sock, err = req.NewSocket(); err != nil { + logger.Error("鍒涘缓璇锋眰socket澶辫触: %s", err.Error()) + } + errSize := sock.SetOption(mangos.OptionMaxRecvSize,5*1024*1024) + if errSize != nil { + logger.Error("Failed set MaxRecvSize: %v", err) + } + //sock.AddTransport(ipc.NewTransport()) + sock.AddTransport(tcp.NewTransport()) + serverIP, _ := GetLocalIP() + if err = sock.Dial("tcp://"+serverIP+":4010"); err != nil { + logger.Error("璇锋眰socket鎷ㄥ彿澶辫触: %s", err.Error()) + } + logger.Info("搴忓垪鍖栨暟鎹�") + + for { + select { + // case <-ctx.Done(): + // return + case data := <- sender: + logger.Info("姣斿杩涚▼鍏ュ弬锛�",data.TableIds,data.CompareThreshold) + bytes,err1 := proto.Marshal(data) + logger.Info("鏁版嵁闀垮害涓猴細",len(bytes)) + if err1 != nil { + logger.Info("搴忓垪鍖栧け璐ワ細",err1) + } + logger.Debug("鎺ㄩ�佹暟鎹�") + //bytes := []byte("ndfasojdfaidsos") + if err = sock.Send(bytes); err != nil { + logger.Error("鎺ㄩ�乻ocket鍙戦�佹暟鎹け璐�: %s", err.Error()) + //os.Exit(1) + } + if msg, err = sock.Recv(); err != nil { + logger.Error("鎺ユ敹鍝嶅簲澶辫触: %s", err.Error()) + //os.Exit(1) + } + logger.Debug("鏁版嵁鎺ㄩ�佹垚鍔燂紒鏀跺埌鍝嶅簲,鏁版嵁闀垮害涓猴細",len(msg)) + receiver <- msg + default: + + } + } +} +func getCompareMsg(data *protomsg.CompareArgs) []byte{ + sender <- data + return <-receiver +} +// 鑾峰彇鏈満ip +func GetLocalIP() (ipv4 string, err error) { + var ( + addrs []net.Addr + addr net.Addr + ipNet *net.IPNet // IP鍦板潃 + isIpNet bool + ) + // 鑾峰彇鎵�鏈夌綉鍗� + if addrs, err = net.InterfaceAddrs(); err != nil { + return + } + // 鍙栫涓�涓潪lo鐨勭綉鍗P + for _, addr = range addrs { + // 杩欎釜缃戠粶鍦板潃鏄疘P鍦板潃: ipv4, ipv6 + if ipNet, isIpNet = addr.(*net.IPNet); isIpNet && !ipNet.IP.IsLoopback() { + // 璺宠繃IPV6 + if ipNet.IP.To4() != nil { + ipv4 = ipNet.IP.String() // 192.168.1.1 + return + } + } + } + + err = errors.New("ipv4 not found") + return +} -- Gitblit v1.8.0