From 993dc81c942d6b1315684cda034c8859a6f4d02b Mon Sep 17 00:00:00 2001 From: panlei <2799247126@qq.com> Date: 星期三, 06 十一月 2019 11:15:40 +0800 Subject: [PATCH] --- --- labelFilter/req.go | 80 ++++++++++++++++++++++++---------------- 1 files changed, 48 insertions(+), 32 deletions(-) diff --git a/labelFilter/req.go b/labelFilter/req.go index 77b7e7c..2c08412 100644 --- a/labelFilter/req.go +++ b/labelFilter/req.go @@ -4,22 +4,23 @@ "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) + logger.Info("+++++++",format,v) //os.Exit(1) } @@ -69,8 +70,9 @@ 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 urlPool[url.Url] == nil { + urlPool[url.Url] = make(chan []byte,100) + logger.Info("鍒濆鍖栦俊鎭細",urlPool) go GoPush(url.Url) } } @@ -90,17 +92,22 @@ logger.Error("浼犺緭鐨勬暟鎹秴杩囧ぇ灏忛檺鍒�") return } - errTimeOut := sock.SetOption(mangos.OptionRecvDeadline,time.Millisecond * 2000) + errTimeOut := sock.SetOption(mangos.OptionRecvDeadline,time.Millisecond * 1500) if errTimeOut != nil { logger.Error("鎺ユ敹鍝嶅簲瓒呮椂") return } - errWrite := sock.SetOption(mangos.OptionWriteQLen,10) + errTimeOut1 := sock.SetOption(mangos.OptionSendDeadline,time.Millisecond * 1500) + if errTimeOut1 != nil { + logger.Error("鍙戦�佽秴鏃�") + return + } + errWrite := sock.SetOption(mangos.OptionWriteQLen,5) if errWrite != nil { logger.Error("璁剧疆浼犺緭缂撳瓨澶у皬澶辫触") return } - errRead := sock.SetOption(mangos.OptionReadQLen,10) + errRead := sock.SetOption(mangos.OptionReadQLen,5) if errRead != nil { logger.Error("璁剧疆浼犺緭缂撳瓨澶у皬澶辫触") return @@ -114,30 +121,39 @@ //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) - } - 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)) - } - + case data := <- urlPool[url]: + //logger.Info("鎺ユ敹鍒版暟鎹�",data.Cid) + //bytes,err1 := proto.Marshal(data) + go func(data []byte) { + logger.Info("鏁版嵁闀垮害涓猴細",len(data)) + //if err1 != nil { + // logger.Info("搴忓垪鍖栧け璐ワ細",err1) + //} + start := time.Now() + logger.Debug("groutine"+url+"鎺ㄩ�佹暟鎹�") + //bytes := []byte("ndfasojdfaidsos") + if err = sock.Send(data); err != nil { + Die("groutine"+url+"鎺ㄩ�乻ocket鍙戦�佹暟鎹け璐�: ", err.Error()) + } + pushTime := time.Since(start) + msg, err = sock.Recv(); + if err != nil { + Die("groutine"+url+"鎺ユ敹鍝嶅簲澶辫触: ", err.Error(),pushTime,time.Since(start)) + } else { + logger.Debug("浜嬩欢鎺ㄩ�佹垚鍔燂紒groutine"+url+"鏀跺埌鍝嶅簲",string(msg),pushTime,time.Since(start)) + } + }(data) default: } -- Gitblit v1.8.0