fix: multiple expiry documents for same staking tx can exist (#97)

internal/db/model/timelock.go

@@ -3,7 +3,7 @@ package model
 import "github.com/babylonlabs-io/babylon-staking-indexer/internal/types"
 
 type TimeLockDocument struct {
-	StakingTxHashHex   string                   `bson:"_id"` // Primary key
+	StakingTxHashHex   string                   `bson:"staking_tx_hash_hex"`
 	ExpireHeight       uint32                   `bson:"expire_height"`
 	DelegationSubState types.DelegationSubState `bson:"delegation_sub_state"`
 }

internal/db/timelock.go

@@ -2,13 +2,11 @@ package db
 
 import (
 	"context"
-	"errors"
 	"fmt"
 
 	"github.com/babylonlabs-io/babylon-staking-indexer/internal/db/model"
 	"github.com/babylonlabs-io/babylon-staking-indexer/internal/types"
 	"go.mongodb.org/mongo-driver/bson"
-	"go.mongodb.org/mongo-driver/mongo"
 	"go.mongodb.org/mongo-driver/mongo/options"
 )
 
@@ -22,21 +20,7 @@ func (db *Database) SaveNewTimeLockExpire(
 	_, err := db.client.Database(db.dbName).
 		Collection(model.TimeLockCollection).
 		InsertOne(ctx, tlDoc)
-	if err != nil {
-		var writeErr mongo.WriteException
-		if errors.As(err, &writeErr) {
-			for _, e := range writeErr.WriteErrors {
-				if mongo.IsDuplicateKeyError(e) {
-					return &DuplicateKeyError{
-						Key:     tlDoc.StakingTxHashHex,
-						Message: "timelock already exists",
-					}
-				}
-			}
-		}
-		return err
-	}
-	return nil
+	return err
 }
 
 func (db *Database) FindExpiredDelegations(ctx context.Context, btcTipHeight, limit uint64) ([]model.TimeLockDocument, error) {

internal/services/expiry_checker.go

@@ -4,6 +4,7 @@ import (
 	"context"
 	"fmt"
 	"net/http"
+	"strconv"
 
 	"github.com/babylonlabs-io/babylon-staking-indexer/internal/types"
 	"github.com/babylonlabs-io/babylon-staking-indexer/internal/utils"
@@ -47,6 +48,8 @@ func (s *Service) checkExpiry(ctx context.Context) *types.Error {
 		log.Debug().
 			Str("staking_tx", delegation.StakingTxHashHex).
 			Str("current_state", delegation.State.String()).
+			Str("new_sub_state", tlDoc.DelegationSubState.String()).
+			Str("expire_height", strconv.FormatUint(uint64(tlDoc.ExpireHeight), 10)).
 			Msg("checking if delegation is expired")
 
 		// Check if the delegation is in a qualified state to transition to Withdrawable