recvRoutine添加日志

liuxiaolong

2021-01-06 c4e51a45643377624f80829066d99008c6623458

recvRoutine添加日志

 hbusc.go

@@ -55,12 +55,13 @@
}

//获取其他进程发给此socket的消息,可以是sub的接收，也可以是reply的接收。
func recvRoutine(ctx context.Context, s *bhomebus.Socket, wg *sync.WaitGroup, ch chan<-TransInfo) {
func recvRoutine(ctx context.Context, s *bhomebus.Socket, wg *sync.WaitGroup, ch chan<-TransInfo, logFn func(...interface{})) {
    var data []byte
    var key int
    for {
        select {
        case <-ctx.Done():
            logFn("recvRoutine ctx.Done")
            wg.Done()
            return
        default:
@@ -75,6 +76,8 @@
                    data = []byte{}
                    key = 0
                }
            } else {
                time.Sleep(10 * time.Millisecond)
            }
        }
    }
@@ -195,7 +198,7 @@
    sockReply.ForceBind(int(regR.ReplyKey))
    handle.wg.Add(1)
    //serve server reply
    go recvRoutine(ctx, sockReply, handle.wg, handle.chReply)
    go recvRoutine(ctx, sockReply, handle.wg, handle.chReply, handle.printLog)
    handle.sockRep = &sockServer{
        sock: sockReply,
        info: &ri.Proc,
@@ -224,7 +227,7 @@

    //启动订阅信息接收
    handle.wg.Add(1)
    go recvRoutine(ctx, sockSub, handle.wg, handle.chSub)
    go recvRoutine(ctx, sockSub, handle.wg, handle.chSub, handle.printLog)
    handle.sockSub = &sockClient{
        sock: sockSub,
        peer: -1,