62 lines
1.2 KiB
Go
62 lines
1.2 KiB
Go
|
package router
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"errors"
|
||
|
|
"github.com/gin-gonic/gin"
|
||
|
|
log "github.com/sirupsen/logrus"
|
||
|
|
ginprometheus "github.com/zsais/go-gin-prometheus"
|
||
|
|
"net/http"
|
||
|
|
)
|
||
|
|
|
||
|
|
type Handler struct {
|
||
|
|
srv *http.Server
|
||
|
|
}
|
||
|
|
|
||
|
|
type Deps struct {
|
||
|
|
Addr string
|
||
|
|
GinMode string
|
||
|
|
}
|
||
|
|
|
||
|
|
const pkgLogHeader string = "Router |"
|
||
|
|
|
||
|
|
func NewHandler(deps Deps) *Handler {
|
||
|
|
engine := gin.Default()
|
||
|
|
|
||
|
|
if deps.GinMode == "release" {
|
||
|
|
gin.SetMode(gin.ReleaseMode)
|
||
|
|
err := engine.SetTrustedProxies([]string{"172.20.0.0/16"})
|
||
|
|
if err != nil {
|
||
|
|
log.WithError(err).Errorf("%v Set proxies failed", pkgLogHeader)
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
engine.GET("/", func(c *gin.Context) { c.JSON(200, gin.H{"msg": "v2"}) })
|
||
|
|
|
||
|
|
p := ginprometheus.NewPrometheus("gin")
|
||
|
|
p.Use(engine)
|
||
|
|
|
||
|
|
srv := http.Server{
|
||
|
|
Addr: deps.Addr,
|
||
|
|
Handler: engine,
|
||
|
|
}
|
||
|
|
|
||
|
|
return &Handler{
|
||
|
|
srv: &srv,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (h *Handler) Run() error {
|
||
|
|
log.Infof("Starting server on %s", h.srv.Addr)
|
||
|
|
if err := h.srv.ListenAndServe(); err != nil && !errors.Is(err, http.ErrServerClosed) {
|
||
|
|
log.WithError(err).Errorf("%v ListenAndServe failed", pkgLogHeader)
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (h *Handler) Shutdown(ctx context.Context) error {
|
||
|
|
return h.srv.Shutdown(ctx)
|
||
|
|
}
|