Golang

关注公众号 jb51net

关闭
首页 > 脚本专栏 > Golang > Golang Gin表名前缀

Golang中Gin数据库表名前缀的三种方法

作者:gskyi

本文主要介绍了Golang中Gin数据库表名前缀的三种方法,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧

在使用 GORM 时,如果你的数据库表名有前缀(例如:prefix_tcm_operation_logs),你可以通过自定义 TableName 方法来处理表前缀。TableName 方法是 GORM 用来获取数据库表名的地方,你可以在其中动态地添加前缀。

1. 方法一:通过 TableName 自定义表名

如果你希望所有表都有相同的前缀,可以在每个模型中手动实现 TableName 方法,动态拼接前缀。

1.1 实现 TableName 方法

假设数据库表前缀为 prefix_,你可以在模型中定义 TableName 方法来返回带有前缀的表名。

package models

import (
	"fmt"
	"time"
	"github.com/jinzhu/gorm"
)

type TcmOperationLogs struct {
	ID          uint64    `gorm:"primaryKey;autoIncrement;column:id" json:"id"`
	Action      string    `gorm:"type:varchar(255);column:action" json:"action"`
	Type        string    `gorm:"type:varchar(255);default:'';column:type" json:"type"`
	UserID      string    `gorm:"type:varchar(255);column:user_id" json:"user_id"`
	DoctorID    string    `gorm:"type:varchar(255);column:doctor_id" json:"doctor_id"`
	AdminID     string    `gorm:"type:varchar(255);column:admin_id" json:"admin_id"`
	Description string    `gorm:"type:text;column:description" json:"description"`
	CreatedAt   time.Time `gorm:"type:datetime;column:created_at" json:"created_at"`
	UpdatedAt   time.Time `gorm:"type:datetime;column:updated_at" json:"updated_at"`
}

// TableName 方法返回带前缀的表名
func (TcmOperationLogs) TableName() string {
	// 假设前缀为 'prefix_'
	return "prefix_tcm_operation_logs"
}

这样,每次 GORM 使用 TcmOperationLogs 模型时,都会自动使用 prefix_tcm_operation_logs 作为表名。

2. 方法二:全局前缀配置

如果你希望 GORM 中所有的表都使用相同的前缀,可以通过全局配置来自动为所有模型添加前缀。

2.1 使用 GORM 的 NamingStrategy 配置

在 GORM v2 中,可以通过设置全局的 NamingStrategy 来为所有表名添加前缀。NamingStrategy 允许你定制 GORM 的表名生成规则。

package main

import (
	"fmt"
	"log"
	"gorm.io/driver/mysql"
	"gorm.io/gorm"
	"gorm.io/gorm/schema"
)

// 定义模型
type TcmOperationLogs struct {
	ID          uint64    `gorm:"primaryKey;autoIncrement;column:id" json:"id"`
	Action      string    `gorm:"type:varchar(255);column:action" json:"action"`
	Type        string    `gorm:"type:varchar(255);default:'';column:type" json:"type"`
	UserID      string    `gorm:"type:varchar(255);column:user_id" json:"user_id"`
	DoctorID    string    `gorm:"type:varchar(255);column:doctor_id" json:"doctor_id"`
	AdminID     string    `gorm:"type:varchar(255);column:admin_id" json:"admin_id"`
	Description string    `gorm:"type:text;column:description" json:"description"`
	CreatedAt   time.Time `gorm:"type:datetime;column:created_at" json:"created_at"`
	UpdatedAt   time.Time `gorm:"type:datetime;column:updated_at" json:"updated_at"`
}

func main() {
	// 连接到 MySQL 数据库
	dsn := "root:password@tcp(127.0.0.1:3306)/your_db?charset=utf8mb4&parseTime=True&loc=Local"
	db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{
		NamingStrategy: schema.NamingStrategy{
			TablePrefix: "prefix_", // 为所有表添加前缀
		},
	})
	if err != nil {
		log.Fatalf("failed to connect to database: %v", err)
	}

	// 自动迁移
	err = db.AutoMigrate(&TcmOperationLogs{})
	if err != nil {
		log.Fatalf("failed to migrate database: %v", err)
	}

	// 之后所有模型的表名都会自动带上前缀
	fmt.Println("Database connected and migrations completed")
}

3. 方法三:使用自定义 DB 和 Model 函数

另一种方法是通过自定义数据库操作函数来动态添加前缀。这种方法不常见,但在一些特殊场景下可能会用到。

package main

import (
	"fmt"
	"log"
	"gorm.io/driver/mysql"
	"gorm.io/gorm"
)

// 定义模型
type TcmOperationLogs struct {
	ID          uint64    `gorm:"primaryKey;autoIncrement;column:id" json:"id"`
	Action      string    `gorm:"type:varchar(255);column:action" json:"action"`
	Type        string    `gorm:"type:varchar(255);default:'';column:type" json:"type"`
	UserID      string    `gorm:"type:varchar(255);column:user_id" json:"user_id"`
	DoctorID    string    `gorm:"type:varchar(255);column:doctor_id" json:"doctor_id"`
	AdminID     string    `gorm:"type:varchar(255);column:admin_id" json:"admin_id"`
	Description string    `gorm:"type:text;column:description" json:"description"`
	CreatedAt   time.Time `gorm:"type:datetime;column:created_at" json:"created_at"`
	UpdatedAt   time.Time `gorm:"type:datetime;column:updated_at" json:"updated_at"`
}

func main() {
	// 连接到 MySQL 数据库
	dsn := "root:password@tcp(127.0.0.1:3306)/your_db?charset=utf8mb4&parseTime=True&loc=Local"
	db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{})
	if err != nil {
		log.Fatalf("failed to connect to database: %v", err)
	}

	// 自定义查询
	var logs []TcmOperationLogs
	db.Table("prefix_tcm_operation_logs").Find(&logs)
	fmt.Println(logs)
}

4. 总结

通常,如果所有表都需要相同的前缀,使用 方法二 配合 NamingStrategy 会更加简洁和统一。

到此这篇关于Golang中Gin数据库表名前缀的实现的文章就介绍到这了,更多相关Golang Gin表名前缀内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!

您可能感兴趣的文章:
阅读全文