mirror of
https://github.com/rancher/norman.git
synced 2025-06-29 16:58:05 +00:00
125 lines
2.8 KiB
Go
125 lines
2.8 KiB
Go
package generator
|
|
|
|
var k8sClientTemplate = `package {{.version.Version}}
|
|
|
|
import (
|
|
"sync"
|
|
"context"
|
|
|
|
"github.com/rancher/norman/objectclient"
|
|
"github.com/rancher/norman/objectclient/dynamic"
|
|
"github.com/rancher/norman/controller"
|
|
"github.com/rancher/norman/restwatch"
|
|
"k8s.io/client-go/rest"
|
|
)
|
|
|
|
type (
|
|
contextKeyType struct{}
|
|
contextClientsKeyType struct{}
|
|
)
|
|
|
|
type Interface interface {
|
|
RESTClient() rest.Interface
|
|
controller.Starter
|
|
{{range .schemas}}
|
|
{{.CodeNamePlural}}Getter{{end}}
|
|
}
|
|
|
|
type Clients struct {
|
|
Interface Interface
|
|
{{range .schemas}}
|
|
{{.CodeName}} {{.CodeName}}Client{{end}}
|
|
}
|
|
|
|
type Client struct {
|
|
sync.Mutex
|
|
restClient rest.Interface
|
|
starters []controller.Starter
|
|
{{range .schemas}}
|
|
{{.ID}}Controllers map[string]{{.CodeName}}Controller{{end}}
|
|
}
|
|
|
|
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,
|
|
{{range .schemas}}
|
|
{{.CodeName}}: &{{.ID}}Client2{
|
|
iface: iface.{{.CodeNamePlural}}(""),
|
|
},{{end}}
|
|
}
|
|
}
|
|
|
|
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,
|
|
{{range .schemas}}
|
|
{{.ID}}Controllers: map[string]{{.CodeName}}Controller{},{{end}}
|
|
}, 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...)
|
|
}
|
|
|
|
{{range .schemas}}
|
|
type {{.CodeNamePlural}}Getter interface {
|
|
{{.CodeNamePlural}}(namespace string) {{.CodeName}}Interface
|
|
}
|
|
|
|
func (c *Client) {{.CodeNamePlural}}(namespace string) {{.CodeName}}Interface {
|
|
objectClient := objectclient.NewObjectClient(namespace, c.restClient, &{{.CodeName}}Resource, {{.CodeName}}GroupVersionKind, {{.ID}}Factory{})
|
|
return &{{.ID}}Client{
|
|
ns: namespace,
|
|
client: c,
|
|
objectClient: objectClient,
|
|
}
|
|
}
|
|
{{end}}
|
|
`
|