调整超时为3s

liuxiaolong

2021-02-07 abfd401180d1eb09c8ed1c24797c3e503e45fa08

 hbusc.go

@@ -7,7 +7,6 @@
   "errors"
   "fmt"
   "os"
   "strconv"
   "sync"
   "time"
)
@@ -48,10 +47,9 @@
   sockSub *sockClient  //订阅主题的socket，线程实时接收消息，需要单独处理

   sockWorker    *sockClient  //发给任意的server,短暂的request client
   mtxWorker    sync.Mutex    //SendAndRecv可能不是线程安全的
   //mtxWorker    sync.Mutex    //SendAndRecv可能不是线程安全的

   chSub chan TransInfo
   chReply chan TransInfo
}

//获取其他进程发给此socket的消息,可以是sub的接收，也可以是reply的接收。
@@ -65,7 +63,7 @@
         wg.Done()
         return
      default:
         n := s.RecvfromTimeout(&data, &key, 100) //目前10001返回值表示超时
         n := s.RecvfromTimeout(&data, &key, 1000) //目前10001返回值表示超时
         if n == 0 {
            var info MsgInfo
            if err := json.Unmarshal(data, &info);err == nil {
@@ -80,7 +78,25 @@
               logFn("unmarshal to MsgInfo err:", err)
            }
         } else {
            time.Sleep(10 * time.Millisecond)
            time.Sleep(100 * time.Millisecond)
         }
      }
   }
}

