Further factoring

internal/service/db.go

@@ -1,10 +1,34 @@
 package service
 
 import (
+	"errors"
 	"github.com/uselagoon/lagoon/services/insights-handler/internal/lagoonclient"
 	"gorm.io/gorm"
 )
 
 func CreateFacts(db *gorm.DB, facts *[]lagoonclient.Fact) error {
 	return db.Create(facts).Error
 }
+
+type DeleteFactsOptions struct {
+}
+
+func DeleteFacts(db *gorm.DB, environmentId int, source string) (int64, error) {
+
+	if environmentId == 0 {
+		return 0, errors.New("EnvironmentId cannot be 0")
+	}
+
+	conditions := map[string]interface{}{
+		"environment": environmentId,
+	}
+
+	if source != "" {
+		conditions["source"] = source
+	}
+
+	result := db.Where(conditions).Delete(&lagoonclient.Fact{})
+
+	return result.RowsAffected, result.Error
+
+}

internal/service/service.go

@@ -77,24 +77,15 @@ func DeleteFactsByEnvironmentEndpoint(c *gin.Context) {
 		return
 	}
 
-	// Build conditions for deletion based on optional parameters
-	conditions := map[string]interface{}{
-		"environment": environmentID,
-	}
-
-	if source != "" {
-		conditions["source"] = source
-	}
-
 	// Delete facts based on conditions
-	result := gormDB.Where(conditions).Delete(&lagoonclient.Fact{})
-	if result.Error != nil {
+	rowsAffected, err := DeleteFacts(gormDB, environmentID, source) //gormDB.Where(conditions).Delete(&lagoonclient.Fact{})
+	if err != nil {
 		c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to delete facts from the database"})
 		return
 	}
 
 	// Check if any records were deleted
-	if result.RowsAffected == 0 {
+	if rowsAffected == 0 {
 		c.JSON(http.StatusNotFound, gin.H{"message": "No matching facts found for deletion"})
 		return
 	}