+++ /dev/null
-package synchron
-
-import (
- "encoding/hex"
- "time"
-
- "github.com/jinzhu/gorm"
- log "github.com/sirupsen/logrus"
-
- "github.com/vapor/errors"
- "github.com/vapor/protocol/bc"
- "github.com/vapor/protocol/bc/types"
- "github.com/vapor/toolbar/common"
- "github.com/vapor/toolbar/common/service"
- "github.com/vapor/toolbar/reward/config"
- "github.com/vapor/toolbar/reward/database/orm"
-)
-
-type ChainKeeper struct {
- cfg *config.Chain
- db *gorm.DB
- node *service.Node
-}
-
-func NewChainKeeper(db *gorm.DB, cfg *config.Config) *ChainKeeper {
- return &ChainKeeper{
- cfg: &cfg.Chain,
- db: db,
- node: service.NewNode(cfg.Chain.Upstream),
- }
-}
-
-func (c *ChainKeeper) Run() {
- ticker := time.NewTicker(time.Duration(c.cfg.SyncSeconds) * time.Second)
- for ; true; <-ticker.C {
- for {
- isUpdate, err := c.syncBlock()
- if err != nil {
- log.WithField("error", err).Errorln("blockKeeper fail on process block")
- break
- }
-
- if !isUpdate {
- break
- }
- }
- }
-}
-
-func (c *ChainKeeper) syncBlock() (bool, error) {
- blockState := &orm.BlockState{}
- if err := c.db.First(blockState).Error; err != nil {
- return false, errors.Wrap(err, "query chain")
- }
-
- height, err := c.node.GetBlockCount()
- if err != nil {
- return false, err
- }
-
- if height == blockState.Height {
- return false, nil
- }
-
- nextBlockStr, txStatus, err := c.node.GetBlockByHeight(blockState.Height + 1)
- if err != nil {
- return false, err
- }
-
- nextBlock := &types.Block{}
- if err := nextBlock.UnmarshalText([]byte(nextBlockStr)); err != nil {
- return false, errors.New("Unmarshal nextBlock")
- }
-
- // Normal case, the previous hash of next block equals to the hash of current block,
- // just sync to database directly.
- if nextBlock.PreviousBlockHash.String() == blockState.BlockHash {
- return true, c.AttachBlock(nextBlock, txStatus)
- }
-
- log.WithField("block height", blockState.Height).Debug("the prev hash of remote is not equals the hash of current best block, must rollback")
- currentBlockStr, txStatus, err := c.node.GetBlockByHash(blockState.BlockHash)
- if err != nil {
- return false, err
- }
-
- currentBlock := &types.Block{}
- if err := nextBlock.UnmarshalText([]byte(currentBlockStr)); err != nil {
- return false, errors.New("Unmarshal currentBlock")
- }
-
- return true, c.DetachBlock(currentBlock, txStatus)
-}
-
-func (c *ChainKeeper) AttachBlock(block *types.Block, txStatus *bc.TransactionStatus) error {
- ormDB := c.db.Begin()
- for pos, tx := range block.Transactions {
- statusFail, err := txStatus.GetStatus(pos)
- if err != nil {
- return err
- }
-
- if statusFail {
- log.WithFields(log.Fields{"block height": block.Height, "statusFail": statusFail}).Debug("AttachBlock")
- continue
- }
-
- for _, input := range tx.Inputs {
- vetoInput, ok := input.TypedInput.(*types.VetoInput)
- if !ok {
- continue
- }
-
- outputID, err := input.SpentOutputID()
- if err != nil {
- return err
- }
- utxo := &orm.Utxo{
- VoterAddress: common.GetAddressFromControlProgram(vetoInput.ControlProgram),
- OutputID: outputID.String(),
- }
- // update data
- db := ormDB.Where(utxo).Update("veto_height", block.Height)
- if err := db.Error; err != nil {
- ormDB.Rollback()
- return err
- }
-
- if db.RowsAffected != 1 {
- ormDB.Rollback()
- return ErrInconsistentDB
- }
-
- }
-
- for index, output := range tx.Outputs {
- voteOutput, ok := output.TypedOutput.(*types.VoteOutput)
- if !ok {
- continue
- }
- pubkey := hex.EncodeToString(voteOutput.Vote)
- outputID := tx.OutputID(index)
- utxo := &orm.Utxo{
- Xpub: pubkey,
- VoterAddress: common.GetAddressFromControlProgram(voteOutput.ControlProgram),
- VoteHeight: block.Height,
- VoteNum: voteOutput.Amount,
- VetoHeight: 0,
- OutputID: outputID.String(),
- }
- // insert data
- if err := ormDB.Save(utxo).Error; err != nil {
- ormDB.Rollback()
- return err
- }
- }
- }
-
- return ormDB.Commit().Error
-}
-
-func (c *ChainKeeper) DetachBlock(block *types.Block, txStatus *bc.TransactionStatus) error {
- ormDB := c.db.Begin()
-
- utxo := &orm.Utxo{
- VoteHeight: block.Height,
- }
- // insert data
- if err := ormDB.Where(utxo).Delete(&orm.Utxo{}).Error; err != nil {
- ormDB.Rollback()
- return err
- }
-
- utxo = &orm.Utxo{
- VetoHeight: block.Height,
- }
-
- // update data
- if err := ormDB.Where(utxo).Update("veto_height", 0).Error; err != nil {
- ormDB.Rollback()
- return err
- }
-
- return ormDB.Commit().Error
-}
OSDN Git Service
