Add Prometheus metrics for S3 and IAM requests

pkg/clients/s3/s3_client.go

@@ -5,13 +5,16 @@ import (
 	"net/http"
 	"os"
 	"strings"
+	"time"
 
 	"github.com/aws/aws-sdk-go-v2/aws"
 	"github.com/aws/aws-sdk-go-v2/config"
 	"github.com/aws/aws-sdk-go-v2/credentials"
 	"github.com/aws/aws-sdk-go-v2/service/s3"
 	"github.com/aws/aws-sdk-go-v2/service/s3/types"
 	"github.com/aws/smithy-go/logging"
+	c "github.com/scality/cosi-driver/pkg/constants"
+	"github.com/scality/cosi-driver/pkg/metrics"
 	"github.com/scality/cosi-driver/pkg/util"
 )
 
@@ -65,6 +68,8 @@ var InitS3Client = func(params util.StorageClientParameters) (*S3Client, error)
 }
 
 func (client *S3Client) CreateBucket(ctx context.Context, bucketName string, params util.StorageClientParameters) error {
+	method := "CreateBucket"
+	start := time.Now()
 
 	input := &s3.CreateBucketInput{
 		Bucket: &bucketName,
@@ -77,12 +82,33 @@ func (client *S3Client) CreateBucket(ctx context.Context, bucketName string, par
 	}
 
 	_, err := client.S3Service.CreateBucket(ctx, input)
+
+	duration := time.Since(start).Seconds()
+	status := c.StatusSuccess
+	if err != nil {
+		status = c.StatusError
+	}
+
+	metrics.S3RequestsTotal.WithLabelValues(method, status).Inc()
+	metrics.S3RequestDuration.WithLabelValues(method, status).Observe(duration)
 	return err
 }
 
 func (client *S3Client) DeleteBucket(ctx context.Context, bucketName string) error {
+	method := "DeleteBucket"
+	start := time.Now()
+
 	_, err := client.S3Service.DeleteBucket(ctx, &s3.DeleteBucketInput{
 		Bucket: &bucketName,
 	})
+	duration := time.Since(start).Seconds()
+
+	status := c.StatusSuccess
+	if err != nil {
+		status = c.StatusError
+	}
+
+	metrics.S3RequestsTotal.WithLabelValues(method, status).Inc()
+	metrics.S3RequestDuration.WithLabelValues(method, status).Observe(duration)
 	return err
 }

pkg/constants/constants.go

@@ -13,6 +13,12 @@ const (
 
 // Service initialization constants
 const (
-	MetricsPath = "/metrics"
-	MetricsAddress = ":8080"
+	MetricsPath    = "/metrics" // Path to expose Prometheus metrics
+	MetricsAddress = ":8080"    // Default address to bind the metrics server
+)
+
+// Prometheus metrics status values
+const (
+	StatusSuccess = "success" // Status value for successful requests
+	StatusError   = "error"   // Status value for failed requests
 )