Golang中GORM的数据库错误处理指南

Golang中GORM的数据库错误处理指南 有没有人处理过在Gorm中处理数据库错误的情况？ 在我的代码中有两种情况需要利用基本错误：约束或UUID验证。 （例如：SQL错误 [8169] [S0002]：从字符串转换为唯一标识符时失败。） 当我使用Gorm的模型查找、原始扫描方法时，在正常运行中我能得到预期的结果，但在异常运行中，result.Error并不包含SQL错误（result.Error为nil），例如，当我尝试违反约束插入数据，或者使用无效的UUID进行查询时。当我在那里运行调试查询时，会在控制台看到这个错误。 另一方面，当我使用Exec（也包括sql.Exec）时，在异常运行中会得到一个错误，但正常运行的结果不包含，或者我无法获取到预期的值。


更多关于Golang中GORM的数据库错误处理指南的实战教程也可以访问 https://www.itying.com/category-94-b0.html

2 回复

很抱歉带来了麻烦——在我提取代码时，我找到了一个解决方案。这是 Exec 和 Find 方法的组合。

func main() {
    fmt.Println("hello world")
}

更多关于Golang中GORM的数据库错误处理指南的实战系列教程也可以访问 https://www.itying.com/category-94-b0.html


在Golang中使用GORM处理数据库错误时，需要根据不同的操作类型采用不同的错误处理策略。以下是针对您提到的两种情况的处理示例：

1. 使用GORM模型方法时的错误处理

当使用Find、Create、Save等方法时，错误通常包含在result.Error中，但某些数据库错误可能需要检查原始错误：

package main

import (
    "errors"
    "fmt"
    "gorm.io/driver/sqlserver"
    "gorm.io/gorm"
    "regexp"
)

type User struct {
    ID   string `gorm:"type:uniqueidentifier;primaryKey"`
    Name string `gorm:"unique"`
}

func main() {
    dsn := "sqlserver://username:password@localhost:1433?database=testdb"
    db, err := gorm.Open(sqlserver.Open(dsn), &gorm.Config{})
    if err != nil {
        panic(err)
    }

    // 示例1：处理UUID转换错误
    user := User{ID: "invalid-uuid", Name: "test"}
    result := db.Create(&user)
    
    if result.Error != nil {
        // 检查是否为UUID转换错误
        if isUUIDConversionError(result.Error) {
            fmt.Println("UUID转换错误:", result.Error)
        } else if isConstraintViolationError(result.Error) {
            fmt.Println("约束违反错误:", result.Error)
        } else {
            fmt.Println("其他数据库错误:", result.Error)
        }
    }

    // 示例2：使用Find方法时的错误处理
    var foundUser User
    result = db.Where("id = ?", "invalid-uuid").Find(&foundUser)
    
    if result.Error != nil {
        fmt.Println("查询错误:", result.Error)
    }
    
    // 检查是否找到记录
    if result.RowsAffected == 0 {
        fmt.Println("未找到记录")
    }
}

// 检查是否为UUID转换错误
func isUUIDConversionError(err error) bool {
    // SQL Server错误码8169表示UUID转换失败
    pattern := `SQL.*\[8169\].*unique identifier`
    matched, _ := regexp.MatchString(pattern, err.Error())
    return matched
}

// 检查是否为约束违反错误
func isConstraintViolationError(err error) bool {
    // SQL Server约束违反错误码通常为2627或2601
    pattern := `SQL.*\[(2627|2601)\]`
    matched, _ := regexp.MatchString(pattern, err.Error())
    return matched
}

2. 使用Exec和Raw SQL时的错误处理

当使用Exec或Raw方法时，需要直接检查返回的错误：

package main

import (
    "database/sql"
    "fmt"
    "gorm.io/driver/sqlserver"
    "gorm.io/gorm"
)

func main() {
    dsn := "sqlserver://username:password@localhost:1433?database=testdb"
    db, err := gorm.Open(sqlserver.Open(dsn), &gorm.Config{})
    if err != nil {
        panic(err)
    }

    // 示例1：使用Exec插入数据
    result := db.Exec(`
        INSERT INTO users (id, name) 
        VALUES (CAST(? AS UNIQUEIDENTIFIER), ?)
    `, "invalid-uuid", "test")
    
    if result.Error != nil {
        fmt.Println("Exec执行错误:", result.Error)
    } else {
        rowsAffected, _ := result.RowsAffected()
        fmt.Printf("影响行数: %d\n", rowsAffected)
    }

    // 示例2：使用Raw查询并扫描结果
    var count int64
    err = db.Raw("SELECT COUNT(*) FROM users WHERE id = CAST(? AS UNIQUEIDENTIFIER)", 
        "invalid-uuid").Scan(&count).Error
    
    if err != nil {
        fmt.Println("Raw查询错误:", err)
    } else {
        fmt.Printf("查询结果: %d\n", count)
    }

    // 示例3：使用sql.Exec直接操作
    sqlDB, err := db.DB()
    if err != nil {
        panic(err)
    }
    
    _, err = sqlDB.Exec(`
        INSERT INTO users (id, name) 
        VALUES (CAST(@p1 AS UNIQUEIDENTIFIER), @p2)
    `, sql.Named("p1", "invalid-uuid"), sql.Named("p2", "test"))
    
    if err != nil {
        fmt.Println("sql.Exec错误:", err)
    }
}

3. 统一错误处理封装

建议创建一个统一的错误处理工具函数：

package main

import (
    "errors"
    "fmt"
    "gorm.io/gorm"
    "strings"
)

type DBErrorHandler struct{}

func (h *DBErrorHandler) Handle(err error) error {
    if err == nil {
        return nil
    }

    errStr := err.Error()
    
    // 处理SQL Server特定错误
    switch {
    case strings.Contains(errStr, "[8169]"):
        return errors.New("UUID格式无效")
    case strings.Contains(errStr, "[2627]"):
        return errors.New("违反唯一约束")
    case strings.Contains(errStr, "[2601]"):
        return errors.New("违反唯一索引约束")
    case strings.Contains(errStr, "[547]"):
        return errors.New("违反外键约束")
    case errors.Is(err, gorm.ErrRecordNotFound):
        return errors.New("记录不存在")
    default:
        return err
    }
}

// 使用示例
func createUser(db *gorm.DB, user *User) error {
    result := db.Create(user)
    
    handler := &DBErrorHandler{}
    if err := handler.Handle(result.Error); err != nil {
        return fmt.Errorf("创建用户失败: %w", err)
    }
    
    return nil
}

4. 使用GORM的Hook进行错误处理

可以在模型中定义Hook方法来处理特定错误：

package main

import (
    "gorm.io/gorm"
)

type Product struct {
    ID    string `gorm:"type:uniqueidentifier;primaryKey"`
    Code  string `gorm:"unique"`
    Price uint
}

func (p *Product) BeforeCreate(tx *gorm.DB) error {
    // 在创建前验证UUID格式
    if !isValidUUID(p.ID) {
        return gorm.ErrInvalidData
    }
    return nil
}

func (p *Product) AfterCreate(tx *gorm.DB) error {
    // 检查创建后的错误
    if tx.Error != nil {
        return handleDBError(tx.Error)
    }
    return nil
}

func isValidUUID(id string) bool {
    // 实现UUID验证逻辑
    return len(id) == 36 // 简化示例
}

func handleDBError(err error) error {
    // 错误处理逻辑
    return err
}

这些示例展示了在GORM中处理数据库错误的不同方法，特别是针对UUID转换错误和约束违反错误的处理。关键是要根据具体的数据库驱动和错误信息进行适当的错误类型判断和处理。

回到顶部