feat: 迁移预警规则管理和历史分类查询

internal/server/handler/alert_rule.go

@@ -0,0 +1,89 @@
+package handler
+
+import (
+	"github.com/gin-gonic/gin"
+	"github.com/go-kratos/kratos/v3/errors"
+	"kra/internal/service/business"
+	"kra/internal/service/dto"
+	"strconv"
+)
+
+type AlertRule struct{ service *business.AlertRuleService }
+
+func NewAlertRule(service *business.AlertRuleService) *AlertRule { return &AlertRule{service: service} }
+func (h *AlertRule) List(c *gin.Context) {
+	var req dto.AlertRuleQuery
+	if err := c.ShouldBindQuery(&req); err != nil {
+		Fail(c, "查询参数不正确")
+		return
+	}
+	v, err := h.service.List(c.Request.Context(), &req)
+	if err != nil {
+		Fail(c, errors.FromError(err).Message)
+		return
+	}
+	OKWithData(c, v)
+}
+func (h *AlertRule) Types(c *gin.Context) {
+	v, err := h.service.Types(c.Request.Context())
+	if err != nil {
+		Fail(c, errors.FromError(err).Message)
+		return
+	}
+	OKWithData(c, v)
+}
+func (h *AlertRule) Find(c *gin.Context) {
+	id, err := strconv.ParseUint(c.Param("id"), 10, 64)
+	if err != nil {
+		Fail(c, "规则 ID 不正确")
+		return
+	}
+	v, err := h.service.Find(c.Request.Context(), id)
+	if err != nil {
+		Fail(c, errors.FromError(err).Message)
+		return
+	}
+	OKWithData(c, v)
+}
+func (h *AlertRule) Save(c *gin.Context) {
+	var req dto.AlertRuleRequest
+	if err := c.ShouldBindJSON(&req); err != nil {
+		Fail(c, "预警规则参数不正确")
+		return
+	}
+	id, err := h.service.Save(c.Request.Context(), &req)
+	if err != nil {
+		Fail(c, errors.FromError(err).Message)
+		return
+	}
+	OKWithData(c, gin.H{"id": id})
+}
+func (h *AlertRule) Delete(c *gin.Context) {
+	id, err := strconv.ParseUint(c.Param("id"), 10, 64)
+	if err != nil {
+		Fail(c, "规则 ID 不正确")
+		return
+	}
+	if err := h.service.Delete(c.Request.Context(), id); err != nil {
+		Fail(c, errors.FromError(err).Message)
+		return
+	}
+	OK(c)
+}
+func (h *AlertRule) Toggle(c *gin.Context) {
+	id, err := strconv.ParseUint(c.Param("id"), 10, 64)
+	if err != nil {
+		Fail(c, "规则 ID 不正确")
+		return
+	}
+	enabled, err := strconv.Atoi(c.Query("enabled"))
+	if err != nil {
+		Fail(c, "启用状态参数不正确")
+		return
+	}
+	if err := h.service.Toggle(c.Request.Context(), id, enabled); err != nil {
+		Fail(c, errors.FromError(err).Message)
+		return
+	}
+	OK(c)
+}

internal/server/handler/provider.go

@@ -8,6 +8,7 @@ var ProviderSet = wire.NewSet(
 	NewBusiness,
 	NewBusinessPassword,
 	NewBusinessPromote,
+	NewAlertRule,
 	NewAuthority, NewMenu, NewAPI, NewPermission, NewOrganization,
 	NewAnnouncement, NewEmail, NewPayment, NewTask, NewMedia, NewAudit,
 	NewExport, NewVersion, NewDictionary, NewParameter, NewAPIToken,

internal/server/handler/set.go

@@ -1,6 +1,7 @@
 package handler
 
 type Set struct {
+	AlertRule         *AlertRule
 	BusinessPromote   *BusinessPromote
 	Business          *Business
 	BusinessPassword  *BusinessPassword