74 lines
1.5 KiB
Go
74 lines
1.5 KiB
Go
package web
|
|
|
|
import (
|
|
"platform/web/auth"
|
|
|
|
"github.com/gofiber/contrib/otelfiber/v2"
|
|
"github.com/gofiber/fiber/v2"
|
|
"github.com/gofiber/fiber/v2/middleware/cors"
|
|
"github.com/gofiber/fiber/v2/middleware/logger"
|
|
"github.com/gofiber/fiber/v2/middleware/recover"
|
|
"github.com/gofiber/fiber/v2/middleware/requestid"
|
|
"github.com/google/uuid"
|
|
"github.com/jxskiss/base62"
|
|
"go.opentelemetry.io/otel/attribute"
|
|
"go.opentelemetry.io/otel/trace"
|
|
)
|
|
|
|
func ApplyMiddlewares(app *fiber.App) {
|
|
|
|
// recover
|
|
app.Use(recover.New(recover.Config{
|
|
EnableStackTrace: true,
|
|
}))
|
|
|
|
// logger
|
|
app.Use(logger.New(logger.Config{
|
|
Next: func(c *fiber.Ctx) bool {
|
|
return c.Path() == "/favicon.ico"
|
|
},
|
|
}))
|
|
|
|
// metric
|
|
app.Use(otelfiber.Middleware())
|
|
app.Use(func(c *fiber.Ctx) error {
|
|
err := c.Next()
|
|
|
|
span := trace.SpanFromContext(c.UserContext())
|
|
if !span.IsRecording() {
|
|
return err
|
|
}
|
|
|
|
status := c.Response().StatusCode()
|
|
body := []byte{}
|
|
if status < 200 || status >= 300 {
|
|
body = c.Response().Body()
|
|
if len(body) > 1024 {
|
|
body = body[:1024]
|
|
}
|
|
}
|
|
|
|
span.SetAttributes(attribute.String("http.response.error", string(body)))
|
|
return err
|
|
})
|
|
|
|
// cors
|
|
app.Use(cors.New(cors.Config{
|
|
AllowCredentials: true,
|
|
AllowOriginsFunc: func(origin string) bool {
|
|
return true
|
|
},
|
|
}))
|
|
|
|
// request id
|
|
app.Use(requestid.New(requestid.Config{
|
|
Generator: func() string {
|
|
binary, _ := uuid.New().MarshalBinary()
|
|
return base62.EncodeToString(binary)
|
|
},
|
|
}))
|
|
|
|
// authenticate
|
|
app.Use(auth.Authenticate())
|
|
}
|