From f29ed4a8aece3175514c22c40dda79af6e093af4 Mon Sep 17 00:00:00 2001
From: panlei <2799247126@qq.com>
Date: 星期三, 30 十月 2019 19:28:12 +0800
Subject: [PATCH] 修正事件推送
---
labelFilter/req.go | 83 ++++++++++++++++++++++++++++-------------
main.go | 1
labelFilter/ruleForLabel.go | 2
3 files changed, 59 insertions(+), 27 deletions(-)
diff --git a/labelFilter/req.go b/labelFilter/req.go
index 7fb7bc2..d85261b 100644
--- a/labelFilter/req.go
+++ b/labelFilter/req.go
@@ -1,6 +1,7 @@
package labelFilter
import (
+ "basic.com/dbapi.git"
"basic.com/valib/logger.git"
"fmt"
"github.com/golang/protobuf/proto"
@@ -13,6 +14,8 @@
"time"
)
+var sock mangos.Socket
+var urlPool = make(map[string]chan structure.ResultMsg)
func Die(format string, v ...interface{}) {
logger.Info("+++++++",format)
//os.Exit(1)
@@ -52,49 +55,77 @@
}
}
-func Push(url string,data structure.ResultMsg) {
- var sock mangos.Socket
+func Init(){
+
var err error
- var msg []byte
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())
- sock.AddTransport(tcp.NewTransport())
- if err = sock.Dial(url); err != nil {
- Die("璇锋眰socket鎷ㄥ彿澶辫触: %s", err.Error())
+ var api dbapi.EventPushApi
+ b, allRules := api.FindAllDetails()
+ logger.Info("鏌ョ湅鎵�鏈夎鍒欑粍锛�", allRules)
+ if !b {
+ logger.Error("鏌ヨ鏃堕棿鎺ㄩ�佽鍒欏け璐ワ紒")
}
- logger.Info("搴忓垪鍖栨暟鎹�")
- bytes,err1 := proto.Marshal(data)
- logger.Info("鏁版嵁闀垮害涓猴細",len(bytes))
- if err1 != nil {
- logger.Info("搴忓垪鍖栧け璐ワ細",err1)
+ for _, ruleGroup := range allRules {
+ if ruleGroup.Enable { // 澶ц鍒欏紑鍏冲紑鍚姸鎬�
+ for _, url := range ruleGroup.Urls {
+ // 涓烘瘡涓猽rl寤虹珛涓�涓猚han
+ urlPool[url.Url] = make(chan structure.ResultMsg,50)
+ go GoPush(url.Url)
+ }
+ }
}
- logger.Debug("鎺ㄩ�佹暟鎹�")
- //bytes := []byte("ndfasojdfaidsos")
- if err = sock.Send(bytes); err != nil {
- Die("鎺ㄩ�乻ocket鍙戦�佹暟鎹け璐�: %s", err.Error())
- }
- msg, err = sock.Recv();
- if err != nil {
- Die("鎺ユ敹鍝嶅簲澶辫触: %s", err.Error())
- } else {
- logger.Debug("浜嬩欢鎺ㄩ�佹垚鍔燂紒鏀跺埌鍝嶅簲",string(msg))
- }
- sock.Close()
}
+func GoPush(url string) {
+ var err error
+ var msg []byte
+ sock.AddTransport(tcp.NewTransport())
+ if err = sock.Dial(url); err != nil {
+ logger.Error("璇锋眰socket鎷ㄥ彿澶辫触: %s", err.Error())
+ }
+ logger.Info("搴忓垪鍖栨暟鎹�")
+
+ for {
+ select {
+ // case <-ctx.Done():
+ // return
+ case data := <- urlPool[url]:
+
+ 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("鎺ㄩ�乻ocket鍙戦�佹暟鎹け璐�: %s", err.Error())
+ }
+ msg, err = sock.Recv();
+ if err != nil {
+ Die("鎺ユ敹鍝嶅簲澶辫触: %s", err.Error())
+ } else {
+ logger.Debug("浜嬩欢鎺ㄩ�佹垚鍔燂紒鏀跺埌鍝嶅簲",string(msg))
+ }
+ sock.Close()
+ default:
+
+ }
+ }
+}
//func main() {
// url := "tcp://192.168.1.123:40011"
// Push(url,"hahahaha")
diff --git a/labelFilter/ruleForLabel.go b/labelFilter/ruleForLabel.go
index bb9a92e..587cb39 100644
--- a/labelFilter/ruleForLabel.go
+++ b/labelFilter/ruleForLabel.go
@@ -319,7 +319,7 @@
func pushData(urls []*protomsg.PushUrl, data structure.ResultMsg) {
for _, url := range urls {
logger.Debug("鐪嬬湅鎺ㄩ�佸湴鍧�锛�",url.Url)
- //Push("tcp://"+url.Url, data)
+
}
}
diff --git a/main.go b/main.go
index 5fc3011..989f128 100644
--- a/main.go
+++ b/main.go
@@ -63,6 +63,7 @@
go cache.Init(initchan, *dbIp, *surveyPort, *pubPort)
logger.Info("cache init completed!!!", <-initchan) //dbserver鍒濆鍖栧畬姣�
ruleserver.Init()
+ labelFilter.Init()
go ruleserver.TimeTicker()
go ruleserver.StartServer()
nReciever("ipc:///tmp/sdk-2-rules-process.ipc", deliver.PushPull, 1)
--
Gitblit v1.8.0