如何用同一个GORM结构体创建两张同结构不同名的SQLite表

问题背景 #

我通过GORM创建两张结构完全一致的SQLite表，原本采用嵌套结构体的方式：定义Reservation结构体并指定默认表名reservations，再定义Record结构体嵌套Reservation并指定表名records。但实际操作时，用Reservation结构体查询records表无结果，只有Record结构体能正常查询。希望找到只用Reservation结构体就能创建并操作两张不同名同结构表的方法。

解决方案 #

方法1：直接使用Table()指定表名（推荐） #

无需额外定义结构体，在创建表和查询时通过Table()方法指定目标表名，完全复用Reservation的结构：

// 定义基础结构体
type Reservation struct {
    ID string
    // 其他字段...
}

// 设置默认表名
func (Reservation) TableName() string {
    return "reservations"
}

// 创建两张表并查询示例
func main() {
    db, err := gorm.Open(sqlite.Open("test.db"), &gorm.Config{})
    if err != nil {
        panic("failed to connect database")
    }

    // 创建默认的reservations表
    db.AutoMigrate(&Reservation{})
    // 创建records表，通过Table指定表名，复用Reservation的结构
    db.Table("records").AutoMigrate(&Reservation{})

    // 查询records表数据，用Reservation接收结果
    var res Reservation
    db.Table("records").Order("last_activity_at DESC").First(&res, "id = ?", "your-id")
}

方法2：动态表名（实例级控制） #

修改Reservation结构体，添加一个不映射到数据库的字段来动态指定表名，适合需要频繁切换表名的场景：

type Reservation struct {
    ID string
    // 其他字段...
    TableName string `gorm:"-"` // gorm:"-"标记该字段不生成数据库列
}

// 根据实例的TableName字段返回表名，默认使用reservations
func (r Reservation) TableName() string {
    if r.TableName != "" {
        return r.TableName
    }
    return "reservations"
}

// 创建与查询示例
func main() {
    db, err := gorm.Open(sqlite.Open("test.db"), &gorm.Config{})
    if err != nil {
        panic("failed to connect database")
    }

    // 创建默认表
    db.AutoMigrate(&Reservation{})
    // 创建records表，通过实例指定表名
    db.AutoMigrate(&Reservation{TableName: "records"})

    // 查询records表
    var res Reservation
    res.TableName = "records"
    db.Order("last_activity_at DESC").First(&res, "id = ?", "your-id")
}

方法3：类型别名复用结构 #

通过类型别名创建新的结构体类型，共享Reservation的字段结构，同时各自指定表名：

type Reservation struct {
    ID string
    // 其他字段...
}

func (Reservation) TableName() string {
    return "reservations"
}

// 定义类型别名，结构与Reservation完全一致
type Record Reservation

func (Record) TableName() string {
    return "records"
}

// 创建与查询示例
func main() {
    db, err := gorm.Open(sqlite.Open("test.db"), &gorm.Config{})
    if err != nil {
        panic("failed to connect database")
    }

    // 创建两张表
    db.AutoMigrate(&Reservation{}, &Record{})

    // 查询records表并转成Reservation类型
    var rec Record
    db.Order("last_activity_at DESC").First(&rec, "id = ?", "your-id")
    res := Reservation(rec) // 类型转换直接复用数据
}

内容的提问来源于stack exchange，提问作者DD Jin