优化

internal/server/handler/audit.go

@@ -13,9 +13,15 @@ import (
 	"github.com/gin-gonic/gin"
 )
 
-type Audit struct{ service *service.AuditService }
+type Audit struct {
+	service  *service.AuditService
+	recorder *service.AuditRecorder
+	logs     *service.LogViewerService
+}
 
-func NewAudit(service *service.AuditService) *Audit { return &Audit{service: service} }
+func NewAudit(service *service.AuditService, recorder *service.AuditRecorder, logs *service.LogViewerService) *Audit {
+	return &Audit{service: service, recorder: recorder, logs: logs}
+}
 
 func page(c *gin.Context) (int, int) {
 	value, _ := strconv.Atoi(c.Query("page"))
@@ -152,7 +158,7 @@ func (h *Audit) DeleteDataAccess(c *gin.Context) {
 }
 
 func (h *Audit) LogDates(c *gin.Context) {
-	data, err := h.service.LogDates(c.Request.Context(), c.Query("month"))
+	data, err := h.logs.LogDates(c.Request.Context(), c.Query("month"))
 	if err != nil {
 		failLogViewer(c, err)
 		return
@@ -160,7 +166,7 @@ func (h *Audit) LogDates(c *gin.Context) {
 	httpx.Write(c, httpx.CodeSuccess, data, "获取成功")
 }
 func (h *Audit) LogFiles(c *gin.Context) {
-	data, err := h.service.LogFiles(c.Request.Context(), c.Query("date"))
+	data, err := h.logs.LogFiles(c.Request.Context(), c.Query("date"))
 	if err != nil {
 		failLogViewer(c, err)
 		return
@@ -177,7 +183,7 @@ func (h *Audit) LogContent(c *gin.Context) {
 		}
 		cursor = &value
 	}
-	data, err := h.service.LogContent(c.Request.Context(), c.Query("date"), c.Query("path"), cursor)
+	data, err := h.logs.LogContent(c.Request.Context(), c.Query("date"), c.Query("path"), cursor)
 	if err != nil {
 		failLogViewer(c, err)
 		return
@@ -264,7 +270,7 @@ func (h *Audit) CreateError(c *gin.Context) {
 		httpx.Fail(c, err.Error())
 		return
 	}
-	if err := h.service.CreateErrorRequest(c.Request.Context(), &req); err != nil {
+	if err := h.recorder.CreateErrorRequest(c.Request.Context(), &req); err != nil {
 		httpx.Fail(c, "创建失败:"+err.Error())
 		return
 	}