feat: support float32 type

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

@@ -506,6 +506,48 @@
     <method>com.google.cloud.spanner.connection.StatementResult execute(com.google.cloud.spanner.Statement, java.util.Set)</method>
   </difference>
 
+  <!-- FLOAT32 -->
+  <difference>
+    <differenceType>7012</differenceType>
+    <className>com/google/cloud/spanner/StructReader</className>
+    <method>float getFloat(int)</method>
+  </difference>
+  <difference>
+    <differenceType>7012</differenceType>
+    <className>com/google/cloud/spanner/StructReader</className>
+    <method>float getFloat(java.lang.String)</method>
+  </difference>
+  <difference>
+    <differenceType>7012</differenceType>
+    <className>com/google/cloud/spanner/StructReader</className>
+    <method>float[] getFloatArray(int)</method>
+  </difference>
+  <difference>
+    <differenceType>7012</differenceType>
+    <className>com/google/cloud/spanner/StructReader</className>
+    <method>float[] getFloatArray(java.lang.String)</method>
+  </difference>
+  <difference>
+    <differenceType>7012</differenceType>
+    <className>com/google/cloud/spanner/StructReader</className>
+    <method>java.util.List getFloatList(int)</method>
+  </difference>
+  <difference>
+    <differenceType>7012</differenceType>
+    <className>com/google/cloud/spanner/StructReader</className>
+    <method>java.util.List getFloatList(java.lang.String)</method>
+  </difference>
+  <difference>
+    <differenceType>7013</differenceType>
+    <className>com/google/cloud/spanner/Value</className>
+    <method>float getFloat32()</method>
+  </difference>
+  <difference>
+    <differenceType>7013</differenceType>
+    <className>com/google/cloud/spanner/Value</className>
+    <method>java.util.List getFloat32Array()</method>
+  </difference>
+
   <!-- (Internal change, use stream timeout) -->
   <difference>
     <differenceType>7012</differenceType>
@@ -569,7 +611,7 @@
     <method>void setSpan(io.opencensus.trace.Span)</method>
     <to>void setSpan(com.google.cloud.spanner.ISpan)</to>
   </difference>
-  
+
   <!-- Added DirectedReadOptions -->
   <difference>
     <differenceType>7012</differenceType>
@@ -580,5 +622,5 @@
     <differenceType>7012</differenceType>
     <className>com/google/cloud/spanner/connection/Connection</className>
     <method>void setDirectedRead(com.google.spanner.v1.DirectedReadOptions)</method>
-  </difference>  
+  </difference>
 </differences>

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

@@ -173,16 +173,44 @@ static double valueProtoToFloat64(com.google.protobuf.Value proto) {
     return proto.getNumberValue();
   }
 
+  static float valueProtoToFloat32(com.google.protobuf.Value proto) {
+    if (proto.getKindCase() == KindCase.STRING_VALUE) {
+      switch (proto.getStringValue()) {
+        case "-Infinity":
+          return Float.NEGATIVE_INFINITY;
+        case "Infinity":
+          return Float.POSITIVE_INFINITY;
+        case "NaN":
+          return Float.NaN;
+        default:
+          // Fall-through to handling below to produce an error.
+      }
+    }
+    if (proto.getKindCase() != KindCase.NUMBER_VALUE) {
+      throw newSpannerException(
+          ErrorCode.INTERNAL,
+          "Invalid value for column type "
+              + Type.float32()
+              + " expected NUMBER_VALUE or STRING_VALUE with value one of"
+              + " \"Infinity\", \"-Infinity\", or \"NaN\" but was "
+              + proto.getKindCase()
+              + (proto.getKindCase() == KindCase.STRING_VALUE
+                  ? " with value \"" + proto.getStringValue() + "\""
+                  : ""));
+    }
+    return (float) proto.getNumberValue();
+  }
+
   static NullPointerException throwNotNull(int columnIndex) {
     throw new NullPointerException(
         "Cannot call array getter for column " + columnIndex + " with null elements");
   }
 
   /**
-   * Memory-optimized base class for {@code ARRAY<INT64>} and {@code ARRAY<FLOAT64>} types. Both of
-   * these involve conversions from the type yielded by JSON parsing, which are {@code String} and
-   * {@code BigDecimal} respectively. Rather than construct new wrapper objects for each array
-   * element, we use primitive arrays and a {@code BitSet} to track nulls.
+   * Memory-optimized base class for {@code ARRAY<INT64>}, {@code ARRAY<FLOAT32>} and {@code
+   * ARRAY<FLOAT64>} types. All of these involve conversions from the type yielded by JSON parsing,
+   * which are {@code String} and {@code BigDecimal} respectively. Rather than construct new wrapper
+   * objects for each array element, we use primitive arrays and a {@code BitSet} to track nulls.
    */
   abstract static class PrimitiveArray<T, A> extends AbstractList<T> {
     private final A data;
@@ -264,6 +292,31 @@ Long get(long[] array, int i) {
     }
   }
 
