Method dispatch

p2p/client.go:295–368

(replyMsg, receivedMsg chan P2PMessage)

}

func (c *client) dispatch(replyMsg, receivedMsg chan P2PMessage) (out chan p2pRequest) {
	out = make(chan p2pRequest)
	requests := make(map[uint64]*p2pRequest)
	var nonce uint64
	idleTimer := time.NewTimer(idleTimeout)
	var readTimer bool
	go func() {
		defer close(out)
		for {
			var (
				ok  bool
				req p2pRequest
				msg P2PMessage
			)
			select {
			case <-c.ctx.Done():
				for _, req := range requests {
					req.replyResult(nil, errors.Errorf("client dispatch: %w", c.ctx.Err()))
				}
				if !idleTimer.Stop() && !readTimer {
					<-idleTimer.C
				}
				return
			case <-idleTimer.C:
				readTimer = true
				err := c.close()
				if err != nil {
					errors.Errorf("client close: %w", err)
				}
			case req, ok = <-c.peerSend:
				if ok {
					if !idleTimer.Stop() {
						<-idleTimer.C
					}
					idleTimer.Reset(idleTimeout)
					if req.rType != replyReq {
						req.nonce = nonce
						requests[nonce] = &req
						nonce++
					}
					select {
					case <-c.ctx.Done():
					case out <- req:
					}
				}
			case msg, ok = <-replyMsg:
				if ok {
					if !idleTimer.Stop() {
						<-idleTimer.C
					}
					idleTimer.Reset(idleTimeout)
					p2pRequest := requests[msg.RequestNonce]
					if p2pRequest != nil {
						delete(requests, msg.RequestNonce)
						select {
						case <-p2pRequest.ctx.Done():
						default:
							p2pRequest.replyResult(msg, nil)