From fbd799c8ed9c065bf4b8872f21d2634bedd9868a Mon Sep 17 00:00:00 2001 From: James Tombleson Date: Fri, 5 Apr 2024 17:53:28 -0700 Subject: [PATCH] Added migrations on startup --- main.go | 26 +++++++++++++++++++++----- 1 file changed, 21 insertions(+), 5 deletions(-) diff --git a/main.go b/main.go index 411f3f4..63fb65d 100644 --- a/main.go +++ b/main.go @@ -2,15 +2,17 @@ package main import ( "database/sql" - v1 "go-cook/api/handlers/v1" - "go-cook/api/services" "log" "net/http" + "git.jamestombleson.com/jtom38/go-cook/api/services" + v1 "git.jamestombleson.com/jtom38/go-cook/api/handlers/v1" + _ "github.com/glebarez/go-sqlite" "github.com/go-playground/validator" "github.com/labstack/echo/v4" "github.com/labstack/echo/v4/middleware" + "github.com/pressly/goose/v3" ) // @title Swagger Example API @@ -24,6 +26,20 @@ func main() { cfg := services.NewEnvConfig() + // Check if we should apply migrations on startup + // You can disable this in the ENV configuration of your application. + if !cfg.DisableMigrationsOnStartUp { + err = goose.SetDialect("sqlite3") + if err != nil { + panic(err) + } + + err = goose.Up(db, "api/migrations") + if err != nil { + panic(err) + } + } + e := echo.New() e.Validator = &CustomValidator{ validator: validator.New(), @@ -45,8 +61,8 @@ type CustomValidator struct { func (cv *CustomValidator) Validate(i interface{}) error { if err := cv.validator.Struct(i); err != nil { - // Optionally, you could return the error to give each route more control over the status code - return echo.NewHTTPError(http.StatusBadRequest, err.Error()) + // Optionally, you could return the error to give each route more control over the status code + return echo.NewHTTPError(http.StatusBadRequest, err.Error()) } return nil - } +}