mirror of
https://github.com/rancher/steve.git
synced 2025-06-29 08:17:20 +00:00
83 lines
3.7 KiB
Go
83 lines
3.7 KiB
Go
package metrics
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/rancher/apiserver/pkg/types"
|
|
"github.com/rancher/steve/pkg/metrics"
|
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
"k8s.io/apimachinery/pkg/apis/meta/v1/unstructured"
|
|
k8stypes "k8s.io/apimachinery/pkg/types"
|
|
"k8s.io/apimachinery/pkg/watch"
|
|
"k8s.io/client-go/dynamic"
|
|
)
|
|
|
|
type ResourceClientWithMetrics struct {
|
|
dynamic.ResourceInterface
|
|
}
|
|
|
|
func Wrap(resourceInterface dynamic.ResourceInterface, err error) (ResourceClientWithMetrics, error) {
|
|
client := ResourceClientWithMetrics{}
|
|
if err != nil {
|
|
return client, err
|
|
}
|
|
client.ResourceInterface = resourceInterface
|
|
return client, err
|
|
}
|
|
|
|
func (r ResourceClientWithMetrics) Create(apiOp *types.APIRequest, obj *unstructured.Unstructured, options metav1.CreateOptions, subresources ...string) (*unstructured.Unstructured, error) {
|
|
m := metrics.MetricLogger{Resource: apiOp.Schema.ID, Method: apiOp.Method}
|
|
start := time.Now()
|
|
obj, err := r.ResourceInterface.Create(apiOp.Context(), obj, options, subresources...)
|
|
m.RecordK8sClientResponseTime(err, float64(time.Since(start).Milliseconds()))
|
|
return obj, err
|
|
}
|
|
|
|
func (r ResourceClientWithMetrics) Update(apiOp *types.APIRequest, obj *unstructured.Unstructured, options metav1.UpdateOptions, subresources ...string) (*unstructured.Unstructured, error) {
|
|
m := metrics.MetricLogger{Resource: apiOp.Schema.ID, Method: apiOp.Method}
|
|
start := time.Now()
|
|
obj, err := r.ResourceInterface.Update(apiOp.Context(), obj, options, subresources...)
|
|
m.RecordK8sClientResponseTime(err, float64(time.Since(start).Milliseconds()))
|
|
return obj, err
|
|
}
|
|
|
|
func (r ResourceClientWithMetrics) Delete(apiOp *types.APIRequest, name string, options metav1.DeleteOptions, subresources ...string) error {
|
|
m := metrics.MetricLogger{Resource: apiOp.Schema.ID, Method: apiOp.Method}
|
|
start := time.Now()
|
|
err := r.ResourceInterface.Delete(apiOp.Context(), name, options, subresources...)
|
|
m.RecordK8sClientResponseTime(err, float64(time.Since(start).Milliseconds()))
|
|
return err
|
|
}
|
|
|
|
func (r ResourceClientWithMetrics) Get(apiOp *types.APIRequest, name string, options metav1.GetOptions, subresources ...string) (*unstructured.Unstructured, error) {
|
|
m := metrics.MetricLogger{Resource: apiOp.Schema.ID, Method: apiOp.Method}
|
|
start := time.Now()
|
|
obj, err := r.ResourceInterface.Get(apiOp.Context(), name, options, subresources...)
|
|
m.RecordK8sClientResponseTime(err, float64(time.Since(start).Milliseconds()))
|
|
return obj, err
|
|
}
|
|
|
|
func (r ResourceClientWithMetrics) List(apiOp *types.APIRequest, opts metav1.ListOptions) (*unstructured.UnstructuredList, error) {
|
|
m := metrics.MetricLogger{Resource: apiOp.Schema.ID, Method: apiOp.Method}
|
|
start := time.Now()
|
|
obj, err := r.ResourceInterface.List(apiOp.Context(), opts)
|
|
m.RecordK8sClientResponseTime(err, float64(time.Since(start).Milliseconds()))
|
|
return obj, err
|
|
}
|
|
|
|
func (r ResourceClientWithMetrics) Watch(apiOp *types.APIRequest, opts metav1.ListOptions) (watch.Interface, error) {
|
|
m := metrics.MetricLogger{Resource: apiOp.Schema.ID, Method: apiOp.Method}
|
|
start := time.Now()
|
|
watchInterface, err := r.ResourceInterface.Watch(apiOp.Context(), opts)
|
|
m.RecordK8sClientResponseTime(err, float64(time.Since(start).Milliseconds()))
|
|
return watchInterface, err
|
|
}
|
|
|
|
func (r ResourceClientWithMetrics) Patch(apiOp *types.APIRequest, name string, pt k8stypes.PatchType, data []byte, options metav1.PatchOptions, subresources ...string) (*unstructured.Unstructured, error) {
|
|
m := metrics.MetricLogger{Resource: apiOp.Schema.ID, Method: apiOp.Method}
|
|
start := time.Now()
|
|
obj, err := r.ResourceInterface.Patch(apiOp.Context(), name, pt, data, options, subresources...)
|
|
m.RecordK8sClientResponseTime(err, float64(time.Since(start).Milliseconds()))
|
|
return obj, err
|
|
}
|