panlei
2019-08-24 63ddbfa884c336f0aa4a73bb134d3526cbf18802
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查询底库信息给liker赋值
         //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("推送socket发送数据失败: %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的网卡IP
   for _, addr = range addrs {
      // 这个网络地址是IP地址: 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
}