migration

pkg/postgres/migrations/202411191708_operatorPISplits/up.go

@@ -0,0 +1,33 @@
+package _202411191708_operatorPISplits
+
+import (
+	"database/sql"
+
+	"gorm.io/gorm"
+)
+
+type Migration struct {
+}
+
+func (m *Migration) Up(db *sql.DB, grm *gorm.DB) error {
+	query := `
+		create table if not exists operator_pi_splits (
+			operator varchar not null,
+			activated_at timestamp(6) not null,
+			old_operator_avs_split_bips integer not null,
+			new_operator_avs_split_bips integer not null,
+			block_number bigint not null,
+			transaction_hash varchar not null,
+			log_index bigint not null,
+			unique(transaction_hash, log_index, block_number)
+		);
+	`
+	if err := grm.Exec(query).Error; err != nil {
+		return err
+	}
+	return nil
+}
+
+func (m *Migration) GetName() string {
+	return "202411191708_operatorPISplits"
+}

pkg/postgres/migrations/202411191710_blockNumberFkConstraint/up.go

@@ -0,0 +1,32 @@
+package _202411191710_blockNumberFkConstraint
+
+import (
+	"database/sql"
+	"fmt"
+
+	"gorm.io/gorm"
+)
+
+type Migration struct {
+}
+
+func (m *Migration) Up(db *sql.DB, grm *gorm.DB) error {
+	queries := []string{
+		`alter table operator_directed_reward_submissions add constraint operator_directed_reward_submissions_block_number_fkey foreign key (block_number) references blocks (number) on delete cascade`,
+		`alter table operator_avs_splits add constraint operator_avs_splits_block_number_fkey foreign key (block_number) references blocks (number) on delete cascade`,
+		`alter table operator_pi_splits add constraint operator_pi_splits_block_number_fkey foreign key (block_number) references blocks (number) on delete cascade`,
+	}
+
+	for _, query := range queries {
+		_, err := db.Exec(query)
+		if err != nil {
+			fmt.Printf("Failed to run migration query: %s - %+v\n", query, err)
+			return err
+		}
+	}
+	return nil
+}
+
+func (m *Migration) GetName() string {
+	return "202411191710_blockNumberFkConstraint"
+}

pkg/postgres/migrations/migrator.go

@@ -36,6 +36,8 @@ import (
 	_202411131200_eigenStateModelConstraints "github.com/Layr-Labs/sidecar/pkg/postgres/migrations/202411131200_eigenStateModelConstraints"
 	_202411151931_operatorDirectedRewardSubmissions "github.com/Layr-Labs/sidecar/pkg/postgres/migrations/202411151931_operatorDirectedRewardSubmissions"
 	_202411191550_operatorAVSSplits "github.com/Layr-Labs/sidecar/pkg/postgres/migrations/202411191550_operatorAVSSplits"
+	_202411191708_operatorPISplits "github.com/Layr-Labs/sidecar/pkg/postgres/migrations/202411191708_operatorPISplits"
+	_202411191710_blockNumberFkConstraint "github.com/Layr-Labs/sidecar/pkg/postgres/migrations/202411191710_blockNumberFkConstraint"
 	"go.uber.org/zap"
 	"gorm.io/gorm"
 )
@@ -107,6 +109,8 @@ func (m *Migrator) MigrateAll() error {
 		&_202411131200_eigenStateModelConstraints.Migration{},
 		&_202411151931_operatorDirectedRewardSubmissions.Migration{},
 		&_202411191550_operatorAVSSplits.Migration{},
+		&_202411191708_operatorPISplits.Migration{},
+		&_202411191710_blockNumberFkConstraint.Migration{},
 	}
 
 	for _, migration := range migrations {