func recvandsendRoutine(ctx context.Context, s *bhomebus.Socket, wg *sync.WaitGroup,serveFn func(rdata []byte, rkey int, sdata *[]byte) bool, logFn func(...interface{})) {
   for {
      select {
      case <-ctx.Done():
         logFn("recvandsendRoutine ctx.Done")
         wg.Done()
         return
      default:
         n := s.RecvandsendTimeout(1000, serveFn) //目前10001返回值表示超时
         if n != 0 {
            logFn("RecvandsendTimeout success")
         } else {
            //time.Sleep(100 * time.Millisecond)
         }
      }
   }
@@ -88,11 +104,11 @@

//Register
func Register(ctx context.Context, q chan os.Signal, config *Config, ri *RegisterInfo) (*BHBus,error) {
   handle := &BHBus{
   handle := &BHBus {
      ctx: ctx,
      conf: config,
      m: make(map[string]*sockServer),
      chSub: make(chan TransInfo, config.chSize),
      chReply: make(chan TransInfo, config.chSize),
   }

   var err error
@@ -199,6 +215,7 @@

   //维持心跳的socket
   sockHB := bhomebus.OpenSocket()
   handle.printLog("open sockHB")
   handle.sockHB = &sockClient{
      sock: sockHB,
      peer: int(regR.HeartbeatKey),
@@ -209,9 +226,10 @@
   if ri.PubTopic != nil && len(ri.PubTopic) > 0 {
      sockReply := bhomebus.OpenSocket()
      sockReply.ForceBind(int(regR.ReplyKey))
      handle.wg.Add(1)
      handle.printLog("after pubTopic forceBind")
      //handle.wg.Add(1)
      //serve server reply
      go recvRoutine(ctx, sockReply, handle.wg, handle.chReply, handle.printLog)
      //go recvRoutine(ctx, sockReply, handle.wg, handle.chReply, handle.printLog)
      handle.sockRep = &sockServer{
         sock: sockReply,
         info: &ri.Proc,
@@ -231,6 +249,7 @@
      //订阅消息的socket
      sockSub := bhomebus.OpenSocket()
      //订阅所有主题
      handle.printLog("start Sub topics")
      for _,v := range ri.SubTopic {
         subN := sockSub.Sub(v)
         handle.printLog("subTopic:", v, " ret n:", subN)
@@ -270,8 +289,8 @@
      return err
   }

   h.mtxWorker.Lock()
   defer h.mtxWorker.Unlock()
   //h.mtxWorker.Lock()
   //defer h.mtxWorker.Unlock()
   netNodes := append([]bhomebus.NetNode{}, bhomebus.NetNode{
      Key: h.conf.regKey,
   })
@@ -405,7 +424,7 @@
            if err == nil {
               return nodes, nil
            } else {
               h.printLog("unmarshal err:", err, "nodes:", nodes)
               h.printLog("unmarshal err:", err, "nodes:", nodes, " center reply.Data:", reply.Data)
               return nil, fmt.Errorf("unmarshal to nodes err:%s", err.Error())
            }
         } else {
@@ -465,19 +484,19 @@
   return nil, fmt.Errorf("request err")
}

func (h *BHBus) Reply(replyKey int, i *Reply) error {
   data,err := json.Marshal(*i)
   if err != nil {
      return err
   }

   n := h.sockRep.sock.SendtoTimeout(data, replyKey, h.conf.sendTimeOut)
   h.printLog("reply to key:", replyKey, " n:",n)
   if n != 0 {
      return errors.New("reply sendToTimeOut n:"+strconv.Itoa(n))
   }
   return nil
}
//func (h *BHBus) Reply(replyKey int, i *Reply) error {
//   data,err := json.Marshal(*i)
//   if err != nil {
//      return err
//   }
//
//   n := h.sockRep.sock.SendtoTimeout(data, replyKey, h.conf.sendTimeOut)
//   h.printLog("reply to key:", replyKey, " n:",n)
//   if n != 0 {
//      return errors.New("reply sendToTimeOut n:"+strconv.Itoa(n))
//   }
//   return nil
//}


//只发送请求，不需要应答.
@@ -487,8 +506,8 @@
   if err != nil {
      return err
   }
   h.mtxWorker.Lock()
   defer h.mtxWorker.Unlock()
   //h.mtxWorker.Lock()
   //defer h.mtxWorker.Unlock()
   n := h.sockWorker.sock.SendtoTimeout(data, key, h.conf.sendTimeOut)
   if n != 0 {
      return fmt.Errorf("sendOnly ret n:%d", n)
@@ -504,8 +523,8 @@
   rNodes := append([]bhomebus.NetNode{}, bhomebus.NetNode{
      Key: KEY_QUERY,
   })
   h.mtxWorker.Lock()
   defer h.mtxWorker.Unlock()
   //h.mtxWorker.Lock()
   //defer h.mtxWorker.Unlock()
   var ret []bhomebus.Mesg
   n := h.sockWorker.sock.SendandrecvTimeout(rNodes, data,&ret, h.conf.sendTimeOut)
   h.printLog("requestCenter n:", n, "len(ret):", len(ret))
@@ -535,6 +554,14 @@
   return err
}

func (h *BHBus) PubTimeout(nodes []bhomebus.NetNode, msg *MsgInfo, timeout int) int {
   data,err := json.Marshal(*msg)
   if err == nil {
      return h.sockPub.sock.PubTimeout(nodes, msg.Topic, data, timeout)
   }
   return -1
}

//追加订阅的主题消息
func (h *BHBus) Sub(topics []string) {
   if topics != nil {
@@ -557,18 +584,13 @@


//获取sub 或者需要reply的消息
func (h *BHBus) GetMsg() (subMsg *MsgInfo, replyMsg *MsgInfo, replyKey int) {
   if h.sockHB == nil && h.sockRep == nil && h.sockPub == nil && h.sockSub == nil && h.sockWorker == nil {
      return nil,nil, -1
func (h *BHBus) GetMsg() (subMsg *MsgInfo) {
   if h.sockHB == nil && h.sockPub == nil && h.sockSub == nil && h.sockWorker == nil {
      return nil
   }
   if len(h.chSub) >0 {
      m  := <-h.chSub
      subMsg = m.info
   }
   if len(h.chReply) >0 {
      m := <-h.chReply
      replyMsg = m.info
      replyKey = m.port
   }
   return
}