gorm中动态使用表名

用户表(前台、后台)

type User struct {
	ID        int            `gorm:"primaryKey;autoIncrement"`
	Name      sql.NullString `gorm:"default:'隔壁老王'"`
	Age       uint8          `gorm:"default:55"`
	UUID      uuid.UUID
	CreatedAt time.Time      `gorm:"autoCreateTime"`
	Deleted   gorm.DeletedAt `gorm:"autoDeleteTime"`
}
type AdminUser struct {
	ID        int            `gorm:"primaryKey;autoIncrement"`
	Name      sql.NullString `gorm:"default:'隔壁老王'"`
	Age       uint8          `gorm:"default:55"`
	UUID      uuid.UUID
	CreatedAt time.Time      `gorm:"autoCreateTime"`
	Deleted   gorm.DeletedAt `gorm:"autoDeleteTime"`
}


type DynamicUser interface {
	IsAdmin() bool
}

func (*User) IsAdmin() bool {
	return true
}
func (*AdminUser) IsAdmin() bool {
	return false
}

动态获取表名

func UserTable(u DynamicUser) func(tx *gorm.DB) *gorm.DB {
	return func(tx *gorm.DB) *gorm.DB {
		if u.IsAdmin() {
			return tx.Table("users")
		}
		return tx.Table("admin_users")
	}
}

测试

type APIUser struct {
	ID int
	Name string
	Age int
}

// TableName 不支持动态变化,它会被缓存下来以便后续使用。想要使用动态表名,你可以使用 Scopes,例如:
var apiUser APIUser
var user DynamicUser
user = &User{}

db.Scopes(UserTable(user)).Scan(&apiUser)
fmt.Println(apiUser)

user = &AdminUser{}
db.Scopes(UserTable(user)).Scan(&apiUser)
fmt.Println(apiUser)

posted @   专职  阅读(2871)  评论(0编辑  收藏  举报
编辑推荐:
· 一个奇形怪状的面试题:Bean中的CHM要不要加volatile?
· [.NET]调用本地 Deepseek 模型
· 一个费力不讨好的项目,让我损失了近一半的绩效!
· .NET Core 托管堆内存泄露/CPU异常的常见思路
· PostgreSQL 和 SQL Server 在统计信息维护中的关键差异
阅读排行:
· DeepSeek “源神”启动!「GitHub 热点速览」
· 我与微信审核的“相爱相杀”看个人小程序副业
· 上周热点回顾(2.17-2.23)
· 如何使用 Uni-app 实现视频聊天(源码,支持安卓、iOS)
· spring官宣接入deepseek,真的太香了~
点击右上角即可分享
微信分享提示