pkg/adapters/go-zero/option.go

package go_zero

import "net/http"

type (
	Option  func(*options)
	options struct {
		resourceExtract func(r *http.Request) string
		blockFallback   func(r *http.Request) (int, string)
	}
)

func evaluateOptions(opts []Option) *options {
	optCopy := &options{}
	for _, opt := range opts {
		opt(optCopy)
	}

	return optCopy
}

// WithResourceExtractor sets the resource extractor of the web requests.
func WithResourceExtractor(fn func(r *http.Request) string) Option {
	return func(opts *options) {
		opts.resourceExtract = fn
	}
}

// WithBlockFallback sets the fallback handler when requests are blocked.
func WithBlockFallback(fn func(r *http.Request) (int, string)) Option {
	return func(opts *options) {
		opts.blockFallback = fn
	}
}