mirror of
https://github.com/rancher/types.git
synced 2025-09-12 13:03:45 +00:00
Add NodeController
This commit is contained in:
@@ -5,6 +5,40 @@ import (
|
||||
runtime "k8s.io/apimachinery/pkg/runtime"
|
||||
)
|
||||
|
||||
// DeepCopyInto is an autogenerated deepcopy function, copying the receiver, writing into out. in must be non-nil.
|
||||
func (in *NodeList) DeepCopyInto(out *NodeList) {
|
||||
*out = *in
|
||||
out.TypeMeta = in.TypeMeta
|
||||
out.ListMeta = in.ListMeta
|
||||
if in.Items != nil {
|
||||
in, out := &in.Items, &out.Items
|
||||
*out = make([]core_v1.Node, len(*in))
|
||||
for i := range *in {
|
||||
(*in)[i].DeepCopyInto(&(*out)[i])
|
||||
}
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
// DeepCopy is an autogenerated deepcopy function, copying the receiver, creating a new NodeList.
|
||||
func (in *NodeList) DeepCopy() *NodeList {
|
||||
if in == nil {
|
||||
return nil
|
||||
}
|
||||
out := new(NodeList)
|
||||
in.DeepCopyInto(out)
|
||||
return out
|
||||
}
|
||||
|
||||
// DeepCopyObject is an autogenerated deepcopy function, copying the receiver, creating a new runtime.Object.
|
||||
func (in *NodeList) DeepCopyObject() runtime.Object {
|
||||
if c := in.DeepCopy(); c != nil {
|
||||
return c
|
||||
} else {
|
||||
return nil
|
||||
}
|
||||
}
|
||||
|
||||
// DeepCopyInto is an autogenerated deepcopy function, copying the receiver, writing into out. in must be non-nil.
|
||||
func (in *PodList) DeepCopyInto(out *PodList) {
|
||||
*out = *in
|
||||
|
@@ -15,6 +15,7 @@ type Interface interface {
|
||||
controller.Starter
|
||||
|
||||
PodsGetter
|
||||
NodesGetter
|
||||
}
|
||||
|
||||
type Client struct {
|
||||
@@ -22,7 +23,8 @@ type Client struct {
|
||||
restClient rest.Interface
|
||||
starters []controller.Starter
|
||||
|
||||
podControllers map[string]PodController
|
||||
podControllers map[string]PodController
|
||||
nodeControllers map[string]NodeController
|
||||
}
|
||||
|
||||
func NewForConfig(config rest.Config) (Interface, error) {
|
||||
@@ -39,7 +41,8 @@ func NewForConfig(config rest.Config) (Interface, error) {
|
||||
return &Client{
|
||||
restClient: restClient,
|
||||
|
||||
podControllers: map[string]PodController{},
|
||||
podControllers: map[string]PodController{},
|
||||
nodeControllers: map[string]NodeController{},
|
||||
}, nil
|
||||
}
|
||||
|
||||
@@ -67,3 +70,16 @@ func (c *Client) Pods(namespace string) PodInterface {
|
||||
objectClient: objectClient,
|
||||
}
|
||||
}
|
||||
|
||||
type NodesGetter interface {
|
||||
Nodes(namespace string) NodeInterface
|
||||
}
|
||||
|
||||
func (c *Client) Nodes(namespace string) NodeInterface {
|
||||
objectClient := clientbase.NewObjectClient(namespace, c.restClient, &NodeResource, NodeGroupVersionKind, nodeFactory{})
|
||||
return &nodeClient{
|
||||
ns: namespace,
|
||||
client: c,
|
||||
objectClient: objectClient,
|
||||
}
|
||||
}
|
||||
|
188
apis/core/v1/zz_generated_node_controller.go
Normal file
188
apis/core/v1/zz_generated_node_controller.go
Normal file
@@ -0,0 +1,188 @@
|
||||
package v1
|
||||
|
||||
import (
|
||||
"context"
|
||||
|
||||
"github.com/rancher/norman/clientbase"
|
||||
"github.com/rancher/norman/controller"
|
||||
"k8s.io/api/core/v1"
|
||||
"k8s.io/apimachinery/pkg/api/errors"
|
||||
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
||||
"k8s.io/apimachinery/pkg/labels"
|
||||
"k8s.io/apimachinery/pkg/runtime"
|
||||
"k8s.io/apimachinery/pkg/runtime/schema"
|
||||
"k8s.io/apimachinery/pkg/watch"
|
||||
"k8s.io/client-go/tools/cache"
|
||||
)
|
||||
|
||||
var (
|
||||
NodeGroupVersionKind = schema.GroupVersionKind{
|
||||
Version: "v1",
|
||||
Group: "",
|
||||
Kind: "Node",
|
||||
}
|
||||
NodeResource = metav1.APIResource{
|
||||
Name: "nodes",
|
||||
SingularName: "node",
|
||||
Namespaced: false,
|
||||
Kind: NodeGroupVersionKind.Kind,
|
||||
}
|
||||
)
|
||||
|
||||
type NodeList struct {
|
||||
metav1.TypeMeta `json:",inline"`
|
||||
metav1.ListMeta `json:"metadata,omitempty"`
|
||||
Items []v1.Node
|
||||
}
|
||||
|
||||
type NodeHandlerFunc func(key string, obj *v1.Node) error
|
||||
|
||||
type NodeLister interface {
|
||||
List(namespace string, selector labels.Selector) (ret []*v1.Node, err error)
|
||||
Get(namespace, name string) (*v1.Node, error)
|
||||
}
|
||||
|
||||
type NodeController interface {
|
||||
Informer() cache.SharedIndexInformer
|
||||
Lister() NodeLister
|
||||
AddHandler(handler NodeHandlerFunc)
|
||||
Enqueue(namespace, name string)
|
||||
Sync(ctx context.Context) error
|
||||
Start(ctx context.Context, threadiness int) error
|
||||
}
|
||||
|
||||
type NodeInterface interface {
|
||||
ObjectClient() *clientbase.ObjectClient
|
||||
Create(*v1.Node) (*v1.Node, error)
|
||||
Get(name string, opts metav1.GetOptions) (*v1.Node, error)
|
||||
Update(*v1.Node) (*v1.Node, error)
|
||||
Delete(name string, options *metav1.DeleteOptions) error
|
||||
List(opts metav1.ListOptions) (*v1.NodeList, error)
|
||||
Watch(opts metav1.ListOptions) (watch.Interface, error)
|
||||
DeleteCollection(deleteOpts *metav1.DeleteOptions, listOpts metav1.ListOptions) error
|
||||
Controller() NodeController
|
||||
}
|
||||
|
||||
type nodeLister struct {
|
||||
controller *nodeController
|
||||
}
|
||||
|
||||
func (l *nodeLister) List(namespace string, selector labels.Selector) (ret []*v1.Node, err error) {
|
||||
err = cache.ListAllByNamespace(l.controller.Informer().GetIndexer(), namespace, selector, func(obj interface{}) {
|
||||
ret = append(ret, obj.(*v1.Node))
|
||||
})
|
||||
return
|
||||
}
|
||||
|
||||
func (l *nodeLister) Get(namespace, name string) (*v1.Node, error) {
|
||||
obj, exists, err := l.controller.Informer().GetIndexer().GetByKey(namespace + "/" + name)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if !exists {
|
||||
return nil, errors.NewNotFound(schema.GroupResource{
|
||||
Group: NodeGroupVersionKind.Group,
|
||||
Resource: "node",
|
||||
}, name)
|
||||
}
|
||||
return obj.(*v1.Node), nil
|
||||
}
|
||||
|
||||
type nodeController struct {
|
||||
controller.GenericController
|
||||
}
|
||||
|
||||
func (c *nodeController) Lister() NodeLister {
|
||||
return &nodeLister{
|
||||
controller: c,
|
||||
}
|
||||
}
|
||||
|
||||
func (c *nodeController) AddHandler(handler NodeHandlerFunc) {
|
||||
c.GenericController.AddHandler(func(key string) error {
|
||||
obj, exists, err := c.Informer().GetStore().GetByKey(key)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
if !exists {
|
||||
return handler(key, nil)
|
||||
}
|
||||
return handler(key, obj.(*v1.Node))
|
||||
})
|
||||
}
|
||||
|
||||
type nodeFactory struct {
|
||||
}
|
||||
|
||||
func (c nodeFactory) Object() runtime.Object {
|
||||
return &v1.Node{}
|
||||
}
|
||||
|
||||
func (c nodeFactory) List() runtime.Object {
|
||||
return &NodeList{}
|
||||
}
|
||||
|
||||
func (s *nodeClient) Controller() NodeController {
|
||||
s.client.Lock()
|
||||
defer s.client.Unlock()
|
||||
|
||||
c, ok := s.client.nodeControllers[s.ns]
|
||||
if ok {
|
||||
return c
|
||||
}
|
||||
|
||||
genericController := controller.NewGenericController(NodeGroupVersionKind.Kind+"Controller",
|
||||
s.objectClient)
|
||||
|
||||
c = &nodeController{
|
||||
GenericController: genericController,
|
||||
}
|
||||
|
||||
s.client.nodeControllers[s.ns] = c
|
||||
s.client.starters = append(s.client.starters, c)
|
||||
|
||||
return c
|
||||
}
|
||||
|
||||
type nodeClient struct {
|
||||
client *Client
|
||||
ns string
|
||||
objectClient *clientbase.ObjectClient
|
||||
controller NodeController
|
||||
}
|
||||
|
||||
func (s *nodeClient) ObjectClient() *clientbase.ObjectClient {
|
||||
return s.objectClient
|
||||
}
|
||||
|
||||
func (s *nodeClient) Create(o *v1.Node) (*v1.Node, error) {
|
||||
obj, err := s.objectClient.Create(o)
|
||||
return obj.(*v1.Node), err
|
||||
}
|
||||
|
||||
func (s *nodeClient) Get(name string, opts metav1.GetOptions) (*v1.Node, error) {
|
||||
obj, err := s.objectClient.Get(name, opts)
|
||||
return obj.(*v1.Node), err
|
||||
}
|
||||
|
||||
func (s *nodeClient) Update(o *v1.Node) (*v1.Node, error) {
|
||||
obj, err := s.objectClient.Update(o.Name, o)
|
||||
return obj.(*v1.Node), err
|
||||
}
|
||||
|
||||
func (s *nodeClient) Delete(name string, options *metav1.DeleteOptions) error {
|
||||
return s.objectClient.Delete(name, options)
|
||||
}
|
||||
|
||||
func (s *nodeClient) List(opts metav1.ListOptions) (*v1.NodeList, error) {
|
||||
obj, err := s.objectClient.List(opts)
|
||||
return obj.(*v1.NodeList), err
|
||||
}
|
||||
|
||||
func (s *nodeClient) Watch(opts metav1.ListOptions) (watch.Interface, error) {
|
||||
return s.objectClient.Watch(opts)
|
||||
}
|
||||
|
||||
func (s *nodeClient) DeleteCollection(deleteOpts *metav1.DeleteOptions, listOpts metav1.ListOptions) error {
|
||||
return s.objectClient.DeleteCollection(deleteOpts, listOpts)
|
||||
}
|
2
main.go
2
main.go
@@ -17,6 +17,6 @@ func main() {
|
||||
generator.Generate(workloadSchema.Schemas)
|
||||
generator.Generate(authzSchema.Schemas)
|
||||
// Group by API group
|
||||
generator.GenerateNativeTypes(v1.Pod{})
|
||||
generator.GenerateNativeTypes(v1.Pod{}, v1.Node{})
|
||||
generator.GenerateNativeTypes(v1beta2.Deployment{})
|
||||
}
|
||||
|
Reference in New Issue
Block a user