feat: add support for UUID and PG_UUID types across Spanner migration utilities

Author: pratickchokhani

v2/spanner-common/src/main/java/com/google/cloud/teleport/v2/spanner/ddl/Column.java

@@ -164,6 +164,10 @@ private static String typeString(Type type, Integer size) {
         return Type.Code.JSON.getName();
       case PG_JSONB:
         return Type.Code.PG_JSONB.getName();
+      case UUID:
+        return Type.Code.UUID.getName();
+      case PG_UUID:
+        return Type.Code.PG_UUID.getName();
       case TOKENLIST:
         return Type.Code.TOKENLIST.getName();
       case ARRAY:
@@ -250,6 +254,10 @@ public Builder bytes() {
       return type(Type.bytes());
     }
 
+    public Builder uuid() {
+      return type(Type.uuid());
+    }
+
     public Builder timestamp() {
       return type(Type.timestamp());
     }
@@ -306,6 +314,10 @@ public Builder pgJsonb() {
       return type(Type.pgJsonb());
     }
 
+    public Builder pgUuid() {
+      return type(Type.pgUuid());
+    }
+
     public Builder max() {
       return size(-1);
     }
@@ -377,6 +389,9 @@ private static SizedType parseSpannerType(String spannerType, Dialect dialect) {
           if (spannerType.equals(Type.Code.NUMERIC.getName())) {
             return t(Type.numeric(), null);
           }
+          if (spannerType.equals(Type.Code.UUID.getName())) {
+            return t(Type.uuid(), null);
+          }
           if (spannerType.equals(Type.Code.JSON.getName())) {
             return t(Type.json(), null);
           }
@@ -432,6 +447,9 @@ private static SizedType parseSpannerType(String spannerType, Dialect dialect) {
           if (spannerType.equals(Type.Code.PG_NUMERIC.getName())) {
             return t(Type.pgNumeric(), null);
           }
+          if (spannerType.equals(Type.Code.PG_UUID.getName())) {
+            return t(Type.pgUuid(), null);
+          }
           if (spannerType.equals(Type.Code.PG_JSONB.getName())) {
             return t(Type.pgJsonb(), null);
           }

v2/spanner-common/src/main/java/com/google/cloud/teleport/v2/spanner/migrations/convertors/ChangeEventSpannerConvertor.java

@@ -111,6 +111,8 @@ public static com.google.cloud.spanner.Key changeEventToPrimaryKey(
           case STRING:
           case PG_VARCHAR:
           case PG_TEXT:
+          case UUID:
+          case PG_UUID:
             pk.append(
                 ChangeEventTypeConvertor.toString(
                     changeEvent, keyColName, /* requiredField= */ true));

v2/spanner-common/src/main/java/com/google/cloud/teleport/v2/spanner/migrations/convertors/ChangeEventTypeConvertor.java

@@ -71,6 +71,8 @@ public static Value toValue(
       case STRING:
       case PG_VARCHAR:
       case PG_TEXT:
+      case UUID:
+      case PG_UUID:
         return Value.string(toString(changeEvent, key, requiredField));
       case NUMERIC:
       case PG_NUMERIC:

v2/spanner-common/src/main/java/com/google/cloud/teleport/v2/spanner/migrations/spanner/SpannerReadUtils.java

@@ -111,6 +111,7 @@ private static void bindGoogleSqlValue(
         break;
       case STRING:
       case JSON:
+      case UUID:
         stmtBuilder.bind(bindName).to((String) value);
         break;
       case NUMERIC:
@@ -153,6 +154,7 @@ private static void bindPgValue(
       case PG_VARCHAR:
       case PG_TEXT:
       case PG_JSONB:
+      case PG_UUID:
         stmtBuilder.bind(bindName).to((String) value);
         break;
       case PG_NUMERIC:

v2/spanner-common/src/main/java/com/google/cloud/teleport/v2/spanner/type/Type.java

@@ -44,6 +44,7 @@ public final class Type implements Serializable {
   private static final Type TYPE_BYTES = new Type(Type.Code.BYTES, null, null);
   private static final Type TYPE_TIMESTAMP = new Type(Type.Code.TIMESTAMP, null, null);
   private static final Type TYPE_DATE = new Type(Type.Code.DATE, null, null);
+  private static final Type TYPE_UUID = new Type(Type.Code.UUID, null, null);
   private static final Type TYPE_ARRAY_BOOL = new Type(Type.Code.ARRAY, TYPE_BOOL, null);
   private static final Type TYPE_ARRAY_INT64 = new Type(Type.Code.ARRAY, TYPE_INT64, null);
   private static final Type TYPE_ARRAY_FLOAT32 = new Type(Type.Code.ARRAY, TYPE_FLOAT32, null);
@@ -66,6 +67,7 @@ public final class Type implements Serializable {
   private static final Type TYPE_PG_BYTEA = new Type(Type.Code.PG_BYTEA, null, null);
   private static final Type TYPE_PG_TIMESTAMPTZ = new Type(Type.Code.PG_TIMESTAMPTZ, null, null);
   private static final Type TYPE_PG_DATE = new Type(Type.Code.PG_DATE, null, null);
+  private static final Type TYPE_PG_UUID = new Type(Type.Code.PG_UUID, null, null);
   private static final Type TYPE_PG_ARRAY_BOOL = new Type(Type.Code.PG_ARRAY, TYPE_PG_BOOL, null);
   private static final Type TYPE_PG_ARRAY_INT8 = new Type(Type.Code.PG_ARRAY, TYPE_PG_INT8, null);
   private static final Type TYPE_PG_ARRAY_FLOAT4 =
@@ -160,6 +162,11 @@ public static Type date() {
     return TYPE_DATE;
   }
 
+  /** Returns the descriptor for the {@code UUID} type. */
+  public static Type uuid() {
+    return TYPE_UUID;
+  }
+
   public static Type pgBool() {
     return TYPE_PG_BOOL;
   }
@@ -204,6 +211,10 @@ public static Type pgDate() {
     return TYPE_PG_DATE;
   }
 
+  public static Type pgUuid() {
+    return TYPE_PG_UUID;
+  }
+
   public static Type pgCommitTimestamp() {
     return TYPE_PG_COMMIT_TIMESTAMP;
   }
@@ -317,6 +328,7 @@ public enum Code {
     BYTES("BYTES", Dialect.GOOGLE_STANDARD_SQL),
     TIMESTAMP("TIMESTAMP", Dialect.GOOGLE_STANDARD_SQL),
     DATE("DATE", Dialect.GOOGLE_STANDARD_SQL),
+    UUID("UUID", Dialect.GOOGLE_STANDARD_SQL),
     ARRAY("ARRAY", Dialect.GOOGLE_STANDARD_SQL),
     STRUCT("STRUCT", Dialect.GOOGLE_STANDARD_SQL),
     PG_BOOL("boolean", Dialect.POSTGRESQL),
@@ -330,6 +342,7 @@ public enum Code {
     PG_BYTEA("bytea", Dialect.POSTGRESQL),
     PG_TIMESTAMPTZ("timestamp with time zone", Dialect.POSTGRESQL),
     PG_DATE("date", Dialect.POSTGRESQL),
+    PG_UUID("uuid", Dialect.POSTGRESQL),
     PG_ARRAY("array", Dialect.POSTGRESQL),
     PG_COMMIT_TIMESTAMP("spanner.commit_timestamp", Dialect.POSTGRESQL);
 

v2/spanner-to-sourcedb/src/main/java/com/google/cloud/teleport/v2/templates/transforms/AssignShardIdFn.java

@@ -586,6 +586,8 @@ private com.google.cloud.spanner.Key generateKey(String tableName, JsonNode keys
           case STRING:
           case PG_VARCHAR:
           case PG_TEXT:
+          case UUID:
+          case PG_UUID:
             pk.append(
                 DataChangeRecordTypeConvertor.toString(
                     keysJson, keyColName, /* requiredField= */ true));
@@ -645,6 +647,8 @@ private Object getColumnValueFromJson(Column column, JsonNode valuesJson) throws
         case STRING:
         case PG_VARCHAR:
         case PG_TEXT:
+        case UUID:
+        case PG_UUID:
           return DataChangeRecordTypeConvertor.toString(valuesJson, colName, false);
         case NUMERIC:
         case PG_NUMERIC:
@@ -690,6 +694,8 @@ private Object getColumnValueFromRow(Column column, Value value) throws Exceptio
         case STRING:
         case PG_VARCHAR:
         case PG_TEXT:
+        case UUID:
+        case PG_UUID:
           return value.getString();
         case NUMERIC:
         case PG_NUMERIC: