mirror of https://github.com/k3s-io/k3s.git
120 lines
2.6 KiB
Go
120 lines
2.6 KiB
Go
package v1
|
|
|
|
import (
|
|
"context"
|
|
"sync"
|
|
|
|
"github.com/rancher/norman/controller"
|
|
"github.com/rancher/norman/objectclient"
|
|
"github.com/rancher/norman/objectclient/dynamic"
|
|
"github.com/rancher/norman/restwatch"
|
|
"k8s.io/client-go/rest"
|
|
)
|
|
|
|
type (
|
|
contextKeyType struct{}
|
|
contextClientsKeyType struct{}
|
|
)
|
|
|
|
type Interface interface {
|
|
RESTClient() rest.Interface
|
|
controller.Starter
|
|
|
|
ListenerConfigsGetter
|
|
}
|
|
|
|
type Clients struct {
|
|
Interface Interface
|
|
|
|
ListenerConfig ListenerConfigClient
|
|
}
|
|
|
|
type Client struct {
|
|
sync.Mutex
|
|
restClient rest.Interface
|
|
starters []controller.Starter
|
|
|
|
listenerConfigControllers map[string]ListenerConfigController
|
|
}
|
|
|
|
func Factory(ctx context.Context, config rest.Config) (context.Context, controller.Starter, error) {
|
|
c, err := NewForConfig(config)
|
|
if err != nil {
|
|
return ctx, nil, err
|
|
}
|
|
|
|
cs := NewClientsFromInterface(c)
|
|
|
|
ctx = context.WithValue(ctx, contextKeyType{}, c)
|
|
ctx = context.WithValue(ctx, contextClientsKeyType{}, cs)
|
|
return ctx, c, nil
|
|
}
|
|
|
|
func ClientsFrom(ctx context.Context) *Clients {
|
|
return ctx.Value(contextClientsKeyType{}).(*Clients)
|
|
}
|
|
|
|
func From(ctx context.Context) Interface {
|
|
return ctx.Value(contextKeyType{}).(Interface)
|
|
}
|
|
|
|
func NewClients(config rest.Config) (*Clients, error) {
|
|
iface, err := NewForConfig(config)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return NewClientsFromInterface(iface), nil
|
|
}
|
|
|
|
func NewClientsFromInterface(iface Interface) *Clients {
|
|
return &Clients{
|
|
Interface: iface,
|
|
|
|
ListenerConfig: &listenerConfigClient2{
|
|
iface: iface.ListenerConfigs(""),
|
|
},
|
|
}
|
|
}
|
|
|
|
func NewForConfig(config rest.Config) (Interface, error) {
|
|
if config.NegotiatedSerializer == nil {
|
|
config.NegotiatedSerializer = dynamic.NegotiatedSerializer
|
|
}
|
|
|
|
restClient, err := restwatch.UnversionedRESTClientFor(&config)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &Client{
|
|
restClient: restClient,
|
|
|
|
listenerConfigControllers: map[string]ListenerConfigController{},
|
|
}, nil
|
|
}
|
|
|
|
func (c *Client) RESTClient() rest.Interface {
|
|
return c.restClient
|
|
}
|
|
|
|
func (c *Client) Sync(ctx context.Context) error {
|
|
return controller.Sync(ctx, c.starters...)
|
|
}
|
|
|
|
func (c *Client) Start(ctx context.Context, threadiness int) error {
|
|
return controller.Start(ctx, threadiness, c.starters...)
|
|
}
|
|
|
|
type ListenerConfigsGetter interface {
|
|
ListenerConfigs(namespace string) ListenerConfigInterface
|
|
}
|
|
|
|
func (c *Client) ListenerConfigs(namespace string) ListenerConfigInterface {
|
|
objectClient := objectclient.NewObjectClient(namespace, c.restClient, &ListenerConfigResource, ListenerConfigGroupVersionKind, listenerConfigFactory{})
|
|
return &listenerConfigClient{
|
|
ns: namespace,
|
|
client: c,
|
|
objectClient: objectClient,
|
|
}
|
|
}
|