+  static class Float32Array extends PrimitiveArray<Float, float[]> {
+    Float32Array(ListValue protoList) {
+      super(protoList);
+    }
+
+    Float32Array(float[] data, BitSet nulls) {
+      super(data, nulls, data.length);
+    }
+
+    @Override
+    float[] newArray(int size) {
+      return new float[size];
+    }
+
+    @Override
+    void setProto(float[] array, int i, com.google.protobuf.Value protoValue) {
+      array[i] = valueProtoToFloat32(protoValue);
+    }
+
+    @Override
+    Float get(float[] array, int i) {
+      return array[i];
+    }
+  }
+
   static class Float64Array extends PrimitiveArray<Double, double[]> {
     Float64Array(ListValue protoList) {
       super(protoList);
@@ -306,6 +359,11 @@ protected long getLongInternal(int columnIndex) {
     return currRow().getLongInternal(columnIndex);
   }
 
+  @Override
+  protected float getFloatInternal(int columnIndex) {
+    return currRow().getFloatInternal(columnIndex);
+  }
+
   @Override
   protected double getDoubleInternal(int columnIndex) {
     return currRow().getDoubleInternal(columnIndex);
@@ -382,6 +440,16 @@ protected List<Long> getLongListInternal(int columnIndex) {
     return currRow().getLongListInternal(columnIndex);
   }
 
+  @Override
+  protected float[] getFloatArrayInternal(int columnIndex) {
+    return currRow().getFloatArrayInternal(columnIndex);
+  }
+
+  @Override
+  protected List<Float> getFloatListInternal(int columnIndex) {
+    return currRow().getFloatListInternal(columnIndex);
+  }
+
   @Override
   protected double[] getDoubleArrayInternal(int columnIndex) {
     return currRow().getDoubleArrayInternal(columnIndex);

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

@@ -43,6 +43,10 @@ public abstract class AbstractStructReader implements StructReader {
 
   protected abstract long getLongInternal(int columnIndex);
 
+  protected float getFloatInternal(int columnIndex) {
+    throw new UnsupportedOperationException("Not implemented");
+  }
+
   protected abstract double getDoubleInternal(int columnIndex);
 
   protected abstract BigDecimal getBigDecimalInternal(int columnIndex);
@@ -94,6 +98,14 @@ protected Value getValueInternal(int columnIndex) {
 
   protected abstract List<Long> getLongListInternal(int columnIndex);
 
+  protected float[] getFloatArrayInternal(int columnIndex) {
+    throw new UnsupportedOperationException("Not implemented");
+  }
+
+  protected List<Float> getFloatListInternal(int columnIndex) {
+    throw new UnsupportedOperationException("Not implemented");
+  }
+
   protected abstract double[] getDoubleArrayInternal(int columnIndex);
 
   protected abstract List<Double> getDoubleListInternal(int columnIndex);
@@ -164,6 +176,19 @@ public long getLong(String columnName) {
     return getLongInternal(columnIndex);
   }
 
+  @Override
+  public float getFloat(int columnIndex) {
+    checkNonNullOfType(columnIndex, Type.float32(), columnIndex);
+    return getFloatInternal(columnIndex);
+  }
+
+  @Override
+  public float getFloat(String columnName) {
+    int columnIndex = getColumnIndex(columnName);
+    checkNonNullOfType(columnIndex, Type.float32(), columnName);
+    return getFloatInternal(columnIndex);
+  }
+
   @Override
   public double getDouble(int columnIndex) {
     checkNonNullOfType(columnIndex, Type.float64(), columnIndex);
@@ -368,6 +393,32 @@ public List<Long> getLongList(String columnName) {
     return getLongListInternal(columnIndex);
   }
 
+  @Override
+  public float[] getFloatArray(int columnIndex) {
+    checkNonNullOfType(columnIndex, Type.array(Type.float32()), columnIndex);
+    return getFloatArrayInternal(columnIndex);
+  }
+
+  @Override
+  public float[] getFloatArray(String columnName) {
+    int columnIndex = getColumnIndex(columnName);
+    checkNonNullOfType(columnIndex, Type.array(Type.float32()), columnName);
+    return getFloatArrayInternal(columnIndex);
+  }
+
+  @Override
+  public List<Float> getFloatList(int columnIndex) {
+    checkNonNullOfType(columnIndex, Type.array(Type.float32()), columnIndex);
+    return getFloatListInternal(columnIndex);
+  }
+
+  @Override
+  public List<Float> getFloatList(String columnName) {
+    int columnIndex = getColumnIndex(columnName);
+    checkNonNullOfType(columnIndex, Type.array(Type.float32()), columnName);
+    return getFloatListInternal(columnIndex);
+  }
+
   @Override
   public double[] getDoubleArray(int columnIndex) {
     checkNonNullOfType(columnIndex, Type.array(Type.float64()), columnIndex);

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

@@ -125,6 +125,18 @@ public long getLong(String columnName) {
     return delegate.get().getLong(columnName);
   }
 
+  @Override
+  public float getFloat(int columnIndex) {
+    checkValidState();
+    return delegate.get().getFloat(columnIndex);
+  }
+
+  @Override
+  public float getFloat(String columnName) {
+    checkValidState();
+    return delegate.get().getFloat(columnName);
+  }
+
   @Override
   public double getDouble(int columnIndex) {
     checkValidState();
@@ -267,6 +279,30 @@ public List<Long> getLongList(String columnName) {
     return delegate.get().getLongList(columnName);
   }
 
+  @Override
+  public float[] getFloatArray(int columnIndex) {
+    checkValidState();
+    return delegate.get().getFloatArray(columnIndex);
+  }
+
+  @Override
+  public float[] getFloatArray(String columnName) {
+    checkValidState();
+    return delegate.get().getFloatArray(columnName);
+  }
+
+  @Override
+  public List<Float> getFloatList(int columnIndex) {
+    checkValidState();
+    return delegate.get().getFloatList(columnIndex);
+  }
+
+  @Override
+  public List<Float> getFloatList(String columnName) {
+    checkValidState();
+    return delegate.get().getFloatList(columnName);
+  }
+
   @Override
   public double[] getDoubleArray(int columnIndex) {
     checkValidState();