feat: initial commit
All checks were successful
Cadoles/bouncer/pipeline/head This commit looks good
All checks were successful
Cadoles/bouncer/pipeline/head This commit looks good
This commit is contained in:
75
internal/client/query_layer.go
Normal file
75
internal/client/query_layer.go
Normal file
@ -0,0 +1,75 @@
|
||||
package client
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"net/url"
|
||||
|
||||
"forge.cadoles.com/cadoles/bouncer/internal/admin"
|
||||
"forge.cadoles.com/cadoles/bouncer/internal/store"
|
||||
"github.com/pkg/errors"
|
||||
)
|
||||
|
||||
type QueryLayerOptionFunc func(*QueryLayerOptions)
|
||||
|
||||
type QueryLayerOptions struct {
|
||||
Options []OptionFunc
|
||||
Offset *int
|
||||
Limit *int
|
||||
Names []store.LayerName
|
||||
}
|
||||
|
||||
func WithQueryLayerOptions(funcs ...OptionFunc) QueryLayerOptionFunc {
|
||||
return func(opts *QueryLayerOptions) {
|
||||
opts.Options = funcs
|
||||
}
|
||||
}
|
||||
|
||||
func WithQueryLayerLimit(limit int) QueryLayerOptionFunc {
|
||||
return func(opts *QueryLayerOptions) {
|
||||
opts.Limit = &limit
|
||||
}
|
||||
}
|
||||
|
||||
func WithQueryLayerOffset(offset int) QueryLayerOptionFunc {
|
||||
return func(opts *QueryLayerOptions) {
|
||||
opts.Offset = &offset
|
||||
}
|
||||
}
|
||||
|
||||
func WithQueryLayerNames(names ...store.LayerName) QueryLayerOptionFunc {
|
||||
return func(opts *QueryLayerOptions) {
|
||||
opts.Names = names
|
||||
}
|
||||
}
|
||||
|
||||
func (c *Client) QueryLayer(ctx context.Context, proxyName store.ProxyName, funcs ...QueryLayerOptionFunc) ([]*store.LayerHeader, error) {
|
||||
options := &QueryLayerOptions{}
|
||||
for _, fn := range funcs {
|
||||
fn(options)
|
||||
}
|
||||
|
||||
query := url.Values{}
|
||||
|
||||
if options.Names != nil && len(options.Names) > 0 {
|
||||
query.Set("names", joinSlice(options.Names))
|
||||
}
|
||||
|
||||
path := fmt.Sprintf("/api/v1/proxies/%s/layers?%s", proxyName, query.Encode())
|
||||
|
||||
response := withResponse[admin.QueryLayerResponse]()
|
||||
|
||||
if options.Options == nil {
|
||||
options.Options = make([]OptionFunc, 0)
|
||||
}
|
||||
|
||||
if err := c.apiGet(ctx, path, &response, options.Options...); err != nil {
|
||||
return nil, errors.WithStack(err)
|
||||
}
|
||||
|
||||
if response.Error != nil {
|
||||
return nil, errors.WithStack(response.Error)
|
||||
}
|
||||
|
||||
return response.Data.Layers, nil
|
||||
}
|
Reference in New Issue
Block a user