Added option to ommit recording

Author: bharat-gadde-dev

otelginmetrics/config.go

@@ -14,6 +14,7 @@ type config struct {
 	groupedStatus  bool
 	recorder       Recorder
 	attributes     func(serverName, route string, request *http.Request) []attribute.KeyValue
+	shouldRecord   func(serverName, route string, request *http.Request) bool
 }
 
 func defaultConfig() *config {
@@ -23,6 +24,9 @@ func defaultConfig() *config {
 		recordSize:     true,
 		groupedStatus:  true,
 		attributes:     DefaultAttributes,
+		shouldRecord: func(_, _ string, _ *http.Request) bool {
+			return true
+		},
 	}
 }
 

otelginmetrics/middleware.go

@@ -28,9 +28,12 @@ func Middleware(service string, options ...Option) gin.HandlerFunc {
 		if len(route) <= 0 {
 			route = "nonconfigured"
 		}
+		if !cfg.shouldRecord(service, route, ginCtx.Request) {
+			ginCtx.Next()
+			return
+		}
 
 		start := time.Now()
-
 		reqAttributes := cfg.attributes(service, route, ginCtx.Request)
 
 		if cfg.recordInFlight {

otelginmetrics/option.go

@@ -57,8 +57,18 @@ func WithGroupedStatusDisabled() Option {
 	})
 }
 
+// WithRecorder sets a recorder for recording requests
+// By default the open telemetry recorder is used
 func WithRecorder(recorder Recorder) Option {
 	return optionFunc(func(cfg *config) {
 		cfg.recorder = recorder
 	})
 }
+
+// WithShouldRecordFunc sets a func using which whether a record should be recorded
+// By default the all api calls are recorded
+func WithShouldRecordFunc(shouldRecord func(serverName, route string, request *http.Request) bool) Option {
+	return optionFunc(func(cfg *config) {
+		cfg.shouldRecord = shouldRecord
+	})
+}