Merge pull request #33 from icon-project/IS-873-add-rollback-function

IS-873: add rollback function

cmd/dbtest/cli_calculate.go

@@ -21,5 +21,5 @@ func (cli *CLI) calculate(dbName string, blockHeight uint64, batchCount uint64)
  req.BlockHeight = blockHeight
  req.Path = "noiissdata"
 
- core.DoCalculate(ctx, &req, nil, 0)
+ core.DoCalculate(ctx.Rollback.GetChannel(), ctx, &req, nil, 0)
 }

cmd/icon_rc/main.go

@@ -80,7 +80,7 @@ func main() {
 
  rcm, err := core.InitManager(&cfg)
  if err != nil {
- log.Panicf("Failed to start RewardCalculator manager %+v", err)
+ log.Panicf("Failed to start RewardCalculator manager. %+v", err)
  }
 
  forever := make(chan bool)

cmd/iissdata/cli_read.go

@@ -3,6 +3,8 @@ package main
 import (
  "encoding/hex"
  "fmt"
+ "github.com/icon-project/rewardcalculator/common"
+ "log"
  "os"
  "path/filepath"
 
@@ -24,8 +26,8 @@ func (cli *CLI) read(dbDir string, dbName string) {
 
  iissDB := core.OpenIISSData(path)
  core.LoadIISSData(iissDB)
- core.ReadIISSBP(iissDB)
- core.ReadIISSTX(iissDB)
+ ReadIISSBP(iissDB)
+ ReadIISSTX(iissDB)
 
  //checkIISSTX(iissDB, lossDB)
 
@@ -34,6 +36,62 @@ func (cli *CLI) read(dbDir string, dbName string) {
  iissDB.Close()
 }
 
+func ReadIISSBP(iissDB db.Database) {
+ var bpInfo core.IISSBlockProduceInfo
+
+ iter, _ := iissDB.GetIterator()
+ prefix := util.BytesPrefix([]byte(db.PrefixIISSBPInfo))
+ iter.New(prefix.Start, prefix.Limit)
+ var entries, startBH, miss uint64
+ for entries = 0; iter.Next(); entries++ {
+ err := bpInfo.SetBytes(iter.Value())
+ if err != nil {
+ log.Printf("Failed to load IISS Block Produce information.")
+ continue
+ }
+ bpInfo.BlockHeight = common.BytesToUint64(iter.Key()[len(db.PrefixIISSBPInfo):])
+ if entries == 0 {
+ startBH = bpInfo.BlockHeight
+ }
+ if startBH + entries + miss != bpInfo.BlockHeight {
+ fmt.Printf("Miss BP block height %d\n", startBH + entries + miss)
+ miss++
+ for ; startBH + entries + miss != bpInfo.BlockHeight; miss++ {
+ fmt.Printf("Miss BP block height %d\n", startBH + entries + miss)
+ }
+ }
+ }
+ log.Printf(">> BP total count %d, miss %d", entries, miss)
+ iter.Release()
+ err := iter.Error()
+ if err != nil {
+ log.Printf("There is error while read IISS BP iteration. %+v", err)
+ }
+}
+
+func ReadIISSTX(iissDB db.Database) {
+ var tx core.IISSTX
+
+ iter, _ := iissDB.GetIterator()
+ prefix := util.BytesPrefix([]byte(db.PrefixIISSTX))
+ iter.New(prefix.Start, prefix.Limit)
+ entries := 0
+ for entries = 0; iter.Next(); entries++ {
+ err := tx.SetBytes(iter.Value())
+ if err != nil {
+ log.Printf("Failed to load IISS TX data")
+ continue
+ }
+ //log.Printf("[IISSTX] TX : %s", tx.String())
+ }
+ log.Printf(">> TX total count %d", entries)
+ iter.Release()
+ err := iter.Error()
+ if err != nil {
+ log.Printf("There is error while read IISS TX iteration. %+v", err)
+ }
+}
+
 func checkIISSTX(iissDB db.Database, result db.Database) {
  var tx core.IISSTX
 

cmd/sendipc/cli.go

@@ -72,6 +72,10 @@ func (cli *CLI) Run() {
  queryCRCmd := flag.NewFlagSet("query_calculate_result", flag.ExitOnError)
  queryCRBlockHeight := queryCRCmd.Uint64("blockheight", 0, "Block height(Required)")
 
+ rollbackCmd := flag.NewFlagSet("rollback", flag.ExitOnError)
+ rollbackBlockHeight := rollbackCmd.Uint64("blockheight", 0, "Rollback block height(Required)")
+ rollbackBlockHash := rollbackCmd.String("blockhash", "", "Rollback block hash(Required)")
+
  // Parse the CLI
  switch cmd {
  case "version":
@@ -116,6 +120,12 @@ func (cli *CLI) Run() {
  queryCRCmd.PrintDefaults()
  os.Exit(1)
  }
+ case "rollback":
+ err := rollbackCmd.Parse(os.Args[3:])
+ if err != nil {
+ rollbackCmd.PrintDefaults()
+ os.Exit(1)
+ }
  default:
  cli.printUsage()
  os.Exit(1)
@@ -202,4 +212,12 @@ func (cli *CLI) Run() {
  if monitorCmd.Parsed() {
  cli.monitor(conn, *monitorConfig, *monitorURL)
  }
+
+ if rollbackCmd.Parsed() {
+ if *rollbackBlockHeight == 0 || *rollbackBlockHash == "" {
+ rollbackCmd.PrintDefaults()
+ os.Exit(1)
+ }
+ cli.rollback(conn, *rollbackBlockHeight, *rollbackBlockHash)
+ }
 }

cmd/sendipc/cli_claim.go

@@ -16,22 +16,24 @@ func (cli *CLI) claim(conn ipc.Connection, address string, blockHeight uint64) {
  // Send CLAIM and get response
  req.Address.SetString(address)
  req.BlockHeight = blockHeight
- req.BlockHash = make([]byte, 8)
+ req.BlockHash = make([]byte, core.BlockHashSize)
  binary.BigEndian.PutUint64(req.BlockHash, blockHeight)
 
+ fmt.Printf("send CLAIM message: %s\n", req.String())
  conn.SendAndReceive(core.MsgClaim, cli.id, &req, &resp)
  cli.id++
- fmt.Printf("CLAIM command get response: %s\n", resp.String())
+ fmt.Printf("CLAIM message get response: %s\n", resp.String())
 
  // send COMMIT_CLAIM and get ack
  var commitClaim core.CommitClaim
+ commitClaim.Success = true
  commitClaim.Address = req.Address
  commitClaim.BlockHeight = req.BlockHeight
- commitClaim.BlockHash = req.BlockHash
- commitClaim.Success = true
+ commitClaim.BlockHash = make([]byte, core.BlockHashSize)
+ copy(commitClaim.BlockHash, req.BlockHash)
 
  fmt.Printf("Send COMMIT_CLAIM message: %s\n", commitClaim.String())
- conn.SendAndReceive(core.MsgClaim, cli.id, &req, &resp)
+ conn.SendAndReceive(core.MsgCommitClaim, cli.id, &commitClaim, &resp)
  cli.id++
  fmt.Printf("COMMIT_CLAIM message get ack\n")
 

cmd/sendipc/cli_rollback.go

@@ -0,0 +1,27 @@
+package main
+
+import (
+ "encoding/hex"
+ "fmt"
+
+ "github.com/icon-project/rewardcalculator/common/ipc"
+ "github.com/icon-project/rewardcalculator/core"
+)
+
+func (cli *CLI) rollback(conn ipc.Connection, blockHeight uint64, blockHash string) {
+ var req core.RollBackRequest
+ var resp core.RollBackResponse
+
+ hash, err := hex.DecodeString(blockHash)
+ if err != nil {
+ fmt.Printf("Failed to decode blockHash. %v\n", err)
+ return
+ }
+
+ req.BlockHeight = blockHeight
+ req.BlockHash = make([]byte, core.BlockHashSize)
+ copy(req.BlockHash, hash[0:core.BlockHashSize])
+
+ conn.SendAndReceive(core.MsgRollBack, cli.id, &req, &resp)
+ fmt.Printf("ROLLBACK command get response: %s\n", resp.String())
+}