From de41769fe7e86176ebf9aea51ffcca68ffb62197 Mon Sep 17 00:00:00 2001
From: gintarasm <gintaras.matulas@gmail.com>
Date: Wed, 24 Jan 2024 12:48:41 +0200
Subject: [PATCH] feat: create rows type

---
 .../flink/bigquery/typeutils/RowsType.java    | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)
 create mode 100644 src/main/java/com/vinted/flink/bigquery/typeutils/RowsType.java

diff --git a/src/main/java/com/vinted/flink/bigquery/typeutils/RowsType.java b/src/main/java/com/vinted/flink/bigquery/typeutils/RowsType.java
new file mode 100644
index 0000000..4d51a1f
--- /dev/null
+++ 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)
+        ));
+    }
+}