Add ClusterWithContext to templates

Signed-off-by: Nelo-T. Wallus <[email protected]>

Author: ntnn

cmd/cluster-informer-gen/generators/generic.go

@@ -132,6 +132,7 @@ func (g *genericGenerator) GenerateType(c *generator.Context, t *types.Type, w i
 		"cacheSharedIndexInformer":          c.Universe.Type(cacheSharedIndexInformer),
 		"scopeableCacheSharedIndexInformer": c.Universe.Type(scopeableCacheSharedIndexInformer),
 		"fmtErrorf":                         c.Universe.Type(fmtErrorfFunc),
+		"contextContext":                    c.Universe.Type(contextContext),
 		"groups":                            groups,
 		"schemeGVs":                         schemeGVs,
 		"schemaGroupResource":               c.Universe.Type(schemaGroupResource),
@@ -155,6 +156,7 @@ func (g *genericGenerator) GenerateType(c *generator.Context, t *types.Type, w i
 var genericClusterInformer = `
 type GenericClusterInformer interface {
 	Cluster({{.logicalclusterName|raw}}) {{.genericInformer|raw}}
+	ClusterWithContext({{.contextContext|raw}}, {{.logicalclusterName|raw}}) {{.genericInformer|raw}}
 	Informer() {{.scopeableCacheSharedIndexInformer|raw}}
 	Lister() {{.kcpcacheGenericClusterLister|raw}}
 }
@@ -188,6 +190,15 @@ func (i *genericClusterInformer) Cluster(clusterName {{.logicalclusterName|raw}}
 		lister:   i.Lister().ByCluster(clusterName),
 	}
 }
+
+// ClusterWithContext scopes to a GenericInformer and unregisters all
+// handles registered through it once the provided context is canceled.
+func (i *genericClusterInformer) ClusterWithContext(ctx {{.contextContext|raw}}, clusterName {{.logicalclusterName|raw}}) {{.genericInformer|raw}} {
+	return &genericInformer{
+		informer: i.Informer().ClusterWithContext(ctx, clusterName),
+		lister:   i.Lister().ByCluster(clusterName),
+	}
+}
 `
 
 var genericInformer = `

cmd/cluster-informer-gen/generators/informer.go

@@ -107,6 +107,7 @@ func (g *informerGenerator) GenerateType(c *generator.Context, t *types.Type, w
 		"clientSetInterface":                    clientSetInterface,
 		"clientSetClusterInterface":             clientSetClusterInterface,
 		"contextBackground":                     c.Universe.Function(contextBackgroundFunc),
+		"contextContext":                        c.Universe.Type(contextContext),
 		"group":                                 namer.IC(g.groupGoName),
 		"interfacesTweakListOptionsFunc":        c.Universe.Type(types.Name{Package: g.internalInterfacesPackage, Name: "TweakListOptionsFunc"}),
 		"interfacesSharedInformerFactory":       c.Universe.Type(types.Name{Package: g.internalInterfacesPackage, Name: "SharedInformerFactory"}),
@@ -151,6 +152,7 @@ var typeClusterInformerInterface = `
 // $.type|publicPlural$.
 type $.type|public$ClusterInformer interface {
 	Cluster($.logicalclusterName|raw$) $.informerInterface|raw$
+	ClusterWithContext($.contextContext|raw$, $.logicalclusterName|raw$) $.informerInterface|raw$
 	Informer() $.scopeableCacheSharedIndexInformer|raw$
 	Lister() $.clusterLister|raw$
 }
@@ -227,6 +229,13 @@ func (i *$.type|private$ClusterInformer) Cluster(clusterName $.logicalclusterNam
 		lister:   i.Lister().Cluster(clusterName),
 	}
 }
+
+func (i *$.type|private$ClusterInformer) ClusterWithContext(ctx $.contextContext|raw$, clusterName $.logicalclusterName|raw$) $.informerInterface|raw$ {
+	return &$.type|private$Informer{
+		informer: i.Informer().ClusterWithContext(ctx, clusterName),
+		lister:   i.Lister().Cluster(clusterName),
+	}
+}
 `
 
 var typeInformer = `

cmd/cluster-informer-gen/generators/types.go

@@ -33,6 +33,7 @@ var (
 	kcpcacheNewGenericClusterLister    = types.Name{Package: "github.com/kcp-dev/apimachinery/v2/pkg/cache", Name: "NewGenericClusterLister"}
 	kcpinformersNewSharedIndexInformer = types.Name{Package: "github.com/kcp-dev/apimachinery/v2/third_party/informers", Name: "NewSharedIndexInformer"}
 	cacheTransformFunc                 = types.Name{Package: "k8s.io/client-go/tools/cache", Name: "TransformFunc"}
+	contextContext                     = types.Name{Package: "context", Name: "Context"}
 	contextBackgroundFunc              = types.Name{Package: "context", Name: "Background"}
 	fmtErrorfFunc                      = types.Name{Package: "fmt", Name: "Errorf"}
 	reflectType                        = types.Name{Package: "reflect", Name: "Type"}