pkg/adapters/fiber/middleware.go

package fiber

import (
	"net/http"

	sentinel "github.com/alibaba/sentinel-golang/api"
	"github.com/alibaba/sentinel-golang/core/base"
	"github.com/gofiber/fiber/v2"
)

// SentinelMiddleware returns new gin.HandlerFunc
// Default resource name is {method}:{path}, such as "GET:/api/users/:id"
// Default block fallback is returning 429 code
// Define your own behavior by setting options
func SentinelMiddleware(opts ...Option) fiber.Handler {
	options := evaluateOptions(opts)
	return func(ctx *fiber.Ctx) error {
		resourceName := ctx.Route().Method + ":" + string(ctx.Context().Path())

		if options.resourceExtract != nil {
			resourceName = options.resourceExtract(ctx)
		}

		entry, entryErr := sentinel.Entry(
			resourceName,
			sentinel.WithResourceType(base.ResTypeWeb),
			sentinel.WithTrafficType(base.Inbound),
		)

		if entryErr != nil {
			if options.blockFallback != nil {
				return options.blockFallback(ctx)
			} else {
				return ctx.SendStatus(http.StatusTooManyRequests)
			}
		}

		defer entry.Exit()
		return ctx.Next()
	}
}