在 GORM 中使用 Postgres 自定义数据类型

2021-12-22

项目中使用到了 postgres 的枚举类型，但是使用 GORM 自动迁移表的时候出现了

 1
 2
 3
 4
 5
 6
 7
 8
 9
10


 CREATE TABLE "user" ("id" bigserial,
                      "created_at" timestamptz,
                      "updated_at" timestamptz,
                      "deleted_at" timestamptz,
                      "username" varchar(20),unique,
                      "password" varchar(20),
                      "email" varchar(32),unique,
                      "authority" enum('root', 'one', 'two', 'three', 'four'),default:'one',
                      PRIMARY KEY ("id"))
错误: 语法错误 在 "," 或附近的 (SQLSTATE 42601)

在gorm/issues中找到了解决方案

在 pg 中创建自定义类型

1
2
3
4
5
6


CREATE TYPE authority AS ENUM (
    'root',
    'one',
    'two',
    'three',
    'four');

定义自定义的数据类型的 model

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19


import (
	"database/sql/driver"
)
type Authority string
// 'root', 'one', 'two', 'three', 'four'
const (
	root  Authority = "root"
	one   Authority = "one"
	two   Authority = "two"
	three Authority = "three"
	four  Authority = "four"
)
func (p *Authority) Scan(value interface{}) error {
	*p = Authority(value.([]byte))
	return nil
}
func (p Authority) Value() (driver.Value, error) {
	return string(p), nil
}

定义一个要使用的 Model

1
2
3
4
5
6
7


type User struct {
	global.Model
	Username  string `gorm:"type:varchar(20)" json:"username" `
	Password  string `gorm:"type:varchar(20)" json:"password"`
	Email     string `gorm:"type:varchar(32)" json:"email" `
	Authority string `gorm:"type:authority" json:"authority"`
}

现在你就可以使用 GORM 的AutoMigrate()自动迁移表了

1
2
3
4
5


err = DB.AutoMigrate(&User{})
	if err != nil {
		fmt.Println("数据库创建失败", err)
		return nil
}

#GORM #Postgres