add logs to rollback

baking-bad · Apr 29, 2022 · 09c5335 · 09c5335
1 parent ce215f2
commit 09c5335
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/internal/rollback/rollback.go b/internal/rollback/rollback.go
@@ -86,6 +86,7 @@ func (rm Manager) Rollback(ctx context.Context, db pg.DBI, network types.Network
 }
 
 func (rm Manager) rollbackTokenBalances(tx pg.DBI, level int64) error {
+	logger.Info().Msg("rollback token balances...")
 	transfers, err := rm.transfersRepo.GetAll(level)
 	if err != nil {
 		return err
@@ -145,6 +146,7 @@ func (rm Manager) rollbackAll(tx pg.DBI, level int64) error {
 }
 
 func (rm Manager) rollbackMigrations(tx pg.DBI, level int64) error {
+	logger.Info().Msg("rollback migrations...")
 	if _, err := tx.Model(new(migration.Migration)).
 		Where("contract_id IN (?)", tx.Model(new(contract.Contract)).Column("id").Where("level > ?", level)).
 		Where("level = ?", level).
@@ -155,6 +157,7 @@ func (rm Manager) rollbackMigrations(tx pg.DBI, level int64) error {
 }
 
 func (rm Manager) rollbackBigMapState(tx pg.DBI, level int64) error {
+	logger.Info().Msg("rollback big map states...")
 	states, err := rm.bmdRepo.StatesChangedAfter(level)
 	if err != nil {
 		return err
@@ -201,6 +204,7 @@ type lastAction struct {
 }
 
 func (rm Manager) rollbackOperations(tx pg.DBI, level int64) error {
+	logger.Info().Msg("rollback oeprations...")
 	var ops []operation.Operation
 	if err := tx.Model(&operation.Operation{}).
 		Where("level = ?", level).