mirror of
				https://github.com/k3s-io/kubernetes.git
				synced 2025-10-31 13:50:01 +00:00 
			
		
		
		
	refactored operation generator and operation executor to use more general generated functions for operations, completions, and events.
		
			
				
	
	
		
			65 lines
		
	
	
		
			1.8 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			65 lines
		
	
	
		
			1.8 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
| /*
 | |
| Copyright 2017 The Kubernetes Authors.
 | |
| 
 | |
| Licensed under the Apache License, Version 2.0 (the "License");
 | |
| you may not use this file except in compliance with the License.
 | |
| You may obtain a copy of the License at
 | |
| 
 | |
|     http://www.apache.org/licenses/LICENSE-2.0
 | |
| 
 | |
| Unless required by applicable law or agreed to in writing, software
 | |
| distributed under the License is distributed on an "AS IS" BASIS,
 | |
| WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 | |
| See the License for the specific language governing permissions and
 | |
| limitations under the License.
 | |
| */
 | |
| 
 | |
| package util
 | |
| 
 | |
| import (
 | |
| 	"time"
 | |
| 
 | |
| 	"github.com/prometheus/client_golang/prometheus"
 | |
| )
 | |
| 
 | |
| var storageOperationMetric = prometheus.NewHistogramVec(
 | |
| 	prometheus.HistogramOpts{
 | |
| 		Name:    "storage_operation_duration_seconds",
 | |
| 		Help:    "Storage operation duration",
 | |
| 		Buckets: []float64{.1, .25, .5, 1, 2.5, 5, 10, 15, 25, 50},
 | |
| 	},
 | |
| 	[]string{"volume_plugin", "operation_name"},
 | |
| )
 | |
| 
 | |
| var storageOperationErrorMetric = prometheus.NewCounterVec(
 | |
| 	prometheus.CounterOpts{
 | |
| 		Name: "storage_operation_errors_total",
 | |
| 		Help: "Storage operation errors",
 | |
| 	},
 | |
| 	[]string{"volume_plugin", "operation_name"},
 | |
| )
 | |
| 
 | |
| func init() {
 | |
| 	registerMetrics()
 | |
| }
 | |
| 
 | |
| func registerMetrics() {
 | |
| 	prometheus.MustRegister(storageOperationMetric)
 | |
| 	prometheus.MustRegister(storageOperationErrorMetric)
 | |
| }
 | |
| 
 | |
| // OperationCompleteHook returns a hook to call when an operation is completed
 | |
| func OperationCompleteHook(plugin, operationName string) func(*error) {
 | |
| 	requestTime := time.Now()
 | |
| 	opComplete := func(err *error) {
 | |
| 		timeTaken := time.Since(requestTime).Seconds()
 | |
| 		// Create metric with operation name and plugin name
 | |
| 		if *err != nil {
 | |
| 			storageOperationErrorMetric.WithLabelValues(plugin, operationName).Inc()
 | |
| 		} else {
 | |
| 			storageOperationMetric.WithLabelValues(plugin, operationName).Observe(timeTaken)
 | |
| 		}
 | |
| 	}
 | |
| 	return opComplete
 | |
| }
 |