Feat: optional reset
All checks were successful
All checks were successful
This commit is contained in:
@@ -4,14 +4,24 @@ import (
|
||||
"context"
|
||||
|
||||
"gitea.konchin.com/go2025/backend/models"
|
||||
"github.com/spf13/viper"
|
||||
"github.com/uptrace/bun"
|
||||
)
|
||||
|
||||
func InitDB(ctx context.Context, db *bun.DB) error {
|
||||
return db.ResetModel(ctx,
|
||||
db.RegisterModel(
|
||||
(*models.AliasImage)(nil),
|
||||
(*models.Alias)(nil),
|
||||
(*models.Image)(nil),
|
||||
(*models.Session)(nil),
|
||||
)
|
||||
if viper.GetBool("reset") {
|
||||
return db.ResetModel(ctx,
|
||||
(*models.AliasImage)(nil),
|
||||
(*models.Alias)(nil),
|
||||
(*models.Image)(nil),
|
||||
(*models.Session)(nil),
|
||||
)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user