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