Thread safe Decider. (#4610)
Konstantin
10 months ago
committed by
GitHub
parent
1dd67a801c
commit
a31b4f5640
@ -0,0 +1,179 @@ |
||||
package quorum |
||||
|
||||
import ( |
||||
"math/big" |
||||
"sync" |
||||
|
||||
"github.com/ethereum/go-ethereum/common" |
||||
bls_core "github.com/harmony-one/bls/ffi/go/bls" |
||||
"github.com/harmony-one/harmony/consensus/votepower" |
||||
"github.com/harmony-one/harmony/crypto/bls" |
||||
shardingconfig "github.com/harmony-one/harmony/internal/configs/sharding" |
||||
"github.com/harmony-one/harmony/multibls" |
||||
"github.com/harmony-one/harmony/numeric" |
||||
"github.com/harmony-one/harmony/shard" |
||||
) |
||||
|
||||
var _ Decider = threadSafeDeciderImpl{} |
||||
|
||||
type threadSafeDeciderImpl struct { |
||||
mu *sync.RWMutex |
||||
decider Decider |
||||
} |
||||
|
||||
func NewThreadSafeDecider(decider Decider, mu *sync.RWMutex) Decider { |
||||
return threadSafeDeciderImpl{ |
||||
mu: mu, |
||||
decider: decider, |
||||
} |
||||
} |
||||
|
||||
func (a threadSafeDeciderImpl) String() string { |
||||
a.mu.Lock() |
||||
defer a.mu.Unlock() |
||||
return a.decider.String() |
||||
} |
||||
|
||||
func (a threadSafeDeciderImpl) Participants() multibls.PublicKeys { |
||||
a.mu.Lock() |
||||
defer a.mu.Unlock() |
||||
return a.decider.Participants() |
||||
} |
||||
|
||||
func (a threadSafeDeciderImpl) IndexOf(key bls.SerializedPublicKey) int { |
||||
a.mu.Lock() |
||||
defer a.mu.Unlock() |
||||
return a.decider.IndexOf(key) |
||||
} |
||||
|
||||
func (a threadSafeDeciderImpl) ParticipantsCount() int64 { |
||||
a.mu.Lock() |
||||
defer a.mu.Unlock() |
||||
return a.decider.ParticipantsCount() |
||||
} |
||||
|
||||
func (a threadSafeDeciderImpl) NthNextValidator(slotList shard.SlotList, pubKey *bls.PublicKeyWrapper, next int) (bool, *bls.PublicKeyWrapper) { |
||||
a.mu.Lock() |
||||
defer a.mu.Unlock() |
||||
return a.decider.NthNextValidator(slotList, pubKey, next) |
||||
} |
||||
|
||||
func (a threadSafeDeciderImpl) NthNextHmy(instance shardingconfig.Instance, pubkey *bls.PublicKeyWrapper, next int) (bool, *bls.PublicKeyWrapper) { |
||||
a.mu.Lock() |
||||
defer a.mu.Unlock() |
||||
return a.decider.NthNextHmy(instance, pubkey, next) |
||||
} |
||||
|
||||
func (a threadSafeDeciderImpl) NthNextHmyExt(instance shardingconfig.Instance, wrapper *bls.PublicKeyWrapper, i int) (bool, *bls.PublicKeyWrapper) { |
||||
a.mu.Lock() |
||||
defer a.mu.Unlock() |
||||
return a.decider.NthNextHmyExt(instance, wrapper, i) |
||||
} |
||||
|
||||
func (a threadSafeDeciderImpl) FirstParticipant(instance shardingconfig.Instance) *bls.PublicKeyWrapper { |
||||
a.mu.Lock() |
||||
defer a.mu.Unlock() |
||||
return a.decider.FirstParticipant(instance) |
||||
} |
||||
|
||||
func (a threadSafeDeciderImpl) UpdateParticipants(pubKeys, allowlist []bls.PublicKeyWrapper) { |
||||
a.mu.Lock() |
||||
defer a.mu.Unlock() |
||||
a.decider.UpdateParticipants(pubKeys, allowlist) |
||||
} |
||||
|
||||
func (a threadSafeDeciderImpl) submitVote(p Phase, pubkeys []bls.SerializedPublicKey, sig *bls_core.Sign, headerHash common.Hash, height, viewID uint64) (*votepower.Ballot, error) { |
||||
a.mu.Lock() |
||||
defer a.mu.Unlock() |
||||
return a.decider.submitVote(p, pubkeys, sig, headerHash, height, viewID) |
||||
} |
||||
|
||||
func (a threadSafeDeciderImpl) SignersCount(phase Phase) int64 { |
||||
a.mu.Lock() |
||||
defer a.mu.Unlock() |
||||
return a.decider.SignersCount(phase) |
||||
} |
||||
|
||||
func (a threadSafeDeciderImpl) reset(phases []Phase) { |
||||
a.mu.Lock() |
||||
defer a.mu.Unlock() |
||||
a.decider.reset(phases) |
||||
} |
||||
|
||||
func (a threadSafeDeciderImpl) ReadBallot(p Phase, pubkey bls.SerializedPublicKey) *votepower.Ballot { |
||||
a.mu.Lock() |
||||
defer a.mu.Unlock() |
||||
return a.decider.ReadBallot(p, pubkey) |
||||
} |
||||
|
||||
func (a threadSafeDeciderImpl) TwoThirdsSignersCount() int64 { |
||||
a.mu.Lock() |
||||
defer a.mu.Unlock() |
||||
return a.decider.TwoThirdsSignersCount() |
||||
} |
||||
|
||||
func (a threadSafeDeciderImpl) AggregateVotes(p Phase) *bls_core.Sign { |
||||
a.mu.Lock() |
||||
defer a.mu.Unlock() |
||||
return a.decider.AggregateVotes(p) |
||||
} |
||||
|
||||
func (a threadSafeDeciderImpl) SetVoters(subCommittee *shard.Committee, epoch *big.Int) (*TallyResult, error) { |
||||
a.mu.Lock() |
||||
defer a.mu.Unlock() |
||||
return a.decider.SetVoters(subCommittee, epoch) |
||||
} |
||||
|
||||
func (a threadSafeDeciderImpl) Policy() Policy { |
||||
a.mu.Lock() |
||||
defer a.mu.Unlock() |
||||
return a.decider.Policy() |
||||
} |
||||
|
||||
func (a threadSafeDeciderImpl) AddNewVote(p Phase, pubkeys []*bls.PublicKeyWrapper, sig *bls_core.Sign, headerHash common.Hash, height, viewID uint64) (*votepower.Ballot, error) { |
||||
a.mu.Lock() |
||||
defer a.mu.Unlock() |
||||
return a.decider.AddNewVote(p, pubkeys, sig, headerHash, height, viewID) |
||||
} |
||||
|
||||
func (a threadSafeDeciderImpl) IsQuorumAchievedByMask(mask *bls.Mask) bool { |
||||
a.mu.Lock() |
||||
defer a.mu.Unlock() |
||||
return a.decider.IsQuorumAchievedByMask(mask) |
||||
} |
||||
|
||||
func (a threadSafeDeciderImpl) QuorumThreshold() numeric.Dec { |
||||
a.mu.Lock() |
||||
defer a.mu.Unlock() |
||||
return a.decider.QuorumThreshold() |
||||
} |
||||
|
||||
func (a threadSafeDeciderImpl) IsAllSigsCollected() bool { |
||||
a.mu.Lock() |
||||
defer a.mu.Unlock() |
||||
return a.decider.IsAllSigsCollected() |
||||
} |
||||
|
||||
func (a threadSafeDeciderImpl) ResetPrepareAndCommitVotes() { |
||||
a.mu.Lock() |
||||
defer a.mu.Unlock() |
||||
a.decider.ResetPrepareAndCommitVotes() |
||||
} |
||||
|
||||
func (a threadSafeDeciderImpl) ResetViewChangeVotes() { |
||||
a.mu.Lock() |
||||
defer a.mu.Unlock() |
||||
a.decider.ResetViewChangeVotes() |
||||
} |
||||
|
||||
func (a threadSafeDeciderImpl) CurrentTotalPower(p Phase) (*numeric.Dec, error) { |
||||
a.mu.Lock() |
||||
defer a.mu.Unlock() |
||||
return a.decider.CurrentTotalPower(p) |
||||
} |
||||
|
||||
func (a threadSafeDeciderImpl) IsQuorumAchieved(p Phase) bool { |
||||
a.mu.Lock() |
||||
defer a.mu.Unlock() |
||||
return a.decider.IsQuorumAchieved(p) |
||||
} |
||||
Loading…
Reference in new issue