Add foreign key to requests table

Signed-off-by: AKP <tom@tdpain.net>
This commit is contained in:
akp 2023-04-03 13:17:16 +01:00
parent a001cdf6d8
commit 144696c9a7
No known key found for this signature in database
GPG key ID: AA5726202C8879B7

View file

@ -10,28 +10,25 @@ import (
func init() {
logger := log.With().Str("migration", "20230331164907").Logger()
tables := []any{
&models.Request{},
&models.Session{},
}
Migrations.MustRegister(func(ctx context.Context, db *bun.DB) error {
logger.Info().Msg("up")
for _, table := range tables {
if _, err := db.NewCreateTable().Model(table).Exec(ctx); err != nil {
return err
}
if _, err := db.NewCreateTable().Model(&models.Session{}).Exec(ctx); err != nil {
return err
}
if _, err := db.NewCreateTable().Model(&models.Request{}).ForeignKey("(session_id) REFERENCES sessions(id)").Exec(ctx); err != nil {
return err
}
return nil
}, func(ctx context.Context, db *bun.DB) error {
logger.Info().Msg("down")
for _, table := range tables {
if _, err := db.NewDropTable().Model(table).Exec(ctx); err != nil {
return err
}
if _, err := db.NewDropTable().Model(&models.Request{}).Exec(ctx); err != nil {
return err
}
if _, err := db.NewDropTable().Model(&models.Session{}).Exec(ctx); err != nil {
return err
}
return nil