feat(tvf): add table schema for update operation

nrkno · Nov 8, 2023 · e5ea400 · e5ea400
1 parent c0f9134
commit e5ea400
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/core/src/main/scala/no/nrk/bigquery/internal/RoutineUpdateOperation.scala b/core/src/main/scala/no/nrk/bigquery/internal/RoutineUpdateOperation.scala
@@ -71,7 +71,15 @@ object RoutineUpdateOperation {
       .setLanguage("SQL")
       .setBody(tvf.query.asString)
       // function not public in Builder: .setDescription(tvf.description.orNull)
-      // TODO .setReturnTableType(tvf.schema.???)
+      .setReturnTableType(
+        StandardSQLTableType
+          .newBuilder()
+          .setColumns(
+            tvf.schema.fields
+              .map(field => StandardSQLField.newBuilder(field.name, toSqlDataType(BQType.fromField(field))).build())
+              .asJava)
+          .build()
+      )
       .build()
 
   private def createUdfRoutineInfo(udf: UDF.Persistent[_]) = {