From 81afb6ffbf7f76f49644a1832dcfe241552d7e08 Mon Sep 17 00:00:00 2001 From: liuxiaolong <liuxiaolong@aiotlink.com> Date: 星期五, 05 二月 2021 18:20:09 +0800 Subject: [PATCH] add recvandsend --- micronode.go | 75 ++++++++++++++++++++++--------------- 1 files changed, 44 insertions(+), 31 deletions(-) diff --git a/micronode.go b/micronode.go index df721e8..9fa49a2 100644 --- a/micronode.go +++ b/micronode.go @@ -90,21 +90,24 @@ ms.handlers = funcMap go ms.startHeartbeat() + //鎺ユ敹璁㈤槄鍒扮殑娑堟伅 + go ms.startRecvSubMsg() + //浣滀负server鍚姩 + ms.serve() +} +//寮�濮嬫帴鏀惰闃呮秷鎭� +func (ms *MicroNode) startRecvSubMsg() { for { select { case <- ms.ctx.Done(): return default: - msgS, msgR, keyR := ms.handle.GetMsg() + msgS := ms.handle.GetMsg() if msgS != nil { //鏀跺埌鍏跺畠杩涚▼鐨勫彂甯冩秷鎭� ms.printLog("Recv Sub Message:", string(msgS.Body)) ms.SubCh <- msgS - } - if msgR != nil { - //鏀跺埌鍏跺畠杩涚▼鐨勮姹傛秷鎭� - go ms.serve(msgR, keyR) } time.Sleep(50 * time.Millisecond) @@ -189,39 +192,49 @@ return nil, fmt.Errorf("GetRegisteredClient list failed") } -func (ms *MicroNode) serve(msgR *MsgInfo, p int) { +func (ms *MicroNode) call(rdata []byte, rkey int, sdata *[]byte) bool { + ri := &Reply{} if ms.handlers == nil { - return - } - - var reqBody Request - var ri *Reply - err := json.Unmarshal(msgR.Body, &reqBody) - if err != nil { - ms.printLog("serve unmarshal msgR.Body err:", err) - ri = &Reply { - Msg: err.Error(), - } + ri.Msg = "send wrong addr, check yourself!!!" } else { - ms.printLog("srcProc:", msgR.SrcProc,"reqBody Path:", reqBody.Path, " contentType:", reqBody.ContentType, " formMap:",reqBody.FormMap, " postFormMap:", reqBody.PostFormMap, "to key: ", p) - - if f,ok := ms.handlers[reqBody.Path];ok { - reqBody.SrcProc = msgR.SrcProc - ri = f(&reqBody) - ms.printLog("call funcMap f,reply:", *ri) + var msgR MsgInfo + err := json.Unmarshal(rdata, &msgR) + if err != nil { + ri.Msg = err.Error() } else { - ms.printLog("ms.funcMap not eixst path: ", reqBody.Path) - ri = &Reply{ - Success: false, - Msg: "璇锋眰鐨勬帴鍙d笉瀛樺湪锛岃妫�鏌rl", - Data: "璇锋眰鐨勬帴鍙d笉瀛樺湪锛岃妫�鏌rl", + var reqBody Request + err = json.Unmarshal(rdata, &msgR.Body) + if err != nil { + ri.Msg = err.Error() + } else { + ms.printLog("srcProc:", reqBody.SrcProc,"reqBody Path:", reqBody.Path, " contentType:", reqBody.ContentType, " formMap:",reqBody.FormMap, " postFormMap:", reqBody.PostFormMap, "to key: ", rkey) + if f,ok := ms.handlers[reqBody.Path];ok { + reqBody.SrcProc = msgR.SrcProc + ri = f(&reqBody) + ms.printLog("call funcMap f,reply:", *ri) + } else { + ms.printLog("ms.funcMap not eixst path: ", reqBody.Path) + ri.Msg = "璇锋眰鐨勬帴鍙d笉瀛樺湪锛岃妫�鏌rl" + } } } } + result, err := json.Marshal(*ri) + if err != nil { + sdata = nil + } else { + sdata = &result + } + return ri.Success +} - retErr := ms.handle.Reply(p, ri) - if retErr != nil { - ms.printLog("retErr:", retErr) +func (ms *MicroNode) serve() { + if ms.handlers == nil { + return + } + for i:=0;i<10;i++ { + ms.handle.wg.Add(1) + go recvandsendRoutine(ms.ctx, ms.handle.sockRep.sock, ms.handle.wg, ms.call, ms.fnLog) } } -- Gitblit v1.8.0