1
0
Fork 0
mirror of https://github.com/kyverno/kyverno.git synced 2025-03-10 09:56:55 +00:00
kyverno/pkg/clients/wrappers/utils/wrapper.go
Charles-Edouard Brétéché f0622a8a3b
refactor: use generics in client wrappers (#4525)
Signed-off-by: Charles-Edouard Brétéché <charled.breteche@gmail.com>

Signed-off-by: Charles-Edouard Brétéché <charled.breteche@gmail.com>
2022-09-07 20:56:38 +08:00

55 lines
2.9 KiB
Go

package utils
import (
"context"
"github.com/kyverno/kyverno/pkg/metrics"
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
"k8s.io/apimachinery/pkg/types"
"k8s.io/apimachinery/pkg/watch"
)
func Create[K any](ctx context.Context, m ClientQueryMetric, kind string, ns string, o *K, opts metav1.CreateOptions, inner func(context.Context, *K, metav1.CreateOptions) (*K, error)) (*K, error) {
m.Record(metrics.ClientCreate, metrics.KyvernoClient, kind, ns)
return inner(ctx, o, opts)
}
func Update[K any](ctx context.Context, m ClientQueryMetric, kind string, ns string, o *K, opts metav1.UpdateOptions, inner func(context.Context, *K, metav1.UpdateOptions) (*K, error)) (*K, error) {
m.Record(metrics.ClientUpdate, metrics.KyvernoClient, kind, ns)
return inner(ctx, o, opts)
}
func UpdateStatus[K any](ctx context.Context, m ClientQueryMetric, kind string, ns string, o *K, opts metav1.UpdateOptions, inner func(context.Context, *K, metav1.UpdateOptions) (*K, error)) (*K, error) {
m.Record(metrics.ClientUpdateStatus, metrics.KyvernoClient, kind, ns)
return inner(ctx, o, opts)
}
func Delete(ctx context.Context, m ClientQueryMetric, kind string, ns string, name string, opts metav1.DeleteOptions, inner func(context.Context, string, metav1.DeleteOptions) error) error {
m.Record(metrics.ClientDelete, metrics.KyvernoClient, kind, ns)
return inner(ctx, name, opts)
}
func DeleteCollection(ctx context.Context, m ClientQueryMetric, kind string, ns string, opts metav1.DeleteOptions, listOpts metav1.ListOptions, inner func(context.Context, metav1.DeleteOptions, metav1.ListOptions) error) error {
m.Record(metrics.ClientDeleteCollection, metrics.KyvernoClient, kind, ns)
return inner(ctx, opts, listOpts)
}
func Get[K any](ctx context.Context, m ClientQueryMetric, kind string, ns string, name string, opts metav1.GetOptions, inner func(context.Context, string, metav1.GetOptions) (*K, error)) (*K, error) {
m.Record(metrics.ClientGet, metrics.KyvernoClient, kind, ns)
return inner(ctx, name, opts)
}
func List[K any](ctx context.Context, m ClientQueryMetric, kind string, ns string, opts metav1.ListOptions, inner func(context.Context, metav1.ListOptions) (*K, error)) (*K, error) {
m.Record(metrics.ClientList, metrics.KyvernoClient, kind, ns)
return inner(ctx, opts)
}
func Watch(ctx context.Context, m ClientQueryMetric, kind string, ns string, opts metav1.ListOptions, inner func(context.Context, metav1.ListOptions) (watch.Interface, error)) (watch.Interface, error) {
m.Record(metrics.ClientWatch, metrics.KyvernoClient, kind, ns)
return inner(ctx, opts)
}
func Patch[K any](ctx context.Context, m ClientQueryMetric, kind string, ns string, name string, pt types.PatchType, data []byte, opts metav1.PatchOptions, inner func(context.Context, string, types.PatchType, []byte, metav1.PatchOptions, ...string) (*K, error), subresources ...string) (*K, error) {
m.Record(metrics.ClientPatch, metrics.KyvernoClient, kind, ns)
return inner(ctx, name, pt, data, opts, subresources...)
}