feat(db): create interface for valid tables

Author: traviolus

informative-indexer/db/db.go

@@ -140,16 +140,7 @@ func GetRowsToPruneByBlockHeight(ctx context.Context, dbClient Queryable, table
 	}
 
 	var query string
-	var t interface{}
-
-	if table == "transaction_events" {
-		t = TransactionEvent{}
-	} else if table == "finalize_block_events" {
-		t = FinalizeBlockEvent{}
-	} else if table == "move_events" {
-		t = MoveEvent{}
-	}
-
+	t := ValidTablesMap[table]
 	columns := getColumns(t)
 
 	query = fmt.Sprintf("SELECT %s FROM %s WHERE block_height <= $1", strings.Join(columns, ", "), table)

informative-indexer/db/valid_tables.go

@@ -1,13 +1,19 @@
 package db
 
-var validTableNames = []string{
-	"transaction_events",
-	"finalize_block_events",
-	"move_events",
+import "github.com/jackc/pgx/v5"
+
+type ValidTable interface {
+	Unmarshal(pgx.Rows) (map[string]interface{}, error)
+}
+
+var ValidTablesMap = map[string]ValidTable{
+	"transaction_events":    &TransactionEvent{},
+	"finalize_block_events": &FinalizeBlockEvent{},
+	"move_events":           &MoveEvent{},
 }
 
 func isValidTableName(tableName string) bool {
-	for _, validTable := range validTableNames {
+	for validTable, _ := range ValidTablesMap {
 		if tableName == validTable {
 			return true
 		}
@@ -16,5 +22,9 @@ func isValidTableName(tableName string) bool {
 }
 
 func GetValidTableNames() []string {
-	return validTableNames
+	var keys []string
+	for key := range ValidTablesMap {
+		keys = append(keys, key)
+	}
+	return keys
 }

informative-indexer/prunner/prunner.go

@@ -127,25 +127,13 @@ func fetchRowsToPrune(ctx context.Context, dbClient db.Queryable, tableName stri
 
 	var result []interface{}
 	for rows.Next() {
-		var row map[string]interface{}
-		if tableName == "transaction_events" {
-			transactionEvent := db.TransactionEvent{}
-			row, err = transactionEvent.Unmarshal(rows)
-			if err != nil {
-				return nil, err
-			}
-		} else if tableName == "finalize_block_events" {
-			blockResult := db.FinalizeBlockEvent{}
-			row, err = blockResult.Unmarshal(rows)
-			if err != nil {
-				return nil, err
-			}
-		} else if tableName == "move_events" {
-			eventResult := db.MoveEvent{}
-			row, err = eventResult.Unmarshal(rows)
-			if err != nil {
-				return nil, err
-			}
+		table, ok := db.ValidTablesMap[tableName]
+		if !ok {
+			return nil, fmt.Errorf("failed to get table interface from table name %s", tableName)
+		}
+		row, err := table.Unmarshal(rows)
+		if err != nil {
+			return nil, err
 		}
 		result = append(result, row)
 	}