用泛型给Gorm赋予MyBatis的灵魂

2023-12-27 09:31:02

泛型让Gorm更强大

在Go 1.18版本中，泛型作为一项备受期待的新特性终于正式登场了。这无疑为Go语言的开发带来了新的可能性。

泛型可以让我们在编写代码时使用更通用的类型参数，从而减少代码的重复和提高代码的可重用性。

Gorm作为Go语言中一款优秀的ORM框架，一直深受开发者的喜爱。但在Gorm中，我们经常需要对不同的数据模型进行分页查询。

例如，我们有以下两个数据模型：

type User struct {
    ID       uint64
    Username string
    Password string
}

type Order struct {
    ID        uint64
    UserID    uint64
    ProductID uint64
    Quantity  int
}

如果我们想对这两个数据模型进行分页查询，我们需要分别编写两段代码：

// 查询所有用户
func GetAllUsers(page int, pageSize int) ([]User, error) {
    var users []User
    err := db.Limit(pageSize).Offset((page - 1) * pageSize).Find(&users).Error
    return users, err
}

// 查询所有订单
func GetAllOrders(page int, pageSize int) ([]Order, error) {
    var orders []Order
    err := db.Limit(pageSize).Offset((page - 1) * pageSize).Find(&orders).Error
    return orders, err
}

从上面的代码中，我们可以看到，这两段代码非常相似，只是数据模型不同。

如果我们使用泛型，就可以将这两段代码合并为一段：

func GetAll[T any](page int, pageSize int) ([]T, error) {
    var data []T
    err := db.Limit(pageSize).Offset((page - 1) * pageSize).Find(&data).Error
    return data, err
}

这样，我们就只需要编写一段代码就可以对不同的数据模型进行分页查询了。

性能评估

在实际项目中，我们经常需要对大量数据进行分页查询。因此，分页查询的性能非常重要。

为了评估反射和泛型的性能，我们使用以下代码对一个包含100万条数据的表进行分页查询：

// 使用反射进行分页查询
func GetAllWithReflect(page int, pageSize int) ([]interface{}, error) {
    var data []interface{}
    err := db.Limit(pageSize).Offset((page - 1) * pageSize).Find(&data).Error
    return data, err
}

// 使用泛型进行分页查询
func GetAllWithGeneric[T any](page int, pageSize int) ([]T, error) {
    var data []T
    err := db.Limit(pageSize).Offset((page - 1) * pageSize).Find(&data).Error
    return data, err
}

我们使用以下代码对反射和泛型的性能进行对比：

// 性能对比
func main() {
    // 使用反射进行分页查询
    start := time.Now()
    _, err := GetAllWithReflect(1, 1000)
    if err != nil {
        log.Fatal(err)
    }
    elapsed := time.Since(start)
    log.Printf("反射耗时：%v\n", elapsed)

    // 使用泛型进行分页查询
    start = time.Now()
    _, err = GetAllWithGeneric[User](1, 1000)
    if err != nil {
        log.Fatal(err)
    }
    elapsed = time.Since(start)
    log.Printf("泛型耗时：%v\n", elapsed)
}