فهرست منبع

refactor(admin): 优化 WafLog 查询逻辑

- 使用 errors.Is 替代直接比较错误值,提高代码的可读性和可维护性
- 优化查询逻辑,处理记录不存在的情况
-调整代码格式,提高可读性
fusu 12 ساعت پیش
والد
کامیت
772e78bb6a
1فایلهای تغییر یافته به همراه3 افزوده شده و 2 حذف شده
  1. 3 2
      internal/repository/admin/waflog.go

+ 3 - 2
internal/repository/admin/waflog.go

@@ -2,6 +2,7 @@ package admin
 
 import (
 	"context"
+	"errors"
 	"fmt"
 	"math"
 	"strings"
@@ -421,7 +422,7 @@ func (r *wafLogRepository) exportFromSingleTable(ctx context.Context, req adminA
 	
 	// 5. 执行查询
 	if err := query.Find(&res).Error; err != nil {
-		if err == gorm.ErrRecordNotFound {
+		if errors.Is(err, gorm.ErrRecordNotFound) {
 			return []model.WafLogWithGatewayIP{}, nil
 		}
 		return nil, err
@@ -435,7 +436,7 @@ func (r *wafLogRepository) exportFromMultipleTables(ctx context.Context, req adm
 	var allResults []model.WafLogWithGatewayIP
 	
 	for _, tableName := range tableNames {
-		tableResults, err := r.exportFromSingleTable(ctx, req, tableName, 0, 0)
+		tableResults, err := r.exportFromSingleTable(ctx, req,  tableName, 0, 0)
 		if err != nil {
 			return nil, fmt.Errorf("查询表 %s 失败: %v", tableName, err)
 		}