speatzle/morffix
1
0
Fork 0
Code Issues Pull requests Projects Releases 2 Packages Wiki Activity Actions

add database reset flag

Browse source
This commit is contained in:
Samuel Lorch 2024-05-11 00:38:40 +02:00
parent bf34abf5f8
commit 3aeda54404
2 changed files with 18 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
main.go
View file

@ -25,7 +25,9 @@ var migrations embed.FS
var conf config.Config
func main() {
isserver := flag.Bool("server", false, "Run as Server")
isServer := flag.Bool("server", false, "Run as Server")
resetDB := flag.Bool("reset-db", false, "Reset DB")
flag.Parse()
confPath := "config.toml"
_, err := os.Stat(confPath)
@ -48,9 +50,9 @@ func main() {
conf.Worker.TempDir = "/tmp"
}
if *isserver {
if *isServer {
slog.Info("Starting Server...")
server.Start(conf, templates, static, migrations)
server.Start(conf, templates, static, migrations, *resetDB)
} else {
slog.Info("Starting Worker...")
worker.Start(conf)

17
server/server.go
View file

@ -33,7 +33,7 @@ func sub(a, b int) int {
var db *pgxpool.Pool
func Start(_conf config.Config, tmplFS embed.FS, staticFS embed.FS, migrationsFS embed.FS) {
func Start(_conf config.Config, tmplFS embed.FS, staticFS embed.FS, migrationsFS embed.FS, resetDB bool) {
conf = _conf
// Static Files
@ -70,6 +70,15 @@ func Start(_conf config.Config, tmplFS embed.FS, staticFS embed.FS, migrationsFS
return
}
if resetDB {
slog.Info("Running Down Migrations...")
err = m.Down()
if err != nil {
slog.Error("Unable to Migrate Down", "err", err)
return
}
}
//err = m.Down()
err = m.Up()
if err == migrate.ErrNoChange {
@ -95,10 +104,10 @@ func Start(_conf config.Config, tmplFS embed.FS, staticFS embed.FS, migrationsFS
server := &http.Server{
Addr: conf.Server.Address,
Handler: mux,
ReadTimeout: 1 * time.Second,
WriteTimeout: 1 * time.Second,
ReadTimeout: 10 * time.Second,
WriteTimeout: 10 * time.Second,
IdleTimeout: 30 * time.Second,
ReadHeaderTimeout: 2 * time.Second,
ReadHeaderTimeout: 10 * time.Second,
}
serverClose := make(chan bool)