解决 gorm (postgres) 自增主键自定义类型的问题
在使用 gorm 与 postgres 时，如果将主键设置为自定义类型（例如 bigint），表自动迁移时无法创建自增。这是因为 gorm 无法识别自定义类型的自增属性。
通过调试 gorm 和 postgres 驱动器的代码，找到了解决办法：
在 bigint 类型中添加 gormdbdatatype 方法，用于指定表字段的数据类型：
func (bigint) gormdbdatatype(db *gorm.db, field *schema.field) string { switch db.dialector.name() { case "mysql", "sqlite": return "bigint" case "postgres": if field.autoincrement { return "bigserial" } return "bigint" } return "bigint" }
登录后复制
gormdbdatatype 方法根据数据库类型返回适当的数据类型。对于 postgres，如果字段是自增的，则返回 “berial”，否则返回 “bigint”。
修复后的代码：
type GVA_MODEL struct { ID BigInt `gorm:"primaryKey;autoIncrement;type:bigint;size:64;->" form:"id" json:"id" query:"id"` // 主键ID // ... } type BigInt json.Number // ... (Scan、Value 方法与原代码相同) func (BigInt) GormDBDataType(db *gorm.DB, field *schema.Field) string { switch db.Dialector.Name() { case "mysql", "sqlite": return "bigint" case "postgres": if field.AutoIncrement { return "bigserial" } return "bigint" } return "bigint" }
登录后复制
通过这个修复，在自动迁移时，postgres 将正确地创建自定义类型 bigint 的主键并将其设置为自增。
以上就是Gorm (Postgres) 自增主键自定义类型如何解决？的详细内容，更多请关注php中文网其它相关文章！
微信扫一扫打赏
支付宝扫一扫打赏
