72 lines
1.6 KiB
Go
72 lines
1.6 KiB
Go
package director
|
|
|
|
import (
|
|
"net/http"
|
|
"time"
|
|
|
|
"forge.cadoles.com/cadoles/bouncer/internal/cache"
|
|
"forge.cadoles.com/cadoles/bouncer/internal/cache/memory"
|
|
"forge.cadoles.com/cadoles/bouncer/internal/cache/ttl"
|
|
"forge.cadoles.com/cadoles/bouncer/internal/store"
|
|
"gitlab.com/wpetit/goweb/logger"
|
|
)
|
|
|
|
type Options struct {
|
|
Layers []Layer
|
|
ProxyCache cache.Cache[string, []*store.Proxy]
|
|
LayerCache cache.Cache[string, []*store.Layer]
|
|
HandleError HandleErrorFunc
|
|
}
|
|
|
|
type OptionFunc func(opts *Options)
|
|
|
|
func NewOptions(funcs ...OptionFunc) *Options {
|
|
opts := &Options{
|
|
Layers: make([]Layer, 0),
|
|
ProxyCache: ttl.NewCache(
|
|
memory.NewCache[string, []*store.Proxy](),
|
|
memory.NewCache[string, time.Time](),
|
|
30*time.Second,
|
|
),
|
|
LayerCache: ttl.NewCache(
|
|
memory.NewCache[string, []*store.Layer](),
|
|
memory.NewCache[string, time.Time](),
|
|
30*time.Second,
|
|
),
|
|
HandleError: func(w http.ResponseWriter, r *http.Request, status int, err error) {
|
|
logger.Error(r.Context(), err.Error(), logger.CapturedE(err))
|
|
http.Error(w, http.StatusText(status), status)
|
|
},
|
|
}
|
|
|
|
for _, fn := range funcs {
|
|
fn(opts)
|
|
}
|
|
|
|
return opts
|
|
}
|
|
|
|
func WithLayers(layers ...Layer) OptionFunc {
|
|
return func(opts *Options) {
|
|
opts.Layers = layers
|
|
}
|
|
}
|
|
|
|
func WithProxyCache(cache cache.Cache[string, []*store.Proxy]) OptionFunc {
|
|
return func(opts *Options) {
|
|
opts.ProxyCache = cache
|
|
}
|
|
}
|
|
|
|
func WithLayerCache(cache cache.Cache[string, []*store.Layer]) OptionFunc {
|
|
return func(opts *Options) {
|
|
opts.LayerCache = cache
|
|
}
|
|
}
|
|
|
|
func WithHandleErrorFunc(fn HandleErrorFunc) OptionFunc {
|
|
return func(opts *Options) {
|
|
opts.HandleError = fn
|
|
}
|
|
}
|