mirror of
https://github.com/kyverno/kyverno.git
synced 2025-03-05 07:26:55 +00:00
78 lines
2.2 KiB
Go
78 lines
2.2 KiB
Go
|
package clientset
|
||
|
|
||
|
import (
|
||
|
"github.com/kyverno/kyverno/pkg/clients/dynamic/resource"
|
||
|
"github.com/kyverno/kyverno/pkg/metrics"
|
||
|
"k8s.io/apimachinery/pkg/runtime/schema"
|
||
|
"k8s.io/client-go/dynamic"
|
||
|
)
|
||
|
|
||
|
type namespaceableInterface interface {
|
||
|
Namespace(string) dynamic.ResourceInterface
|
||
|
}
|
||
|
|
||
|
func WrapWithMetrics(inner dynamic.Interface, metrics metrics.MetricsConfigManager, clientType metrics.ClientType) dynamic.Interface {
|
||
|
return &withMetrics{inner, metrics, clientType}
|
||
|
}
|
||
|
|
||
|
func WrapWithTracing(inner dynamic.Interface) dynamic.Interface {
|
||
|
return &withTracing{inner}
|
||
|
}
|
||
|
|
||
|
type withMetrics struct {
|
||
|
inner dynamic.Interface
|
||
|
metrics metrics.MetricsConfigManager
|
||
|
clientType metrics.ClientType
|
||
|
}
|
||
|
|
||
|
type withMetricsNamespaceable struct {
|
||
|
metrics metrics.MetricsConfigManager
|
||
|
resource string
|
||
|
clientType metrics.ClientType
|
||
|
inner namespaceableInterface
|
||
|
}
|
||
|
|
||
|
func (c *withMetricsNamespaceable) Namespace(namespace string) dynamic.ResourceInterface {
|
||
|
recorder := metrics.NamespacedClientQueryRecorder(c.metrics, namespace, c.resource, c.clientType)
|
||
|
return resource.WithMetrics(c.inner.Namespace(namespace), recorder)
|
||
|
}
|
||
|
|
||
|
func (c *withMetrics) Resource(gvr schema.GroupVersionResource) dynamic.NamespaceableResourceInterface {
|
||
|
recorder := metrics.ClusteredClientQueryRecorder(c.metrics, gvr.Resource, c.clientType)
|
||
|
inner := c.inner.Resource(gvr)
|
||
|
return struct {
|
||
|
dynamic.ResourceInterface
|
||
|
namespaceableInterface
|
||
|
}{
|
||
|
resource.WithMetrics(inner, recorder),
|
||
|
&withMetricsNamespaceable{c.metrics, gvr.Resource, c.clientType, inner},
|
||
|
}
|
||
|
}
|
||
|
|
||
|
type withTracing struct {
|
||
|
inner dynamic.Interface
|
||
|
}
|
||
|
|
||
|
type withTracingNamespaceable struct {
|
||
|
client string
|
||
|
kind string
|
||
|
inner namespaceableInterface
|
||
|
}
|
||
|
|
||
|
func (c *withTracingNamespaceable) Namespace(namespace string) dynamic.ResourceInterface {
|
||
|
return resource.WithTracing(c.inner.Namespace(namespace), c.client, c.kind)
|
||
|
}
|
||
|
|
||
|
func (c *withTracing) Resource(gvr schema.GroupVersionResource) dynamic.NamespaceableResourceInterface {
|
||
|
inner := c.inner.Resource(gvr)
|
||
|
client := gvr.GroupResource().String()
|
||
|
kind := gvr.Resource
|
||
|
return struct {
|
||
|
dynamic.ResourceInterface
|
||
|
namespaceableInterface
|
||
|
}{
|
||
|
resource.WithTracing(inner, client, kind),
|
||
|
&withTracingNamespaceable{client, kind, inner},
|
||
|
}
|
||
|
}
|