---

panlei

2019-11-02 c16cab7461123245b06de496a92462a7f9e1f1d0

 labelFilter/req.go

@@ -1,18 +1,25 @@
package labelFilter

import (
   "basic.com/dbapi.git"
   "basic.com/valib/logger.git"
   "fmt"
   "github.com/golang/protobuf/proto"
   "nanomsg.org/go-mangos"
   "nanomsg.org/go-mangos/protocol/rep"
   "nanomsg.org/go-mangos/protocol/req"
   "nanomsg.org/go-mangos/transport/ipc"
   "nanomsg.org/go-mangos/transport/tcp"
   "ruleprocess/structure"
   "strings"
   "time"
)

var urlPool = make(map[string]chan []byte,100)
//var urlChans = make([]urlChan,100)
//type urlChan struct {
//   url string
//   ch chan *structure.ResultMsg
//}
//var pool chan *structure.ResultMsg = make(chan *structure.ResultMsg)
func Die(format string, v ...interface{}) {
   logger.Info("+++++++",format)
   //os.Exit(1)
@@ -52,48 +59,101 @@
   }
}

func Push(url string,data structure.ResultMsg) {
   var sock mangos.Socket
func Init(){

   var api dbapi.EventPushApi
   b, allRules := api.FindAllDetails()
   logger.Info("初始化事件推送，查看所有规则组：", allRules)
   if !b {
      logger.Error("查询时间推送规则失败！")
   }
   for _, ruleGroup := range allRules {
      if ruleGroup.Enable { // 大规则开关开启状态
         for _, url := range ruleGroup.Urls {
            // 为每个url建立一个chan

            if strings.Contains(url.Url,"102") {
               urlPool[url.Url] = make(chan []byte,10)
               logger.Info("初始化信息：",urlPool)
               go GoPush(url.Url)
            }
         }
      }
   }
}

func GoPush(url string) {
   var err error
   var msg []byte

   var sock mangos.Socket
   if sock, err = req.NewSocket(); err != nil {
      Die("创建请求socket失败: %s", err.Error())
   }
   errSize := sock.SetOption(mangos.OptionMaxRecvSize,30*1024*1024)
   if errSize != nil {
      fmt.Errorf("Failed set MaxRecvSize: %v", err)
      logger.Error("传输的数据超过大小限制")
      return
   }
   errTimeOut := sock.SetOption(mangos.OptionRecvDeadline,time.Millisecond * 2000)
   if errTimeOut != nil {
      fmt.Errorf("Failed set MaxRecvDeadline: %v", err)
      logger.Error("接收响应超时")
      return
   }
   //sock.AddTransport(ipc.NewTransport())
   //errWrite := sock.SetOption(mangos.OptionWriteQLen,10)
   //if errWrite != nil {
   //   logger.Error("设置传输缓存大小失败")
   //   return
   //}
   //errRead := sock.SetOption(mangos.OptionReadQLen,10)
   //if errRead != nil {
   //   logger.Error("设置传输缓存大小失败")
   //   return
   //}
   sock.AddTransport(tcp.NewTransport())
   if err = sock.Dial(url); err != nil {
      Die("请求socket拨号失败: %s", err.Error())
   if err = sock.Dial("tcp://"+url); err != nil {
      logger.Error("请求socket拨号失败: ", err.Error())
   }
   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 {
      Die("推送socket发送数据失败: %s", err.Error())
   }
   if msg, err = sock.Recv(); err != nil {
      Die("接收响应失败: %s", err.Error())
   } else if msg, err = sock.Recv(); err == nil{
      logger.Debug("事件推送成功！收到响应",string(msg))

   //for v := range pool{
   //   logger.Info("无限循环",v.Cid)
   //}
   //var ch chan *structure.ResultMsg
   //for _, v := range urlChans  {
   //   if v.url == url{
   //      ch = v.ch
   //   }
   //}
   logger.Info("chan信息：",urlPool[url])
   for {
      select {
      // case <-ctx.Done():
      //    return
      case data := <- urlPool[url]:
         //logger.Info("接收到数据",data.Cid)
         //bytes,err1 := proto.Marshal(data)
         logger.Info("数据长度为：",len(data))
         //if err1 != nil {
         //   logger.Info("序列化失败：",err1)
         //}
         logger.Debug("groutine"+url+"推送数据")
         //bytes := []byte("ndfasojdfaidsos")
         if err = sock.Send(data); err != nil {
            Die("groutine"+url+"推送socket发送数据失败: ", err.Error())
         }
         msg, err = sock.Recv();
         if err != nil {
            Die("groutine"+url+"接收响应失败: ", err.Error())
         } else {
            logger.Debug("事件推送成功！groutine"+url+"收到响应",string(msg))
         }

      default:

      }
   }
   sock.Close()
}

//func main() {
//   url := "tcp://192.168.1.123:40011"
//   Push(url,"hahahaha")