Gorm(Postgres)使用自定义类型主键时如何实现自增？

gorm(postgres)使用自定义类型主键时无法设置自增问题

使用gorm(postgres)给表设置自定义类型主键时，由于自动迁移无法为主键添加自增功能，导致程序出现问题。

问题原因在于，gorm在创建表时，会根据模型中的字段类型来判断是否需要添加自增功能。但在使用自定义类型时，gorm无法识别自定义类型的具体特性，因此无法判断是否需要添加自增。

解决方案

为了解决此问题，需要在自定义类型中实现gormdbdatatype接口。该接口用于告知gorm字段的具体数据类型，以便gorm在创建表时可以正确设置字段的类型和约束。

在提供的代码中，已对bigint自定义类型实现了gormdbdatatype接口，并根据不同的数据库驱动程序返回不同的数据类型。对于postgres数据库，如果字段为自增字段，则返回"bigserial"类型，否则返回"bigint"类型。

具体实现如下：

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接口，gorm可以识别bigint自定义类型为postgres中的"bigserial"类型，从而在自动创建表时正确添加自增约束。

以上就是Gorm(Postgres)使用自定义类型主键时如何实现自增？的详细内容，更多请关注其它相关文章！