74 lines
1.6 KiB
Go
74 lines
1.6 KiB
Go
package consensus
|
|
|
|
import (
|
|
"sync"
|
|
|
|
types2 "github.com/Secured-Finance/dione/consensus/types"
|
|
"github.com/sirupsen/logrus"
|
|
)
|
|
|
|
type CommitPool struct {
|
|
mut sync.RWMutex
|
|
commitMsgs map[string][]*types2.Message
|
|
}
|
|
|
|
func NewCommitPool() *CommitPool {
|
|
return &CommitPool{
|
|
commitMsgs: map[string][]*types2.Message{},
|
|
}
|
|
}
|
|
|
|
func (cp *CommitPool) CreateCommit(prepareMsg *types2.Message, privateKey []byte) (*types2.Message, error) {
|
|
var message types2.Message
|
|
message.Type = types2.MessageTypeCommit
|
|
newCMsg := prepareMsg.Payload
|
|
message.Payload = newCMsg
|
|
return &message, nil
|
|
}
|
|
|
|
func (cp *CommitPool) IsExistingCommit(commitMsg *types2.Message) bool {
|
|
cp.mut.RLock()
|
|
defer cp.mut.RUnlock()
|
|
|
|
consensusMessage := commitMsg.Payload
|
|
var exists bool
|
|
for _, v := range cp.commitMsgs[consensusMessage.Task.ConsensusID] {
|
|
if v.From == commitMsg.From {
|
|
exists = true
|
|
}
|
|
}
|
|
return exists
|
|
}
|
|
|
|
func (cp *CommitPool) IsValidCommit(commit *types2.Message) bool {
|
|
consensusMsg := commit.Payload
|
|
err := verifyTaskSignature(consensusMsg)
|
|
if err != nil {
|
|
logrus.Errorf("failed to verify task signature: %v", err)
|
|
return false
|
|
}
|
|
return true
|
|
}
|
|
|
|
func (cp *CommitPool) AddCommit(commit *types2.Message) {
|
|
cp.mut.Lock()
|
|
defer cp.mut.Unlock()
|
|
|
|
consensusID := commit.Payload.Task.ConsensusID
|
|
if _, ok := cp.commitMsgs[consensusID]; !ok {
|
|
cp.commitMsgs[consensusID] = []*types2.Message{}
|
|
}
|
|
|
|
cp.commitMsgs[consensusID] = append(cp.commitMsgs[consensusID], commit)
|
|
}
|
|
|
|
func (cp *CommitPool) CommitSize(consensusID string) int {
|
|
cp.mut.RLock()
|
|
defer cp.mut.RUnlock()
|
|
|
|
if v, ok := cp.commitMsgs[consensusID]; ok {
|
|
return len(v)
|
|
}
|
|
return 0
|
|
}
|