delete custom rule

actiontech · Jul 27, 2023 · 75dd528 · 75dd528
1 parent aa530cd
commit 75dd528
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/sqle/model/rule.go b/sqle/model/rule.go
@@ -596,3 +596,17 @@ func (s *Storage) GetAllRulesByTmpNameAndProjectIdInstanceDBType(ruleTemplateNam
 	}
 	return rules, customRules, nil
 }
+
+func (s *Storage) DeleteCustomRule(ruleId string) error {
+	tx := s.db.Begin()
+	if err := tx.Where("rule_id = ?", ruleId).Delete(&CustomRule{}).Error; err != nil {
+		tx.Rollback()
+		return err
+	}
+	if err := tx.Where("rule_id = ?", ruleId).Delete(&RuleTemplateCustomRule{}).Error; err != nil {
+		tx.Rollback()
+		return err
+	}
+
+	return errors.New(errors.ConnectStorageError, tx.Commit().Error)
+}