Add delete all pushgateway metric endpoint

appclacks · Jul 4, 2024 · 6a46a73 · 6a46a73
1 parent 7c45066
commit 6a46a73
Show file tree

Hide file tree

Showing 8 changed files with 41 additions and 2 deletions.
diff --git a/internal/database/pushgateway.go b/internal/database/pushgateway.go
@@ -185,3 +185,15 @@ func (c *Database) CleanPushgatewayMetrics(ctx context.Context) (int64, error) {
 	}
 	return affected, nil
 }
+
+func (c *Database) DeleteAllPushgatewayMetrics(ctx context.Context) error {
+	result, err := c.db.ExecContext(ctx, "TRUNCATE pushgateway_metric")
+	if err != nil {
+		return fmt.Errorf("fail to clean all pushgateway metrics: %w", err)
+	}
+	_, err = result.RowsAffected()
+	if err != nil {
+		return fmt.Errorf("fail to check affected row: %w", err)
+	}
+	return nil
+}
diff --git a/internal/database/pushgateway_test.go b/internal/database/pushgateway_test.go
@@ -180,4 +180,14 @@ func TestPushgatewayCRUD(t *testing.T) {
 	assert.NoError(t, err)
 	assert.Equal(t, int64(5), deleteCount)
 
+	metric, err = TestComponent.GetMetrics(context.Background())
+	assert.NoError(t, err)
+	assert.Len(t, metric, 5)
+
+	err = TestComponent.DeleteAllPushgatewayMetrics(context.Background())
+	assert.NoError(t, err)
+
+	metric, err = TestComponent.GetMetrics(context.Background())
+	assert.NoError(t, err)
+	assert.Len(t, metric, 0)
 }
diff --git a/internal/http/handlers/builder.go b/internal/http/handlers/builder.go
@@ -23,6 +23,7 @@ type PushgatewayService interface {
 	DeleteMetricsByName(ctx context.Context, name string) error
 	DeleteMetricByID(ctx context.Context, id string) error
 	PrometheusMetrics(ctx context.Context) (string, error)
+	DeleteAllPushgatewayMetrics(ctx context.Context) error
 }
 
 type Builder struct {

diff --git a/internal/http/handlers/pushgateway.go b/internal/http/handlers/pushgateway.go
@@ -85,6 +85,15 @@ func (b *Builder) DeleteMetric(ec echo.Context) error {
 	return ec.JSON(http.StatusOK, NewResponse("metrics deleted"))
 }
 
+func (b *Builder) DeleteAllPushgatewayMetrics(ec echo.Context) error {
+
+	err := b.pushgateway.DeleteAllPushgatewayMetrics(ec.Request().Context())
+	if err != nil {
+		return err
+	}
+	return ec.JSON(http.StatusOK, NewResponse("metrics deleted"))
+}
+
 func (b *Builder) ListPushgatewayMetrics(ec echo.Context) error {
 	metrics, err := b.pushgateway.GetMetrics(ec.Request().Context())
 	if err != nil {

diff --git a/internal/http/integration_test.go b/internal/http/integration_test.go
@@ -871,8 +871,8 @@ func TestIntegration(t *testing.T) {
 		},
 		{
 			url:            "/api/v1/pushgateway",
-			expectedStatus: 404,
-			body:           "Not Found",
+			expectedStatus: 200,
+			body:           "metrics deleted",
 			method:         "DELETE",
 			headers: map[string]string{
 				"Authorization": basicAuth(testUser, testPassword),

diff --git a/internal/http/server.go b/internal/http/server.go
@@ -121,6 +121,7 @@ func NewServer(logger *slog.Logger, config Configuration, registry *prometheus.R
 	apiGroup.GET("/healthcheck", builder.ListHealthchecks)
 	apiGroup.GET("/cabourotte/discovery", builder.CabourotteDiscovery)
 	apiGroup.POST("/pushgateway", builder.CreateOrUpdatePushgatewayMetric)
+	apiGroup.DELETE("/pushgateway", builder.DeleteAllPushgatewayMetrics)
 	apiGroup.DELETE("/pushgateway/:identifier", builder.DeleteMetric)
 	apiGroup.GET("/pushgateway", builder.ListPushgatewayMetrics)
 	apiGroup.GET("/pushgateway/prometheus/metrics", builder.PushgatewayMetrics)

diff --git a/pkg/pushgateway/metric.go b/pkg/pushgateway/metric.go
@@ -63,6 +63,11 @@ func (s *Service) DeleteMetricByID(ctx context.Context, id string) error {
 	return s.store.DeleteMetricByID(ctx, id)
 }
 
+func (s *Service) DeleteAllPushgatewayMetrics(ctx context.Context) error {
+	s.logger.Debug("deleting all push gateway metrics")
+	return s.store.DeleteAllPushgatewayMetrics(ctx)
+}
+
 func (s *Service) CleanPushgatewayMetrics(ctx context.Context) {
 	numberMetricsDeleted, err := s.store.CleanPushgatewayMetrics(ctx)
 	if err != nil {

diff --git a/pkg/pushgateway/service.go b/pkg/pushgateway/service.go
@@ -16,6 +16,7 @@ type Store interface {
 	DeleteMetricsByName(ctx context.Context, name string) error
 	DeleteMetricByID(ctx context.Context, id string) error
 	CleanPushgatewayMetrics(ctx context.Context) (int64, error)
+	DeleteAllPushgatewayMetrics(ctx context.Context) error
 }
 
 type Service struct {