dione/consensus/commit_pool.go

62 lines
1.5 KiB
Go
Raw Normal View History

2020-11-15 13:46:58 +00:00
package consensus
import (
types2 "github.com/Secured-Finance/dione/consensus/types"
"github.com/sirupsen/logrus"
2020-11-15 13:46:58 +00:00
)
type CommitPool struct {
commitMsgs map[string][]*types2.Message
2020-11-15 13:46:58 +00:00
}
func NewCommitPool() *CommitPool {
return &CommitPool{
commitMsgs: map[string][]*types2.Message{},
2020-11-15 13:46:58 +00:00
}
}
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
2020-11-15 13:46:58 +00:00
return &message, nil
}
func (cp *CommitPool) IsExistingCommit(commitMsg *types2.Message) bool {
2020-11-15 13:46:58 +00:00
consensusMessage := commitMsg.Payload
var exists bool
for _, v := range cp.commitMsgs[consensusMessage.ConsensusID] {
if v.From == commitMsg.From {
exists = true
}
}
return exists
}
func (cp *CommitPool) IsValidCommit(commit *types2.Message) bool {
2020-11-15 13:46:58 +00:00
consensusMsg := commit.Payload
err := verifyTaskSignature(consensusMsg)
2020-11-15 13:46:58 +00:00
if err != nil {
logrus.Errorf("failed to verify task signature: %v", err)
2020-11-15 13:46:58 +00:00
return false
}
return true
}
func (cp *CommitPool) AddCommit(commit *types2.Message) {
2020-11-15 13:46:58 +00:00
consensusID := commit.Payload.ConsensusID
if _, ok := cp.commitMsgs[consensusID]; !ok {
cp.commitMsgs[consensusID] = []*types2.Message{}
2020-11-15 13:46:58 +00:00
}
cp.commitMsgs[consensusID] = append(cp.commitMsgs[consensusID], commit)
}
func (cp *CommitPool) CommitSize(consensusID string) int {
if v, ok := cp.commitMsgs[consensusID]; ok {
return len(v)
}
return 0
}