From 1755ac2fecccd53f27f94c7dcad270551086cdb1 Mon Sep 17 00:00:00 2001 From: Sagar Ahire Date: Wed, 19 Apr 2023 15:59:06 +0530 Subject: [PATCH] Add a method to add schema information to a relation --- .../gcp/bigquery/relational/BigQueryRelation.java | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/src/main/java/io/cdap/plugin/gcp/bigquery/relational/BigQueryRelation.java b/src/main/java/io/cdap/plugin/gcp/bigquery/relational/BigQueryRelation.java index 796cfa2d67..18d12b378b 100644 --- a/src/main/java/io/cdap/plugin/gcp/bigquery/relational/BigQueryRelation.java +++ b/src/main/java/io/cdap/plugin/gcp/bigquery/relational/BigQueryRelation.java @@ -349,6 +349,16 @@ public Relation window(WindowAggregationDefinition definition) { return new BigQueryRelation(datasetName, columns, featureFlagsProvider, this, supplier); } + /** + * Adds schema information to the relation. This can be used for validation purposes. + * + * @param schema The schema. + * @return A new relation with the schema added. + */ + public Relation addSchema(@Nullable Schema schema) { + return new BigQueryRelation(datasetName, columns, featureFlagsProvider, this, sqlStatementSupplier, schema); + } + private static String buildBaseSelect(Map columns, String sourceTable, String datasetName) {