Gorm（六）错误处理 & RowsAffected

在 Gorm 中，错误处理和 RowsAffected 是数据库操作后不可或缺的部分，用于判断操作结果、排查问题并确保业务逻辑正确执行。以下是详细说明：

一、错误处理（Error Handling）

Gorm 的所有数据库操作方法（如 Create、Find、Update、Delete 等）都会返回一个 *gorm.Result 结构体（或直接返回 error），其中包含操作过程中可能出现的错误。正确处理这些错误是保证程序健壮性的关键。

1. 常见错误类型

Gorm 定义了一些常见的错误常量，同时也会返回数据库驱动本身的错误（如连接失败、SQL 语法错误等）：

• gorm.ErrRecordNotFound：查询单条记录时未找到（如 First、Last、Take）。
• 数据库驱动错误：如连接超时、主键冲突、字段不存在等（具体错误信息由数据库返回）。

2. 错误处理示例

（1）查询单条记录的错误处理

var user User
result := db.First(&user, 100) // 查询 ID=100 的用户

// 处理错误
if result.Error != nil {
  if errors.Is(result.Error, gorm.ErrRecordNotFound) {
    // 明确处理"记录不存在"的场景
    fmt.Println("用户不存在")
  } else {
    // 其他错误（如数据库连接失败、SQL 错误等）
    fmt.Printf("查询失败：%v\n", result.Error)
  }
}

（2）创建/更新/删除的错误处理

// 创建记录
user := User{Name: "Alice", Email: "alice@example.com"}
result := db.Create(&user)
if result.Error != nil {
  // 处理错误（如唯一索引冲突、字段验证失败等）
  fmt.Printf("创建失败：%v\n", result.Error)
  return
}

// 更新记录
result = db.Model(&User{}).Where("id = ?", 1).Update("name", "Bob")
if result.Error != nil {
  fmt.Printf("更新失败：%v\n", result.Error)
  return
}

（3）批量操作的错误处理

var users []User
// 批量查询
result := db.Where("age > 30").Find(&users)
if result.Error != nil {
  fmt.Printf("批量查询失败：%v\n", result.Error)
  return
}

3. 关键原则

• 必须检查错误 ：任何数据库操作后都应检查 result.Error，避免忽略潜在问题（如网络中断连、权限不足）。
• 区分错误类型 ：使用 errors.Is 或 errors.As 区分特定错误（如 gorm.ErrRecordNotFound），针对性处理业务逻辑。
• 日志记录：生产环境中，建议将错误详细日志记录（如错误信息、堆栈跟踪），便于排查问题。

二、RowsAffected：受影响的行数

*gorm.Result 结构体中的 RowsAffected 字段表示操作影响的记录行数（类型为 int64），用于判断操作是否实际生效（如更新是否命中记录、删除是否成功删除数据等）。

1. 常见使用场景

（1）判断更新/删除是否生效

// 更新操作
result := db.Model(&User{}).Where("id = ?", 1).Update("age", 20)
if result.Error != nil {
  // 处理错误
} else if result.RowsAffected == 0 {
  // 无记录被更新（如 ID=1 的用户不存在）
  fmt.Println("未找到要更新的用户")
} else {
  fmt.Printf("成功更新 %d 条记录\n", result.RowsAffected)
}

// 删除操作
result = db.Delete(&User{}, 1)
if result.Error != nil {
  // 处理错误
} else if result.RowsAffected == 0 {
  fmt.Println("未找到要删除的用户")
}

（2）批量操作的行数统计

// 批量删除
result := db.Where("age < 18").Delete(&User{})
if result.Error != nil {
  // 处理错误
} else {
  fmt.Printf("成功删除 %d 条未成年用户记录\n", result.RowsAffected)
}

（3）插入操作的行数验证

// 单条插入
user := User{Name: "Alice"}
result := db.Create(&user)
if result.Error != nil {
  // 处理错误
} else if result.RowsAffected == 1 {
  fmt.Println("插入成功")
}

// 批量插入
users := []User{{Name: "Bob"}, {Name: "Charlie"}}
result = db.CreateInBatches(users, 100)
if result.Error != nil {
  // 处理错误
} else {
  fmt.Printf("成功插入 %d 条记录\n", result.RowsAffected)
}

2. 注意事项

• 查询操作的 RowsAffected ：Find 等查询操作的 RowsAffected 等于返回的记录数，可用于判断是否有符合条件的记录（替代 len(users) == 0 的检查，但需注意性能，Count 更适合纯统计）。
• 软删除的 RowsAffected ：软删除（更新 deleted_at）的 RowsAffected 与物理删除一致，均为被标记的记录数。
• 事务中的 RowsAffected ：事务内的操作 RowsAffected 仅表示当前步骤的影响行数，最终是否生效取决于事务是否提交。

三、综合示例：结合错误处理与 RowsAffected

func UpdateUserAge(userID uint, newAge int) (bool, error) {
  result := db.Model(&User{}).Where("id = ?", userID).Update("age", newAge)
  
  // 先处理错误
  if result.Error != nil {
    return false, fmt.Errorf("更新失败：%w", result.Error) // 包装错误，保留堆栈
  }
  
  // 再判断影响行数
  if result.RowsAffected == 0 {
    return false, nil // 无记录被更新（非错误，业务上可能视为"未找到"）
  }
  
  return true, nil // 更新成功
}

总结

• 错误处理 ：通过检查 result.Error 判断操作是否异常，区分特定错误类型（如 ErrRecordNotFound），确保程序稳定。
• RowsAffected：通过受影响的行数判断操作是否实际生效（如更新是否命中记录、删除是否成功），是业务逻辑判断的重要依据。

二者结合使用，可全面掌控数据库操作的结果，保证业务逻辑的正确性和程序的健壮性。