From c16cab7461123245b06de496a92462a7f9e1f1d0 Mon Sep 17 00:00:00 2001 From: panlei <2799247126@qq.com> Date: 星期六, 02 十一月 2019 17:22:14 +0800 Subject: [PATCH] --- --- labelFilter/req.go | 62 ++++++++++++++++++------------- 1 files changed, 36 insertions(+), 26 deletions(-) diff --git a/labelFilter/req.go b/labelFilter/req.go index 77b7e7c..ae18f79 100644 --- a/labelFilter/req.go +++ b/labelFilter/req.go @@ -4,20 +4,22 @@ "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 structure.ResultMsg) -var pool chan *structure.ResultMsg = make(chan *structure.ResultMsg) +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) @@ -69,8 +71,10 @@ if ruleGroup.Enable { // 澶ц鍒欏紑鍏冲紑鍚姸鎬� for _, url := range ruleGroup.Urls { // 涓烘瘡涓猽rl寤虹珛涓�涓猚han - if strings.Contains(url.Url,"114") { - //urlPool[url.Url] = make(chan structure.ResultMsg,10) + + if strings.Contains(url.Url,"102") { + urlPool[url.Url] = make(chan []byte,10) + logger.Info("鍒濆鍖栦俊鎭細",urlPool) go GoPush(url.Url) } } @@ -95,16 +99,16 @@ 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 - } + //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("tcp://"+url); err != nil { logger.Error("璇锋眰socket鎷ㄥ彿澶辫触: ", err.Error()) @@ -114,21 +118,27 @@ //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 := <- pool: - logger.Info("鎺ユ敹鍒版暟鎹�",data.Cid) - bytes,err1 := proto.Marshal(data) - logger.Info("鏁版嵁闀垮害涓猴細",len(bytes)) - if err1 != nil { - logger.Info("搴忓垪鍖栧け璐ワ細",err1) - } + 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(bytes); err != nil { + if err = sock.Send(data); err != nil { Die("groutine"+url+"鎺ㄩ�乻ocket鍙戦�佹暟鎹け璐�: ", err.Error()) } msg, err = sock.Recv(); -- Gitblit v1.8.0