fix: default value with typecast (#211)

Co-authored-by: Aleksandr Tuliakov <tuliakov@basetrack.net>

migrator.go

@@ -478,7 +478,7 @@ func (m Migrator) ColumnTypes(value interface{}) (columnTypes []gorm.ColumnType,
  }
 
  if column.DefaultValueValue.Valid {
- column.DefaultValueValue.String = regexp.MustCompile(`'?(.*\b|)'?:+[\w\s]+$`).ReplaceAllString(column.DefaultValueValue.String, "$1")
+ column.DefaultValueValue.String = parseDefaultValueValue(column.DefaultValueValue.String)
  }
 
  if datetimePrecision.Valid {
@@ -784,3 +784,7 @@ func (m Migrator) RenameColumn(dst interface{}, oldName, field string) error {
  m.resetPreparedStmts()
  return nil
 }
+
+func parseDefaultValueValue(defaultValue string) string {
+ return regexp.MustCompile(`^(.*?)(?:::.*)?$`).ReplaceAllString(defaultValue, "$1")
+}

migrator_test.go

@@ -0,0 +1,102 @@
+package postgres
+
+import "testing"
+
+func Test_parseDefaultValueValue(t *testing.T) {
+ type args struct {
+ defaultValue string
+ }
+ tests := []struct {
+ name string
+ args args
+ want string
+ }{
+ {
+ name: "it should works with number without colons",
+ args: args{defaultValue: "0"},
+ want: "0",
+ },
+ {
+ name: "it should works with number and two colons",
+ args: args{defaultValue: "0::int8"},
+ want: "0",
+ },
+ {
+ name: "it should works with number and three colons",
+ args: args{defaultValue: "0:::int8"},
+ want: "0",
+ },
+ {
+ name: "it should works with empty string without colons",
+ args: args{defaultValue: "''"},
+ want: "''",
+ },
+ {
+ name: "it should works with empty string with two colons",
+ args: args{defaultValue: "''::character varying"},
+ want: "''",
+ },
+ {
+ name: "it should works with empty string with three colons",
+ args: args{defaultValue: "'':::character varying"},
+ want: "''",
+ },
+ {
+ name: "it should works with string without colons",
+ args: args{defaultValue: "'field'"},
+ want: "'field'",
+ },
+ {
+ name: "it should works with string with two colons",
+ args: args{defaultValue: "'field'::character varying"},
+ want: "'field'",
+ },
+ {
+ name: "it should works with string with three colons",
+ args: args{defaultValue: "'field':::character varying"},
+ want: "'field'",
+ },
+ {
+ name: "it should works with value with two colons",
+ args: args{defaultValue: "field"},
+ want: "field",
+ },
+ {
+ name: "it should works with value without colons",
+ args: args{defaultValue: "field::character varying"},
+ want: "field",
+ },
+ {
+ name: "it should works with value with three colons",
+ args: args{defaultValue: "field:::character varying"},
+ want: "field",
+ },
+ {
+ name: "it should works with function without colons",
+ args: args{defaultValue: "now()"},
+ want: "now()",
+ },
+ {
+ name: "it should works with function with two colons",
+ args: args{defaultValue: "now()::timestamp without time zone"},
+ want: "now()",
+ },
+ {
+ name: "it should works with json without colons",
+ args: args{defaultValue: "{}"},
+ want: "{}",
+ },
+ {
+ name: "it should works with json with two colons",
+ args: args{defaultValue: "{}::jsonb"},
+ want: "{}",
+ },
+ }
+ for _, tt := range tests {
+ t.Run(tt.name, func(t *testing.T) {
+ if got := parseDefaultValueValue(tt.args.defaultValue); got != tt.want {
+ t.Errorf("parseDefaultValueValue() = %v, want %v", got, tt.want)
+ }
+ })
+ }
+}