sql/postgres: more flexiable support for udf types (#2920)

ariga · Jul 3, 2024 · 1f36577 · 1f36577
1 parent 2cdfe29
commit 1f36577
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/sql/postgres/convert.go b/sql/postgres/convert.go
@@ -288,13 +288,16 @@ func columnType(c *columnDesc) (schema.Type, error) {
 	case typeOID, typeRegClass, typeRegCollation, typeRegConfig, typeRegDictionary, typeRegNamespace,
 		typeRegOper, typeRegOperator, typeRegProc, typeRegProcedure, typeRegRole, typeRegType:
 		typ = &OIDType{T: t}
-	case TypeUserDefined:
-		typ = &UserDefinedType{T: c.fmtype}
 	case typeAny, typeAnyElement, typeAnyArray, typeAnyNonArray, typeAnyEnum, typeInternal,
 		typeRecord, typeTrigger, typeEventTrigger, typeVoid, typeUnknown:
 		typ = &PseudoType{T: t}
+	// TypeUserDefined or any other base type.
 	default:
-		typ = &schema.UnsupportedType{T: t}
+		ft := c.fmtype
+		if ft == "" {
+			ft = t
+		}
+		typ = &UserDefinedType{T: ft}
 	}
 	switch c.typtype {
 	case "d", "e":