MicroFunc添加Context参数

liuxiaolong

2021-04-02 a35153875f213929601a39c47f0823b310210321

 micronode.go

@@ -7,6 +7,7 @@
   "errors"
   "fmt"
   "os"
   "sync"
   "time"
)

@@ -20,6 +21,9 @@
   fnLog       func(...interface{})

   SubCh       chan *MsgInfo

   mtx       sync.Mutex
   started    bool
}

func NewMicroNode(ctx context.Context,q chan os.Signal, serverId string, reg *RegisterInfo, fnLog func(v ...interface{})) (*MicroNode, error){
@@ -83,10 +87,21 @@
}

func (ms *MicroNode) StartClient() {
   ms.mtx.Lock()
   defer ms.mtx.Unlock()
   if !ms.started {
      ms.started = true

   go ms.startHeartbeat()
   }
}

func (ms *MicroNode) StartServer(funcMap map[string]MicroFunc) {
   ms.mtx.Lock()
   if !ms.started {
      ms.started = true
      ms.mtx.Unlock()

   ms.handlers = funcMap

   go ms.startHeartbeat()
@@ -111,13 +126,12 @@
      }
   }




   //接收订阅到的消息
   //go ms.startRecvSubMsg()
   //作为server启动
   //ms.serve()
   }
   ms.mtx.Unlock()
}

//开始接收订阅消息
@@ -284,7 +298,11 @@

      if f,ok := ms.handlers[reqBody.Path];ok {
         reqBody.SrcProc = msgR.SrcProc
         ri = f(&reqBody)
         ctx := Context{
            ms,
            ms,
         }
         ri = f(&ctx, &reqBody)
         ms.printLog("call funcMap f,reply.Success:", ri.Success)
      } else {
         ms.printLog("ms.funcMap not eixst path: ", reqBody.Path)