6 "github.com/jinzhu/gorm"
7 log "github.com/sirupsen/logrus"
9 "github.com/vapor/errors"
10 "github.com/vapor/protocol/bc/types"
11 apinode "github.com/vapor/toolbar/api_node"
12 "github.com/vapor/toolbar/common"
13 "github.com/vapor/toolbar/vote_reward/config"
14 "github.com/vapor/toolbar/vote_reward/database/orm"
17 var ErrInconsistentDB = errors.New("inconsistent db status")
19 type ChainKeeper struct {
25 func NewChainKeeper(db *gorm.DB, cfg *config.Config, targetHeight uint64) (*ChainKeeper, error) {
26 keeper := &ChainKeeper{
28 node: apinode.NewNode(cfg.NodeIP),
29 targetHeight: targetHeight,
32 chainStatus := &orm.ChainStatus{}
33 if err := db.First(chainStatus).Error; err == nil {
35 } else if err != gorm.ErrRecordNotFound {
36 return nil, errors.Wrap(err, "fail on get chainStatus")
39 if err := keeper.initBlockState(); err != nil {
40 return nil, errors.Wrap(err, "fail on init chainStatus")
45 func (c *ChainKeeper) SyncBlock() error {
47 chainStatus := &orm.ChainStatus{}
48 if err := c.db.First(chainStatus).Error; err != nil {
49 return errors.Wrap(err, "fail on syncBlock query chainStatus")
52 if chainStatus.BlockHeight >= c.targetHeight {
57 if err := c.syncChainStatus(dbTX, chainStatus); err != nil {
62 if err := dbTX.Commit().Error; err != nil {
69 func (c *ChainKeeper) syncChainStatus(db *gorm.DB, chainStatus *orm.ChainStatus) error {
70 nextBlock, err := c.node.GetBlockByHeight(chainStatus.BlockHeight + 1)
75 // Normal case, the previous hash of next block equals to the hash of current block,
76 // just sync to database directly.
77 if nextBlock.PreviousBlockHash.String() == chainStatus.BlockHash {
78 return c.AttachBlock(db, chainStatus, nextBlock)
81 log.WithField("block height", chainStatus.BlockHeight).Debug("the prev hash of remote is not equals the hash of current best block, must rollback")
82 currentBlock, err := c.node.GetBlockByHash(chainStatus.BlockHash)
87 return c.DetachBlock(db, chainStatus, currentBlock)
90 func (c *ChainKeeper) AttachBlock(db *gorm.DB, chainStatus *orm.ChainStatus, block *types.Block) error {
91 for _, tx := range block.Transactions {
92 for _, input := range tx.Inputs {
93 if input.TypedInput.InputType() != types.VetoInputType {
97 outputID, err := input.SpentOutputID()
102 result := db.Model(&orm.Utxo{}).Where(&orm.Utxo{OutputID: outputID.String()}).Update("veto_height", block.Height)
103 if err := result.Error; err != nil {
105 } else if result.RowsAffected != 1 {
106 return ErrInconsistentDB
110 for i, output := range tx.Outputs {
111 voteOutput, ok := output.TypedOutput.(*types.VoteOutput)
117 Xpub: hex.EncodeToString(voteOutput.Vote),
118 VoteAddress: common.GetAddressFromControlProgram(voteOutput.ControlProgram),
119 VoteHeight: block.Height,
120 VoteNum: voteOutput.Amount,
121 OutputID: tx.OutputID(i).String(),
124 if err := db.Save(utxo).Error; err != nil {
130 return c.updateChainStatus(db, chainStatus, block)
133 func (c *ChainKeeper) DetachBlock(db *gorm.DB, chainStatus *orm.ChainStatus, block *types.Block) error {
134 if err := db.Where(&orm.Utxo{VoteHeight: block.Height}).Delete(&orm.Utxo{}).Error; err != nil {
138 if err := db.Where(&orm.Utxo{VetoHeight: block.Height}).Update("veto_height", 0).Error; err != nil {
142 return c.updateChainStatus(db, chainStatus, block)
145 func (c *ChainKeeper) initBlockState() error {
146 block, err := c.node.GetBlockByHeight(0)
148 return errors.Wrap(err, "fail on get genenis block")
151 blockHash := block.Hash()
152 chainStatus := &orm.ChainStatus{
153 BlockHeight: block.Height,
154 BlockHash: blockHash.String(),
156 return c.db.Save(chainStatus).Error
159 func (c *ChainKeeper) updateChainStatus(db *gorm.DB, chainStatus *orm.ChainStatus, block *types.Block) error {
160 blockHash := block.Hash()
161 result := db.Model(&orm.ChainStatus{}).Where(chainStatus).Updates(&orm.ChainStatus{
162 BlockHeight: block.Height,
163 BlockHash: blockHash.String(),
165 if err := result.Error; err != nil {
167 } else if result.RowsAffected != 1 {
168 return ErrInconsistentDB