Go2025/backend
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Feat: optional reset
All checks were successful
Go test / run-go-vet (push) Successful in 7s
Details
Go test / check-swagger-up-to-date (push) Successful in 10s
Details
Go test / run-go-test (push) Successful in 30s
Details
Go test / cleanup-go-test (push) Successful in 4s
Details
Go test / release-image (push) Successful in 3m36s
Details

Browse Source
This commit is contained in:
Yi-Ting Shih
2025-12-12 22:38:36 +08:00
parent e537d83c19
commit 44a3bf64c1
2 changed files with 13 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
cmds/serve.go
View File

@@ -145,6 +145,8 @@ func init() {
String("cors-origin", "", "CORS origin") String("cors-origin", "", "CORS origin")
serveCmd.Flags(). serveCmd.Flags().
String("preshared-key", "poop", "Preshared key for Discord Bot") String("preshared-key", "poop", "Preshared key for Discord Bot")
serveCmd.Flags().
Bool("reset", false, "Reset database")
serveCmd.Flags(). serveCmd.Flags().
Int64("access-token-timeout", 300, "Timeout of Access Token JWT") Int64("access-token-timeout", 300, "Timeout of Access Token JWT")

10
utils/initDB.go
View File

@@ -4,10 +4,18 @@ import (
"context" "context"
"gitea.konchin.com/go2025/backend/models" "gitea.konchin.com/go2025/backend/models"
"github.com/spf13/viper"
"github.com/uptrace/bun" "github.com/uptrace/bun"
) )
func InitDB(ctx context.Context, db *bun.DB) error { func InitDB(ctx context.Context, db *bun.DB) error {
db.RegisterModel(
(*models.AliasImage)(nil),
(*models.Alias)(nil),
(*models.Image)(nil),
(*models.Session)(nil),
)
if viper.GetBool("reset") {
return db.ResetModel(ctx, return db.ResetModel(ctx,
(*models.AliasImage)(nil), (*models.AliasImage)(nil),
(*models.Alias)(nil), (*models.Alias)(nil),
@@ -15,3 +23,5 @@ func InitDB(ctx context.Context, db *bun.DB) error {
(*models.Session)(nil), (*models.Session)(nil),
) )
} }
return nil
}