diff --git a/.gitignore b/.gitignore index 2b5ef8a6..49db1d6e 100644 --- a/.gitignore +++ b/.gitignore @@ -28,3 +28,6 @@ node_modules /release .release_version chart_releases +/snapshots/**/*.sql +*.sql +*.dump diff --git a/pkg/eigenState/avsOperators/avsOperators_test.go b/pkg/eigenState/avsOperators/avsOperators_test.go index f440d9b2..53d624ec 100644 --- a/pkg/eigenState/avsOperators/avsOperators_test.go +++ b/pkg/eigenState/avsOperators/avsOperators_test.go @@ -127,7 +127,7 @@ func Test_AvsOperatorState(t *testing.T) { Scan(&states) if statesRes.Error != nil { - t.Fatalf("Failed to fetch registered_avs_operators: %v", statesRes.Error) + t.Fatalf("Failed to fetch avs_operator_state_changes: %v", statesRes.Error) } assert.Equal(t, 1, len(states)) diff --git a/pkg/eigenState/stakerDelegations/stakerDelegations_test.go b/pkg/eigenState/stakerDelegations/stakerDelegations_test.go index bab05c6d..c1593c68 100644 --- a/pkg/eigenState/stakerDelegations/stakerDelegations_test.go +++ b/pkg/eigenState/stakerDelegations/stakerDelegations_test.go @@ -40,7 +40,6 @@ func setup() ( func teardown(model *StakerDelegationsModel) { model.DB.Exec("truncate table staker_delegation_changes cascade") - model.DB.Exec("truncate table delegated_stakers cascade") model.DB.Exec("truncate table staker_delegation_changes cascade") } @@ -125,7 +124,7 @@ func Test_DelegatedStakersState(t *testing.T) { Scan(&states) if statesRes.Error != nil { - t.Fatalf("Failed to fetch delegated_stakers: %v", statesRes.Error) + t.Fatalf("Failed to fetch staker_delegation_changes: %v", statesRes.Error) } assert.Equal(t, 1, len(states)) diff --git a/pkg/postgres/migrations/202411191947_cleanupUnusedTables/up.go b/pkg/postgres/migrations/202411191947_cleanupUnusedTables/up.go new file mode 100644 index 00000000..0f1abd22 --- /dev/null +++ b/pkg/postgres/migrations/202411191947_cleanupUnusedTables/up.go @@ -0,0 +1,27 @@ +package _202411191947_cleanupUnusedTables + +import ( + "database/sql" + "gorm.io/gorm" +) + +type Migration struct { +} + +func (m *Migration) Up(db *sql.DB, grm *gorm.DB) error { + queries := []string{ + `drop table delegated_stakers`, + `drop table registered_avs_operators`, + } + for _, query := range queries { + _, err := db.Exec(query) + if err != nil { + return err + } + } + return nil +} + +func (m *Migration) GetName() string { + return "202411191947_cleanupUnusedTables" +} diff --git a/pkg/postgres/migrations/migrator.go b/pkg/postgres/migrations/migrator.go index d80671ee..da3bfbc1 100644 --- a/pkg/postgres/migrations/migrator.go +++ b/pkg/postgres/migrations/migrator.go @@ -32,6 +32,7 @@ import ( _202411120947_disabledDistributionRoots "github.com/Layr-Labs/sidecar/pkg/postgres/migrations/202411120947_disabledDistributionRoots" _202411130953_addHashColumns "github.com/Layr-Labs/sidecar/pkg/postgres/migrations/202411130953_addHashColumns" _202411131200_eigenStateModelConstraints "github.com/Layr-Labs/sidecar/pkg/postgres/migrations/202411131200_eigenStateModelConstraints" + _202411191947_cleanupUnusedTables "github.com/Layr-Labs/sidecar/pkg/postgres/migrations/202411191947_cleanupUnusedTables" "go.uber.org/zap" "gorm.io/gorm" "time" @@ -102,6 +103,7 @@ func (m *Migrator) MigrateAll() error { &_202411120947_disabledDistributionRoots.Migration{}, &_202411130953_addHashColumns.Migration{}, &_202411131200_eigenStateModelConstraints.Migration{}, + &_202411191947_cleanupUnusedTables.Migration{}, } for _, migration := range migrations {