diff --git a/config/context.go b/config/context.go index 6948f52b..337620be 100644 --- a/config/context.go +++ b/config/context.go @@ -17,6 +17,7 @@ import ( extv1beta1 "github.com/rancher/types/apis/extensions/v1beta1" managementv3 "github.com/rancher/types/apis/management.cattle.io/v3" managementSchema "github.com/rancher/types/apis/management.cattle.io/v3/schema" + knetworkingv1 "github.com/rancher/types/apis/networking.k8s.io/v1" projectv3 "github.com/rancher/types/apis/project.cattle.io/v3" projectSchema "github.com/rancher/types/apis/project.cattle.io/v3/schema" rbacv1 "github.com/rancher/types/apis/rbac.authorization.k8s.io/v1" @@ -187,6 +188,7 @@ type UserContext struct { Extensions extv1beta1.Interface BatchV1 batchv1.Interface BatchV1Beta1 batchv1beta1.Interface + Networking knetworkingv1.Interface } func (w *UserContext) controllers() []controller.Starter { @@ -198,6 +200,7 @@ func (w *UserContext) controllers() []controller.Starter { w.Extensions, w.BatchV1, w.BatchV1Beta1, + w.Networking, } } @@ -381,6 +384,11 @@ func NewUserContext(scaledContext *ScaledContext, config rest.Config, clusterNam return nil, err } + context.Networking, err = knetworkingv1.NewForConfig(config) + if err != nil { + return nil, err + } + context.Extensions, err = extv1beta1.NewForConfig(config) if err != nil { return nil, err diff --git a/main.go b/main.go index cb48fe48..02e73ce6 100644 --- a/main.go +++ b/main.go @@ -15,6 +15,7 @@ import ( "k8s.io/api/core/v1" extv1beta1 "k8s.io/api/extensions/v1beta1" + knetworkingv1 "k8s.io/api/networking/v1" rbacv1 "k8s.io/api/rbac/v1" ) @@ -51,6 +52,9 @@ func main() { rbacv1.ClusterRoleBinding{}, rbacv1.ClusterRole{}, }) + generator.GenerateNativeTypes(knetworkingv1.SchemeGroupVersion, []interface{}{ + knetworkingv1.NetworkPolicy{}, + }, nil) generator.GenerateNativeTypes(batchv1.SchemeGroupVersion, []interface{}{ batchv1.Job{}, }, nil)