-
Notifications
You must be signed in to change notification settings - Fork 4
/
quorum.go
106 lines (91 loc) · 2.29 KB
/
quorum.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
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
package dqpbft
import (
"github.com/ibalajiarun/go-consensus/peer/peerpb"
pb "github.com/ibalajiarun/go-consensus/protocols/dqpbft/dqpbftpb"
)
type Quorum struct {
p *DQPBFT
maxView pb.View
msgs map[pb.View]map[peerpb.PeerID]*pb.AgreementMessage
}
// newQuorum returns a new Quorum
func newQuorum(p *DQPBFT) *Quorum {
return &Quorum{
p: p,
msgs: make(map[pb.View]map[peerpb.PeerID]*pb.AgreementMessage),
}
}
// log adds id to quorum log records
func (q *Quorum) log(id peerpb.PeerID, m *pb.AgreementMessage) {
if q.maxView > m.View {
q.p.logger.Panicf("quorum view is greater %v > %v", q.maxView, m.View)
}
q.maxView = m.View
if _, exists := q.msgs[m.View]; !exists {
q.msgs[m.View] = make(map[peerpb.PeerID]*pb.AgreementMessage)
}
if _, exists := q.msgs[m.View][id]; !exists {
q.msgs[m.View][id] = m
} else {
q.p.logger.Panicf("multiple messages vcSender same replica %d", id)
}
}
// Majority quorum satisfied
func (q *Quorum) Majority(m *pb.AgreementMessage) bool {
if msgPairs, ok := q.msgs[m.View]; ok {
if len(msgPairs) <= 2*q.p.f {
return false
}
size := 0
for _, otherMsg := range msgPairs {
if m.Equals(otherMsg) {
size++
}
}
return size > 2*q.p.f
}
return false
}
type OQuorum struct {
p *DQPBFT
maxView pb.View
msgs map[pb.View]map[peerpb.PeerID]*pb.OAgreementMessage
}
// newQuorum returns a new Quorum
func newOQuorum(p *DQPBFT) *OQuorum {
return &OQuorum{
p: p,
msgs: make(map[pb.View]map[peerpb.PeerID]*pb.OAgreementMessage),
}
}
// log adds id to quorum log records
func (q *OQuorum) log(id peerpb.PeerID, m *pb.OAgreementMessage) {
if q.maxView > m.View {
q.p.logger.Panicf("quorum view is greater %v > %v", q.maxView, m.View)
}
q.maxView = m.View
if _, exists := q.msgs[m.View]; !exists {
q.msgs[m.View] = make(map[peerpb.PeerID]*pb.OAgreementMessage)
}
if _, exists := q.msgs[m.View][id]; !exists {
q.msgs[m.View][id] = m
} else {
q.p.logger.Panicf("multiple messages vcSender same replica %d", id)
}
}
// Majority quorum satisfied
func (q *OQuorum) Majority(m *pb.OAgreementMessage) bool {
if msgPairs, ok := q.msgs[m.View]; ok {
if len(msgPairs) <= 2*q.p.f {
return false
}
size := 0
for _, otherMsg := range msgPairs {
if m.Equals(otherMsg) {
size++
}
}
return size > 2*q.p.f
}
return false
}