From e6982607fbbeaa96d3d14409df780266646b793d Mon Sep 17 00:00:00 2001
From: panlei <2799247126@qq.com>
Date: 星期五, 01 十一月 2019 18:22:44 +0800
Subject: [PATCH] 换一个chan
---
labelFilter/req.go | 122 +++++++++++++++++++++++++++++++---------
1 files changed, 95 insertions(+), 27 deletions(-)
diff --git a/labelFilter/req.go b/labelFilter/req.go
index e69d233..77b7e7c 100644
--- a/labelFilter/req.go
+++ b/labelFilter/req.go
@@ -1,33 +1,37 @@
package labelFilter
import (
+ "basic.com/dbapi.git"
+ "basic.com/valib/logger.git"
"fmt"
- "github.com/gogo/protobuf/proto"
+ "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"
- "os"
- "ruleprocess/logger"
- "ruleprocess/ruleserver"
+ "ruleprocess/structure"
+ "strings"
"time"
)
+
+var urlPool = make(map[string]chan structure.ResultMsg)
+var pool chan *structure.ResultMsg = make(chan *structure.ResultMsg)
func Die(format string, v ...interface{}) {
- fmt.Fprintln(os.Stderr, fmt.Sprintf(format, v...))
logger.Info("+++++++",format)
- os.Exit(1)
+ //os.Exit(1)
}
func date() string {
return time.Now().Format(time.ANSIC)
}
-
-func Node0(url string) {
+var msgChan chan []byte
+func Receive(url string) {
var sock mangos.Socket
var err error
var msg []byte
+ msgChan = make(chan []byte,200)
if sock, err = rep.NewSocket(); err != nil {
Die("can't get new rep socket: %s", err)
}
@@ -39,11 +43,13 @@
for {
// Could also use sock.RecvMsg to get header
msg, err = sock.Recv()
- if string(msg) == "DATE" { // no need to terminate
- fmt.Println("NODE0: RECEIVED DATE REQUEST")
+ if msg != nil { // no need to terminate
+ fmt.Println("Received Data request")
+ // 鎶婃敹鍒扮殑msg濉炶繘閫氶亾
+ msgChan <- msg
+ // 缁欏彂閫佺▼搴忓弽棣堜俊鎭�
d := date()
- fmt.Printf("NODE0: SENDING DATE %s\n", d)
- err = sock.Send([]byte(d))
+ err = sock.Send([]byte("Received Data, --"+d))
if err != nil {
Die("can't send reply: %s", err.Error())
}
@@ -51,31 +57,93 @@
}
}
-func Push(url string,data ruleserver.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 {
+ // 涓烘瘡涓猽rl寤虹珛涓�涓猚han
+ if strings.Contains(url.Url,"114") {
+ //urlPool[url.Url] = make(chan structure.ResultMsg,10)
+ 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())
}
- //sock.AddTransport(ipc.NewTransport())
+ errSize := sock.SetOption(mangos.OptionMaxRecvSize,30*1024*1024)
+ if errSize != nil {
+ logger.Error("浼犺緭鐨勬暟鎹秴杩囧ぇ灏忛檺鍒�")
+ return
+ }
+ errTimeOut := sock.SetOption(mangos.OptionRecvDeadline,time.Millisecond * 2000)
+ if errTimeOut != nil {
+ logger.Error("鎺ユ敹鍝嶅簲瓒呮椂")
+ return
+ }
+ 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.Debug("鎺ㄩ�佹暟鎹�")
- bytes,err := proto.Marshal(data)
- if err = sock.Send(bytes); err != nil {
- Die("鎺ㄩ�乻ocket鍙戦�佹暟鎹け璐�: %s", err.Error())
+ logger.Info("搴忓垪鍖栨暟鎹�")
+
+ //for v := range pool{
+ // logger.Info("鏃犻檺寰幆",v.Cid)
+ //}
+
+ for {
+ select {
+ // case <-ctx.Done():
+ // return
+ case data := <- pool:
+ logger.Info("鎺ユ敹鍒版暟鎹�",data.Cid)
+ bytes,err1 := proto.Marshal(data)
+ logger.Info("鏁版嵁闀垮害涓猴細",len(bytes))
+ if err1 != nil {
+ logger.Info("搴忓垪鍖栧け璐ワ細",err1)
+ }
+ logger.Debug("groutine"+url+"鎺ㄩ�佹暟鎹�")
+ //bytes := []byte("ndfasojdfaidsos")
+ if err = sock.Send(bytes); err != nil {
+ Die("groutine"+url+"鎺ㄩ�乻ocket鍙戦�佹暟鎹け璐�: ", err.Error())
+ }
+ msg, err = sock.Recv();
+ if err != nil {
+ Die("groutine"+url+"鎺ユ敹鍝嶅簲澶辫触: ", err.Error())
+ } else {
+ logger.Debug("浜嬩欢鎺ㄩ�佹垚鍔燂紒groutine"+url+"鏀跺埌鍝嶅簲",string(msg))
+ }
+
+ default:
+
+ }
}
- if msg, err = sock.Recv(); err != nil {
- Die("鎺ユ敹鍝嶅簲澶辫触: %s", err.Error())
- }
- logger.Debug("鏁版嵁鎺ㄩ�佹垚鍔燂紒鏀跺埌鍝嶅簲",string(msg))
sock.Close()
}
-
//func main() {
// url := "tcp://192.168.1.123:40011"
// Push(url,"hahahaha")
--
Gitblit v1.8.0