From 63ddbfa884c336f0aa4a73bb134d3526cbf18802 Mon Sep 17 00:00:00 2001 From: panlei <2799247126@qq.com> Date: 星期六, 24 八月 2019 16:13:31 +0800 Subject: [PATCH] update protomsg --- ruleserver/readyDataForRule.go | 50 ++++++++++++++++++++++++++++++++++++++++++-------- 1 files changed, 42 insertions(+), 8 deletions(-) diff --git a/ruleserver/readyDataForRule.go b/ruleserver/readyDataForRule.go index 58937d5..4c47372 100644 --- a/ruleserver/readyDataForRule.go +++ b/ruleserver/readyDataForRule.go @@ -1,21 +1,22 @@ package ruleserver import ( + "basic.com/dbapi.git" bigCache "basic.com/pubsub/cache.git" - "basic.com/pubsub/cache.git/esutil" "basic.com/pubsub/protomsg.git" "encoding/json" + "errors" "fmt" "github.com/go-yaml/yaml" + "github.com/golang/protobuf/proto" "io/ioutil" "math" "nanomsg.org/go-mangos" "nanomsg.org/go-mangos/protocol/req" "nanomsg.org/go-mangos/transport/tcp" - "os" + "net" "ruleprocess/cache" "ruleprocess/logger" - "github.com/golang/protobuf/proto" "strconv" "time" ) @@ -174,7 +175,8 @@ CompareThreshold:compareThreshold, Source:false, } - bytes := Push("tcp://192.168.1.182:40010",comArg,sock) + serverIp, _ := GetLocalIP() + bytes := Push("tcp://"+serverIp+":4010",comArg,sock) var scResult protomsg.SdkCompareResult err1 := proto.Unmarshal(bytes, &scResult) if err1 != nil { @@ -190,16 +192,20 @@ logger.Info("------------------------------------------------------------------------------------------------------------------------") //logger.Info("=====================浜哄憳id鐨勯泦鍚堜负锛�",ids) if len(ids) > 0 { - baseinfos, err1 := esutil.Dbpersoninfosbyid(ids, dbTablePersons, serverIp, serverPort) + 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 := esutil.Dbtablefosbyid(tableIds, "dbtables", serverIp, serverPort) + 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)) } @@ -354,12 +360,40 @@ //bytes := []byte("ndfasojdfaidsos") if err = sock.Send(bytes); err != nil { logger.Error("鎺ㄩ�乻ocket鍙戦�佹暟鎹け璐�: %s", err.Error()) - os.Exit(1) + //os.Exit(1) } if msg, err = sock.Recv(); err != nil { logger.Error("鎺ユ敹鍝嶅簲澶辫触: %s", err.Error()) - os.Exit(1) + //os.Exit(1) } logger.Debug("鏁版嵁鎺ㄩ�佹垚鍔燂紒鏀跺埌鍝嶅簲,鏁版嵁闀垮害涓猴細",len(msg)) return msg } + +// 鑾峰彇鏈満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