如何从 gorm 中的模型获取表名？

在使用Golang的ORM库GORM时，有时我们需要获取模型对应的数据库表名。GORM提供了多种方法来实现这一点。下面我将介绍两种主要的方法来从GORM模型获取表名。

1. 使用模型的TableName方法

在GORM中，每个模型都可以通过实现TableName方法来指定其对应的数据库表名。如果模型没有实现这个方法，GORM会默认使用模型的结构体名称的蛇形复数形式作为表名。例如：

go
type User struct {
    gorm.Model
    Name string
}

// 默认表名是 `users`

// 自定义表名
func (User) TableName() string {
    return "my_custom_users"
}

在这个例子中，虽然默认的表名为users，但通过定义TableName方法，我们可以指定表名为my_custom_users。这个方法可以直接被调用来获取表名：

go
func main() {
    user := User{}
    tableName := user.TableName()  // 返回 "my_custom_users"
    fmt.Println(tableName)
}

2. 使用gorm库的ToTableName方法

如果你需要在不实例化模型的情况下获取表名，或者你想获取默认的表名而不调用模型的TableName方法，你可以使用ToTableName方法。这个方法属于gorm库的schema工具，可以直接从模型的类型信息中解析出表名。

go
import (
    "fmt"
    "gorm.io/gorm"
    "gorm.io/gorm/schema"
)

type Product struct {
    gorm.Model
    Code string
    Price uint
}

func main() {
    // 初始化GORM
    db, _ := gorm.Open(...) // 假设已正确配置数据库
    
    // 使用schema.Parse获取模型的schema信息
    schema, _ := schema.Parse(&Product{}, &sync.Map{}, schema.NamingStrategy{})

    // 获取表名
    tableName := schema.Table
    fmt.Println(tableName)  // 默认会输出 "products"
}

这种方法特别适合在还没有数据库实例的情况下获取表名，或者在编写通用函数需要操作表名时使用。

总结

根据具体情况选择适合的方法。如果你已经有模型的实例，使用TableName方法非常直接方便。如果你需要在全局范围内或在没有模型实例的情况下获取表名，gorm/schema的ToTableName方法将是一个好的选择。