Merge pull request #14 from vinted/feat/add-rows-type

feat: create rows type
vinted · Jan 24, 2024 · 1eb5b22 · 1eb5b22
2 parents 6dab5c0 + de41769
commit 1eb5b22
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/src/main/java/com/vinted/flink/bigquery/typeutils/RowsType.java b/src/main/java/com/vinted/flink/bigquery/typeutils/RowsType.java
@@ -0,0 +1,19 @@
+package com.vinted.flink.bigquery.typeutils;
+
+import com.google.cloud.bigquery.TableId;
+import com.vinted.flink.bigquery.model.Rows;
+import org.apache.flink.api.common.typeinfo.TypeInformation;
+import org.apache.flink.api.common.typeinfo.Types;
+
+import java.util.Map;
+
+public class RowsType {
+    public static <A> TypeInformation<Rows<A>> of(TypeInformation<A> typeInfo) {
+        return Types.POJO((Class<Rows<A>>)(Class<?>)Rows.class, Map.of(
+                "data", Types.LIST(typeInfo),
+                "offset", Types.LONG,
+                "stream", Types.STRING,
+                "table", TypeInformation.of(TableId.class)
+        ));
+    }
+}