From e56e8b16a8ad544d3beb69c90cfb702b51d5d1ab Mon Sep 17 00:00:00 2001 From: Adrika Gupta Date: Mon, 13 Apr 2026 09:12:29 +0000 Subject: [PATCH] fix classcastexception in bq source --- .../gcp/bigquery/source/PartitionedBigQueryInputFormat.java | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/src/main/java/io/cdap/plugin/gcp/bigquery/source/PartitionedBigQueryInputFormat.java b/src/main/java/io/cdap/plugin/gcp/bigquery/source/PartitionedBigQueryInputFormat.java index e0834f357..163403f7f 100644 --- a/src/main/java/io/cdap/plugin/gcp/bigquery/source/PartitionedBigQueryInputFormat.java +++ b/src/main/java/io/cdap/plugin/gcp/bigquery/source/PartitionedBigQueryInputFormat.java @@ -24,6 +24,7 @@ import com.google.api.services.bigquery.model.TableReference; import com.google.cloud.bigquery.RangePartitioning; import com.google.cloud.bigquery.StandardTableDefinition; +import com.google.cloud.bigquery.TableDefinition; import com.google.cloud.bigquery.TableDefinition.Type; import com.google.cloud.bigquery.TimePartitioning; import com.google.cloud.hadoop.io.bigquery.AbstractBigQueryInputFormat; @@ -146,7 +147,7 @@ private void processQuery(JobContext context) throws IOException, InterruptedExc readTimeout); Type type = Objects.requireNonNull(bigQueryTable).getDefinition().getType(); Boolean isPartitionFilterRequired = bigQueryTable.getRequirePartitionFilter(); - StandardTableDefinition tableDefinition = Objects.requireNonNull(bigQueryTable).getDefinition(); + TableDefinition tableDefinition = Objects.requireNonNull(bigQueryTable).getDefinition(); String query; if (type == Type.VIEW || type == Type.MATERIALIZED_VIEW || type == Type.EXTERNAL) { @@ -156,7 +157,7 @@ private void processQuery(JobContext context) throws IOException, InterruptedExc query = generateQuery(partitionFromDate, partitionToDate, filter, datasetProjectId, datasetId, tableName, limit, orderBy, - isPartitionFilterRequired, tableDefinition); + isPartitionFilterRequired, (StandardTableDefinition) tableDefinition); } if (query != null) {