Fix: save validators (#41)

internal/storage/generic.go

@@ -76,7 +76,6 @@ type Transaction interface {
 	UpdateAddresses(ctx context.Context, address ...*Address) error
 	UpdateConstants(ctx context.Context, constants ...*Constant) error
 	UpdateRollups(ctx context.Context, rollups ...*Rollup) error
-	UpdateValidators(ctx context.Context, validators ...*Validator) error
 
 	LastBlock(ctx context.Context) (block Block, err error)
 	State(ctx context.Context, name string) (state State, err error)

internal/storage/postgres/transaction.go

@@ -415,24 +415,6 @@ func (tx Transaction) RetentionBlockSignatures(ctx context.Context, height types
 	return err
 }
 
-func (tx Transaction) UpdateValidators(ctx context.Context, validators ...*models.Validator) error {
-	if len(validators) == 0 {
-		return nil
-	}
-
-	for _, val := range validators {
-		_, err := tx.Tx().NewUpdate().
-			Model(val).
-			Where("pubkey = ?", val.PubKey).
-			Set("power = ?", val.Power).
-			Exec(ctx)
-		if err != nil {
-			return err
-		}
-	}
-	return nil
-}
-
 func (tx Transaction) UpdateConstants(ctx context.Context, constants ...*models.Constant) error {
 	if len(constants) == 0 {
 		return nil

internal/storage/postgres/transaction_test.go

@@ -754,26 +754,28 @@ func (s *TransactionTestSuite) TestRetentionBlockSignatures() {
 	s.Require().Len(signs, 3)
 }
 
-func (s *TransactionTestSuite) TestUpdateValidators() {
+func (s *TransactionTestSuite) TestCreateValidator() {
 	ctx, ctxCancel := context.WithTimeout(context.Background(), 5*time.Second)
 	defer ctxCancel()
 
 	tx, err := BeginTransaction(ctx, s.storage.Transactable)
 	s.Require().NoError(err)
 
-	pk, err := hex.DecodeString("32415f09dbee4297cc9a841c2c2312bf903fc53c48860d788ae66097355a585f")
+	pk, err := hex.DecodeString("52415f09dbee4297cc9a841c2c2312bf903fc53c48860d788ae66097355a5851")
 	s.Require().NoError(err)
 
-	err = tx.UpdateValidators(ctx, &storage.Validator{
+	val := &storage.Validator{
 		PubKey: pk,
 		Power:  decimal.NewFromInt(10000),
-	})
+	}
+	err = tx.SaveValidators(ctx, val)
 	s.Require().NoError(err)
+	s.Require().Greater(val.Id, uint64(0))
 
 	s.Require().NoError(tx.Flush(ctx))
 	s.Require().NoError(tx.Close(ctx))
 
-	validator, err := s.storage.Validator.GetByID(ctx, 1)
+	validator, err := s.storage.Validator.GetByID(ctx, val.Id)
 	s.Require().NoError(err)
 	s.Require().EqualValues("10000", validator.Power.String())
 }

internal/storage/validator.go

@@ -8,15 +8,14 @@ import (
 
 	pkgTypes "github.com/celenium-io/astria-indexer/pkg/types"
 
-	"github.com/dipdup-net/indexer-sdk/pkg/storage"
 	sdk "github.com/dipdup-net/indexer-sdk/pkg/storage"
 	"github.com/shopspring/decimal"
 	"github.com/uptrace/bun"
 )
 
 //go:generate mockgen -source=$GOFILE -destination=mock/$GOFILE -package=mock -typed
 type IValidator interface {
-	storage.Table[*Validator]
+	sdk.Table[*Validator]
 
 	ListByPower(ctx context.Context, limit, offset int, order sdk.SortOrder) ([]Validator, error)
 }

pkg/indexer/storage/validators.go

@@ -23,7 +23,7 @@ func (module *Module) saveValidators(
 		vals = append(vals, val)
 	}
 
-	if err := tx.UpdateValidators(ctx, vals...); err != nil {
+	if err := tx.SaveValidators(ctx, vals...); err != nil {
 		return err
 	}