如何让GORM自动迁移PostgreSQL时指定使用INTEGER类型

我的Go模型定义如下：

type Sales_Daily_db struct {
    Nation_shipping string
    Date             datatypes.Date
    Impressions      int `gorm:"type:integer;"`
    Clicks           int `gorm:"type:integer;"`
    Cost             float32
    ATB              float32
    OKL              float32
}

执行AutoMigrate()时，希望PostgreSQL数据库中的impressions和clicks列类型为integer，但即使添加了对应的gorm标签，这些列仍被创建为int4类型。手动尝试在标签中指定int2、int4、int8都能正常生效；指定int标签时，会被强制转为int8。请问如何修正该行为，使PostgreSQL中生成的列类型为integer？

补充说明：使用DBeaver查看数据库。

首先需要明确：在PostgreSQL中，integer是int4的官方别名，两者本质是同一个数据类型，功能完全一致，DBeaver只是默认显示了底层的类型标识符int4。

如果一定要让DBeaver显示列类型为integer，可以通过以下两种方式处理：

• 方式一：自定义GORM类型映射逻辑
  覆盖GORM PostgreSQL方言的默认类型转换，让标记了type:integer;的int字段强制映射为integer：

  import (
      "gorm.io/driver/postgres"
      "gorm.io/gorm"
      "gorm.io/gorm/schema"
  )
  
  func main() {
      dsn := "你的数据库连接串"
      db, err := gorm.Open(postgres.New(postgres.Config{DSN: dsn}), &gorm.Config{})
      if err != nil {
          panic(err)
      }
  
      // 注册自定义创建回调，修改列类型
      db.Callback().Create().Before("gorm:create").Register("custom_integer_type", func(db *gorm.DB) {
          if db.Statement.Schema == nil {
              return
          }
          for _, field := range db.Statement.Schema.Fields {
              if field.FieldType.Name() == "int" && field.Tag.Get("gorm") == "type:integer;" {
                  db.Statement.SetColumnType(field.DBName, "integer")
              }
          }
      })
  
      // 执行自动迁移
      if err := db.AutoMigrate(&Sales_Daily_db{}); err != nil {
          panic(err)
      }
  }
  

• 方式二：手动执行SQL修改列类型
  直接在PostgreSQL中执行DDL语句修改列类型，之后GORM的读写操作不受任何影响：

  ALTER TABLE sales_daily_dbs ALTER COLUMN impressions TYPE integer;
  ALTER TABLE sales_daily_dbs ALTER COLUMN clicks TYPE integer;
  

  执行后，DBeaver会将列类型显示为integer。

内容的提问来源于stack exchange，提问作者Mike Abbey