Trace back everything (tested)
Signed-off-by: Asai Neko <sugar@sne.moe>
This commit is contained in:
@@ -1,7 +1,9 @@
|
||||
package server
|
||||
|
||||
import (
|
||||
"context"
|
||||
"log/slog"
|
||||
"net"
|
||||
"net/http"
|
||||
"nixcn-cms/middleware"
|
||||
"time"
|
||||
@@ -11,14 +13,16 @@ import (
|
||||
"go.opentelemetry.io/contrib/instrumentation/github.com/gin-gonic/gin/otelgin"
|
||||
)
|
||||
|
||||
func Start() {
|
||||
func Start(ctx context.Context) {
|
||||
if !viper.GetBool("server.debug_mode") {
|
||||
gin.SetMode(gin.ReleaseMode)
|
||||
gin.DisableConsoleColor()
|
||||
}
|
||||
|
||||
r := gin.New()
|
||||
r.Use(gin.Recovery(), middleware.GinLogger(), otelgin.Middleware("nixcn-cms-backend"))
|
||||
r.Use(otelgin.Middleware(viper.GetString("server.service_name")))
|
||||
r.Use(middleware.GinLogger())
|
||||
r.Use(gin.Recovery())
|
||||
|
||||
Router(r)
|
||||
|
||||
@@ -26,13 +30,14 @@ func Start() {
|
||||
server := &http.Server{
|
||||
Addr: viper.GetString("server.address"),
|
||||
Handler: r,
|
||||
BaseContext: func(net.Listener) context.Context { return ctx },
|
||||
ReadTimeout: 10 * time.Second,
|
||||
WriteTimeout: 10 * time.Second,
|
||||
MaxHeaderBytes: 1 << 20,
|
||||
}
|
||||
|
||||
slog.Info("[Server] Starting server on " + viper.GetString("server.address"))
|
||||
slog.InfoContext(ctx, "[Server] Starting server on "+viper.GetString("server.address"))
|
||||
if err := server.ListenAndServe(); err != nil {
|
||||
slog.Error("[Server] Error starting server!", "err", err)
|
||||
slog.ErrorContext(ctx, "[Server] Error starting server!", "err", err)
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user