github.com/lzy4123/fabric@v2.1.1+incompatible/orderer/consensus/etcdraft/dispatcher.go (about)

     1  /*
     2  Copyright IBM Corp. All Rights Reserved.
     3  
     4  SPDX-License-Identifier: Apache-2.0
     5  */
     6  
     7  package etcdraft
     8  
     9  import (
    10  	"github.com/hyperledger/fabric-protos-go/orderer"
    11  	"github.com/hyperledger/fabric/common/flogging"
    12  	"github.com/pkg/errors"
    13  )
    14  
    15  //go:generate mockery -dir . -name MessageReceiver -case underscore -output mocks
    16  
    17  // MessageReceiver receives messages
    18  type MessageReceiver interface {
    19  	// Consensus passes the given ConsensusRequest message to the MessageReceiver
    20  	Consensus(req *orderer.ConsensusRequest, sender uint64) error
    21  
    22  	// Submit passes the given SubmitRequest message to the MessageReceiver
    23  	Submit(req *orderer.SubmitRequest, sender uint64) error
    24  }
    25  
    26  //go:generate mockery -dir . -name ReceiverGetter -case underscore -output mocks
    27  
    28  // ReceiverGetter obtains instances of MessageReceiver given a channel ID
    29  type ReceiverGetter interface {
    30  	// ReceiverByChain returns the MessageReceiver if it exists, or nil if it doesn't
    31  	ReceiverByChain(channelID string) MessageReceiver
    32  }
    33  
    34  // Dispatcher dispatches Submit and Step requests to the designated per chain instances
    35  type Dispatcher struct {
    36  	Logger        *flogging.FabricLogger
    37  	ChainSelector ReceiverGetter
    38  }
    39  
    40  // OnConsensus notifies the Dispatcher for a reception of a StepRequest from a given sender on a given channel
    41  func (d *Dispatcher) OnConsensus(channel string, sender uint64, request *orderer.ConsensusRequest) error {
    42  	receiver := d.ChainSelector.ReceiverByChain(channel)
    43  	if receiver == nil {
    44  		d.Logger.Warningf("An attempt to send a consensus request to a non existing channel (%s) was made by %d", channel, sender)
    45  		return errors.Errorf("channel %s doesn't exist", channel)
    46  	}
    47  	return receiver.Consensus(request, sender)
    48  }
    49  
    50  // OnSubmit notifies the Dispatcher for a reception of a SubmitRequest from a given sender on a given channel
    51  func (d *Dispatcher) OnSubmit(channel string, sender uint64, request *orderer.SubmitRequest) error {
    52  	receiver := d.ChainSelector.ReceiverByChain(channel)
    53  	if receiver == nil {
    54  		d.Logger.Warningf("An attempt to submit a transaction to a non existing channel (%s) was made by %d", channel, sender)
    55  		return errors.Errorf("channel %s doesn't exist", channel)
    56  	}
    57  	return receiver.Submit(request, sender)
    58  }