72 lines
1.4 KiB
Go
72 lines
1.4 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,
|
|
}))
|
|
|
|
// metric
|
|
app.Use(otelfiber.Middleware())
|
|
|
|
// logger
|
|
app.Use(logger.New(logger.Config{
|
|
Next: func(c *fiber.Ctx) bool {
|
|
return c.Path() == "/favicon.ico"
|
|
},
|
|
}))
|
|
|
|
// 补充 otel span attr
|
|
app.Use(func(c *fiber.Ctx) error {
|
|
err := c.Next()
|
|
|
|
span := trace.SpanFromContext(c.UserContext())
|
|
if !span.IsRecording() {
|
|
return err
|
|
}
|
|
|
|
str := ""
|
|
if err != nil {
|
|
str = err.Error()
|
|
}
|
|
|
|
span.SetAttributes(attribute.String("http.response.error", str))
|
|
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())
|
|
}
|