Method recvLoop

internal/nodehub/hub_test.go:127–163

()

}

func (m *nodeMock) recvLoop() {
	defer close(m.done)
	for {
		msg, err := m.stream.Recv()
		if err != nil {
			return
		}
		if msg.GetCancelId() != "" {
			select {
			case m.cancelIDs <- msg.GetCancelId():
			default:
			}
			continue
		}
		if msg.GetMethod() == "ack" {
			select {
			case m.acks <- msg.GetBody():
			default:
			}
			continue
		}
		// 普通请求
		m.mu.Lock()
		fn, ok := m.handlers[msg.GetMethod()]
		m.mu.Unlock()
		if m.silent.Load() {
			continue
		}
		if !ok {
			// 默认回 ok=true, body=null
			m.sendNode(&nodev1.NodeMessage{Id: msg.GetId(), Ok: true})
			continue
		}
		ok2, body, errStr := fn(msg.GetId(), msg.GetBody())
		m.sendNode(&nodev1.NodeMessage{Id: msg.GetId(), Ok: ok2, Body: body, Error: errStr})
	}
}

func (m *nodeMock) sendNode(msg *nodev1.NodeMessage) {
	m.sendMu.Lock()