GORM GEN生成代码自定义方法和表名

2023-11-21 20:56:48

GORM GEN是一个强大的go ORM框架，基于gorm，支持多种数据库，包括MySQL、SQLServer、PostgreSQL以及Clickhouse等。GORM GEN通过代码生成的方式，生成安全且友好的ORM代码，同时也支持类似Java Mybatis的动态SQL代码生成。

在这个快速教程中，我们将向您展示如何使用GORM GEN自定义方法和表名。

自定义方法

GORM GEN允许您自定义模型方法。这在您需要在模型上实现自己的自定义逻辑时非常有用。

要自定义方法，您需要在模型文件中添加method标签。该标签接受一个函数作为参数，该函数将作为模型方法实现。

例如，以下代码演示了如何为User模型添加FullName方法：

type User struct {
	ID        uint   `gorm:"primary_key"`
	FirstName string `gorm:"size:255;not null"`
	LastName  string `gorm:"size:255;not null"`

	// 自定义方法
	method:func() string{
		return fmt.Sprintf("%s %s", FirstName, LastName)
	}
}

现在，您可以使用FullName方法来获取用户的全名：

user := User{FirstName: "John", LastName: "Doe"}
fmt.Println(user.FullName()) // 输出: "John Doe"

自定义表名

GORM GEN允许您自定义模型的表名。这在您需要使用与模型名称不同的表名时非常有用。

要自定义表名，您需要在模型文件中添加table标签。该标签接受一个字符串作为参数，该字符串将用作表的名称。

例如，以下代码演示了如何将User模型的表名设置为users：

type User struct {
	ID        uint   `gorm:"primary_key"`
	FirstName string `gorm:"size:255;not null"`
	LastName  string `gorm:"size:255;not null"`

	// 自定义表名
	table: users
}

现在，GORM GEN将使用users作为User模型的表名。