feat: add PG OID support

.gitignore

@@ -31,3 +31,5 @@ docs/
 *.pyc
 
 .flattened-pom.xml
+.java-version
+.vscode/

google-cloud-spanner/clirr-ignored-differences.xml

@@ -192,6 +192,18 @@
     <className>com/google/cloud/spanner/StructReader</className>
     <method>java.util.List getPgJsonbList(java.lang.String)</method>
   </difference>
+  <!-- PG OID -->
+  <difference>
+    <differenceType>7013</differenceType>
+    <className>com/google/cloud/spanner/Value</className>
+    <method>long getPgOid()</method>
+  </difference>
+  <difference>
+    <differenceType>7013</differenceType>
+    <className>com/google/cloud/spanner/Value</className>
+    <method>java.util.List getPgOidArray()</method>
+  </difference>
+
   <difference>
     <differenceType>7012</differenceType>
     <className>com/google/cloud/spanner/StructReader</className>

google-cloud-spanner/src/main/java/com/google/cloud/spanner/AbstractStructReader.java

@@ -165,14 +165,15 @@ public boolean getBoolean(String columnName) {
 
   @Override
   public long getLong(int columnIndex) {
-    checkNonNullOfCodes(columnIndex, Arrays.asList(Code.ENUM, Code.INT64), columnIndex);
+    checkNonNullOfCodes(
+        columnIndex, Arrays.asList(Code.ENUM, Code.PG_OID, Code.INT64), columnIndex);
     return getLongInternal(columnIndex);
   }
 
   @Override
   public long getLong(String columnName) {
     int columnIndex = getColumnIndex(columnName);
-    checkNonNullOfCodes(columnIndex, Arrays.asList(Code.ENUM, Code.INT64), columnName);
+    checkNonNullOfCodes(columnIndex, Arrays.asList(Code.ENUM, Code.PG_OID, Code.INT64), columnName);
     return getLongInternal(columnIndex);
   }
 
@@ -366,30 +367,34 @@ public List<Boolean> getBooleanList(String columnName) {
   @Override
   public long[] getLongArray(int columnIndex) {
     checkNonNullOfCodes(columnIndex, Collections.singletonList(Code.ARRAY), columnIndex);
-    checkArrayElementType(columnIndex, Arrays.asList(Code.ENUM, Code.INT64), columnIndex);
+    checkArrayElementType(
+        columnIndex, Arrays.asList(Code.ENUM, Code.PG_OID, Code.INT64), columnIndex);
     return getLongArrayInternal(columnIndex);
   }
 
   @Override
   public long[] getLongArray(String columnName) {
     int columnIndex = getColumnIndex(columnName);
     checkNonNullOfCodes(columnIndex, Collections.singletonList(Code.ARRAY), columnName);
-    checkArrayElementType(columnIndex, Arrays.asList(Code.ENUM, Code.INT64), columnName);
+    checkArrayElementType(
+        columnIndex, Arrays.asList(Code.ENUM, Code.PG_OID, Code.INT64), columnName);
     return getLongArrayInternal(columnIndex);
   }
 
   @Override
   public List<Long> getLongList(int columnIndex) {
     checkNonNullOfCodes(columnIndex, Collections.singletonList(Code.ARRAY), columnIndex);
-    checkArrayElementType(columnIndex, Arrays.asList(Code.ENUM, Code.INT64), columnIndex);
+    checkArrayElementType(
+        columnIndex, Arrays.asList(Code.ENUM, Code.PG_OID, Code.INT64), columnIndex);
     return getLongListInternal(columnIndex);
   }
 
   @Override
   public List<Long> getLongList(String columnName) {
     int columnIndex = getColumnIndex(columnName);
     checkNonNullOfCodes(columnIndex, Collections.singletonList(Code.ARRAY), columnName);
-    checkArrayElementType(columnIndex, Arrays.asList(Code.ENUM, Code.INT64), columnName);
+    checkArrayElementType(
+        columnIndex, Arrays.asList(Code.ENUM, Code.PG_OID, Code.INT64), columnName);
     return getLongListInternal(columnIndex);
   }
 

google-cloud-spanner/src/main/java/com/google/cloud/spanner/GrpcStruct.java

@@ -114,6 +114,9 @@ private Object writeReplace() {
         case PG_JSONB:
           builder.set(fieldName).to(Value.pgJsonb((String) value));
           break;
+        case PG_OID:
+          builder.set(fieldName).to(Value.pgOid((Long) value));
+          break;
         case BYTES:
           builder
               .set(fieldName)
@@ -158,6 +161,9 @@ private Object writeReplace() {
             case PG_JSONB:
               builder.set(fieldName).toPgJsonbArray((Iterable<String>) value);
               break;
+            case PG_OID:
+              builder.set(fieldName).toPgOidArray((Iterable<Long>) value);
+              break;
             case BYTES:
             case PROTO:
               builder
@@ -262,6 +268,7 @@ private static Object decodeValue(Type fieldType, com.google.protobuf.Value prot
         checkType(fieldType, proto, KindCase.BOOL_VALUE);
         return proto.getBoolValue();
       case INT64:
+      case PG_OID:
       case ENUM:
         checkType(fieldType, proto, KindCase.STRING_VALUE);
         return Long.parseLong(proto.getStringValue());
@@ -319,6 +326,7 @@ private static Struct decodeStructValue(Type structType, ListValue structValue)
   static Object decodeArrayValue(Type elementType, ListValue listValue) {
     switch (elementType.getCode()) {
       case INT64:
+      case PG_OID:
       case ENUM:
         // For int64/float64/float32/enum types, use custom containers.
         // These avoid wrapper object creation for non-null arrays.
@@ -563,6 +571,8 @@ protected Value getValueInternal(int columnIndex) {
         return Value.json(isNull ? null : getJsonInternal(columnIndex));
       case PG_JSONB:
         return Value.pgJsonb(isNull ? null : getPgJsonbInternal(columnIndex));
+      case PG_OID:
+        return Value.pgOid(isNull ? null : getLongInternal(columnIndex));
       case BYTES:
         return Value.internalBytes(isNull ? null : getLazyBytesInternal(columnIndex));
       case PROTO:
@@ -598,6 +608,8 @@ protected Value getValueInternal(int columnIndex) {
             return Value.jsonArray(isNull ? null : getJsonListInternal(columnIndex));
           case PG_JSONB:
             return Value.pgJsonbArray(isNull ? null : getPgJsonbListInternal(columnIndex));
+          case PG_OID:
+            return Value.pgOidArray(isNull ? null : getLongListInternal(columnIndex));
           case BYTES:
             return Value.bytesArray(isNull ? null : getBytesListInternal(columnIndex));
           case PROTO:

google-cloud-spanner/src/main/java/com/google/cloud/spanner/Struct.java

@@ -396,6 +396,7 @@ private Object getAsObject(int columnIndex) {
       case BOOL:
         return getBooleanInternal(columnIndex);
       case INT64:
+      case PG_OID:
       case ENUM:
         return getLongInternal(columnIndex);
       case FLOAT32:
@@ -426,6 +427,7 @@ private Object getAsObject(int columnIndex) {
           case BOOL:
             return getBooleanListInternal(columnIndex);
           case INT64:
+          case PG_OID:
           case ENUM:
             return getLongListInternal(columnIndex);
           case FLOAT32:

google-cloud-spanner/src/main/java/com/google/cloud/spanner/Type.java

@@ -55,6 +55,7 @@ public final class Type implements Serializable {
   private static final Type TYPE_STRING = new Type(Code.STRING, null, null);
   private static final Type TYPE_JSON = new Type(Code.JSON, null, null);
   private static final Type TYPE_PG_JSONB = new Type(Code.PG_JSONB, null, null);
+  private static final Type TYPE_PG_OID = new Type(Code.PG_OID, null, null);
   private static final Type TYPE_BYTES = new Type(Code.BYTES, null, null);
   private static final Type TYPE_TIMESTAMP = new Type(Code.TIMESTAMP, null, null);
   private static final Type TYPE_DATE = new Type(Code.DATE, null, null);
@@ -67,6 +68,7 @@ public final class Type implements Serializable {
   private static final Type TYPE_ARRAY_STRING = new Type(Code.ARRAY, TYPE_STRING, null);
   private static final Type TYPE_ARRAY_JSON = new Type(Code.ARRAY, TYPE_JSON, null);
   private static final Type TYPE_ARRAY_PG_JSONB = new Type(Code.ARRAY, TYPE_PG_JSONB, null);
+  private static final Type TYPE_ARRAY_PG_OID = new Type(Code.ARRAY, TYPE_PG_OID, null);
   private static final Type TYPE_ARRAY_BYTES = new Type(Code.ARRAY, TYPE_BYTES, null);
   private static final Type TYPE_ARRAY_TIMESTAMP = new Type(Code.ARRAY, TYPE_TIMESTAMP, null);
   private static final Type TYPE_ARRAY_DATE = new Type(Code.ARRAY, TYPE_DATE, null);
@@ -137,6 +139,11 @@ public static Type pgJsonb() {
     return TYPE_PG_JSONB;
   }
 
+  /** Returns the descriptor for the {@code PG_OID} type. */
+  public static Type pgOid() {
+    return TYPE_PG_OID;
+  }
+
   /**
    * To get the descriptor for the {@code PROTO} type.
    *
@@ -198,6 +205,8 @@ public static Type array(Type elementType) {
         return TYPE_ARRAY_JSON;
       case PG_JSONB:
         return TYPE_ARRAY_PG_JSONB;
+      case PG_OID:
+        return TYPE_ARRAY_PG_OID;
       case BYTES:
         return TYPE_ARRAY_BYTES;
       case TIMESTAMP:
@@ -280,6 +289,7 @@ public enum Code {
     STRING(TypeCode.STRING, "character varying"),
     JSON(TypeCode.JSON, "unknown"),
     PG_JSONB(TypeCode.JSON, "jsonb", TypeAnnotationCode.PG_JSONB),
+    PG_OID(TypeCode.INT64, "oid", TypeAnnotationCode.PG_OID),
     PROTO(TypeCode.PROTO, "proto"),
     ENUM(TypeCode.ENUM, "enum"),
     BYTES(TypeCode.BYTES, "bytea"),
@@ -592,6 +602,8 @@ static Type fromProto(com.google.spanner.v1.Type proto) {
         return json();
       case PG_JSONB:
         return pgJsonb();
+      case PG_OID:
+        return pgOid();
       case BYTES:
         return bytes();
       case TIMESTAMP: