|
@@ -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)
|
|
|
}
|