bouncer/internal/store/proxy_repository.go

92 lines
1.9 KiB
Go
Raw Normal View History

2023-04-24 20:52:12 +02:00
package store
import (
"context"
"net/url"
)
type ProxyRepository interface {
CreateProxy(ctx context.Context, name ProxyName, to string, from ...string) (*Proxy, error)
UpdateProxy(ctx context.Context, name ProxyName, funcs ...UpdateProxyOptionFunc) (*Proxy, error)
QueryProxy(ctx context.Context, funcs ...QueryProxyOptionFunc) ([]*ProxyHeader, error)
GetProxy(ctx context.Context, name ProxyName) (*Proxy, error)
DeleteProxy(ctx context.Context, name ProxyName) error
}
type UpdateProxyOptionFunc func(*UpdateProxyOptions)
type UpdateProxyOptions struct {
To *string
From []string
Enabled *bool
Weight *int
}
func WithProxyUpdateEnabled(enabled bool) UpdateProxyOptionFunc {
return func(o *UpdateProxyOptions) {
o.Enabled = &enabled
}
}
func WithProxyUpdateWeight(weight int) UpdateProxyOptionFunc {
return func(o *UpdateProxyOptions) {
o.Weight = &weight
}
}
func WithProxyUpdateTo(to string) UpdateProxyOptionFunc {
return func(o *UpdateProxyOptions) {
o.To = &to
}
}
func WithProxyUpdateFrom(from ...string) UpdateProxyOptionFunc {
return func(o *UpdateProxyOptions) {
o.From = from
}
}
type QueryProxyOptionFunc func(*QueryProxyOptions)
type QueryProxyOptions struct {
To *url.URL
Names []ProxyName
Enabled *bool
From []string
}
func DefaultQueryProxyOptions() *QueryProxyOptions {
funcs := []QueryProxyOptionFunc{}
opts := &QueryProxyOptions{}
for _, fn := range funcs {
fn(opts)
}
return opts
}
func WithProxyQueryTo(to *url.URL) QueryProxyOptionFunc {
return func(o *QueryProxyOptions) {
o.To = to
}
}
func WithProxyQueryFrom(from ...string) QueryProxyOptionFunc {
return func(o *QueryProxyOptions) {
o.From = from
}
}
func WithProxyQueryNames(names ...ProxyName) QueryProxyOptionFunc {
return func(o *QueryProxyOptions) {
o.Names = names
}
}
func WithProxyQueryEnabled(enabled bool) QueryProxyOptionFunc {
return func(o *QueryProxyOptions) {
o.Enabled = &enabled
}
}