From fc2431baefd39eab76f9b7872031be85154ff07f Mon Sep 17 00:00:00 2001
From: panlei <2799247126@qq.com>
Date: 星期五, 16 八月 2019 14:57:29 +0800
Subject: [PATCH] 比对底库的ip不再写死,用本地的ip

---
 ruleserver/readyDataForRule.go |   33 ++++++++++++++++++++++++++++++++-
 1 files changed, 32 insertions(+), 1 deletions(-)

diff --git a/ruleserver/readyDataForRule.go b/ruleserver/readyDataForRule.go
index 58937d5..3d4b0e3 100644
--- a/ruleserver/readyDataForRule.go
+++ b/ruleserver/readyDataForRule.go
@@ -5,6 +5,7 @@
 	"basic.com/pubsub/cache.git/esutil"
 	"basic.com/pubsub/protomsg.git"
 	"encoding/json"
+	"errors"
 	"fmt"
 	"github.com/go-yaml/yaml"
 	"io/ioutil"
@@ -12,6 +13,7 @@
 	"nanomsg.org/go-mangos"
 	"nanomsg.org/go-mangos/protocol/req"
 	"nanomsg.org/go-mangos/transport/tcp"
+	"net"
 	"os"
 	"ruleprocess/cache"
 	"ruleprocess/logger"
@@ -174,7 +176,8 @@
 		CompareThreshold:compareThreshold,
 		Source:false,
 	}
-	bytes := Push("tcp://192.168.1.182:40010",comArg,sock)
+	serverIp, _ := GetLocalIP()
+	bytes := Push("tcp://"+serverIp+":40010",comArg,sock)
 	var scResult protomsg.SdkCompareResult
 	err1 := proto.Unmarshal(bytes, &scResult)
 	if err1 != nil {
@@ -363,3 +366,31 @@
 	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