refactor: migrate env-var reads to config package
This commit is contained in:
@@ -1,16 +1,19 @@
|
||||
package server
|
||||
|
||||
import (
|
||||
"mal/internal/config"
|
||||
"net/http"
|
||||
"os"
|
||||
"strings"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
)
|
||||
|
||||
func CORSMiddleware() gin.HandlerFunc {
|
||||
allowAll := os.Getenv("MAL_CORS_ALLOW_ALL") == "1"
|
||||
return CORSMiddlewareWithConfig(config.Config{})
|
||||
}
|
||||
|
||||
func CORSMiddlewareWithConfig(cfg config.Config) gin.HandlerFunc {
|
||||
allowAll := cfg.CORSAllowAll
|
||||
return func(c *gin.Context) {
|
||||
origin := c.GetHeader("Origin")
|
||||
if origin != "" && (allowAll || isAllowedOrigin(origin)) {
|
||||
|
||||
@@ -3,9 +3,9 @@ package server
|
||||
import (
|
||||
"context"
|
||||
"log"
|
||||
"mal/internal/config"
|
||||
"mal/internal/observability"
|
||||
"net/http"
|
||||
"os"
|
||||
"time"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
@@ -19,12 +19,14 @@ var Module = fx.Options(
|
||||
fx.Invoke(RunServer),
|
||||
)
|
||||
|
||||
func ProvideRouter(htmlRender render.HTMLRender, metrics *observability.Metrics) *gin.Engine {
|
||||
if os.Getenv("GIN_MODE") == "" {
|
||||
func ProvideRouter(cfg config.Config, htmlRender render.HTMLRender, metrics *observability.Metrics) *gin.Engine {
|
||||
if cfg.GinMode == "" {
|
||||
gin.SetMode(gin.ReleaseMode)
|
||||
} else {
|
||||
gin.SetMode(cfg.GinMode)
|
||||
}
|
||||
r := gin.New()
|
||||
r.Use(CORSMiddleware(), RequestLogger(metrics), gin.Recovery())
|
||||
r.Use(CORSMiddlewareWithConfig(cfg), RequestLogger(metrics), gin.Recovery())
|
||||
r.Static("/static", "./static")
|
||||
r.Static("/dist", "./dist")
|
||||
r.GET("/metrics", gin.WrapH(metrics.Handler()))
|
||||
@@ -32,11 +34,8 @@ func ProvideRouter(htmlRender render.HTMLRender, metrics *observability.Metrics)
|
||||
return r
|
||||
}
|
||||
|
||||
func RunServer(lifecycle fx.Lifecycle, r *gin.Engine) {
|
||||
port := os.Getenv("PORT")
|
||||
if port == "" {
|
||||
port = "3000"
|
||||
}
|
||||
func RunServer(cfg config.Config, lifecycle fx.Lifecycle, r *gin.Engine) {
|
||||
port := cfg.Port
|
||||
|
||||
srv := newHTTPServer(":"+port, r)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user