Fix Beam to handle float32 arrays and pg schema.

arawind · arawind · commit c1cd281b1113 · 2024-03-21T18:15:05.000+05:30
This was missed in the earlier commits
diff --git a/v1/src/main/java/org/apache/beam/sdk/io/gcp/spanner/MutationSizeEstimator.java b/v1/src/main/java/org/apache/beam/sdk/io/gcp/spanner/MutationSizeEstimator.java
@@ -142,6 +142,8 @@ private static long estimateArrayValue(Value v) {
     switch (v.getType().getArrayElementType().getCode()) {
       case BOOL:
         return v.getBoolArray().size();
+      case FLOAT32:
+        return 4L * v.getFloat32Array().size();
       case INT64:
         return 8L * v.getInt64Array().size();
       case FLOAT64:
diff --git a/v1/src/main/java/org/apache/beam/sdk/io/gcp/spanner/SpannerSchema.java b/v1/src/main/java/org/apache/beam/sdk/io/gcp/spanner/SpannerSchema.java
@@ -215,6 +215,9 @@ private static Type parseSpannerType(String spannerType, Dialect dialect) {
           if ("BIGINT".equals(spannerType)) {
             return Type.int64();
           }
+          if ("REAL".equals(spannerType)) {
+            return Type.float32();
+          }
           if ("DOUBLE PRECISION".equals(spannerType)) {
             return Type.float64();
           }

Original file line number	Diff line number	Diff line change
`@@ -215,6 +215,9 @@ private static Type parseSpannerType(String spannerType, Dialect dialect) {`
`215`	`215`	`if ("BIGINT".equals(spannerType)) {`
`216`	`216`	`return Type.int64();`
`217`	`217`	`}`
	`218`	`+ if ("REAL".equals(spannerType)) {`
	`219`	`+ return Type.float32();`
	`220`	`+ }`
`218`	`221`	`if ("DOUBLE PRECISION".equals(spannerType)) {`
`219`	`222`	`return Type.float64();`
`220`	`223`	`}`