From 427f9ac0b75a4b74b8ffab6bf34101618228c4d8 Mon Sep 17 00:00:00 2001
From: Anwar Hidayat <15167551+irainia@users.noreply.github.com>
Date: Fri, 17 Nov 2023 10:22:31 +0700
Subject: [PATCH] fix: bug when updating table to add cluster not reflected
 (#182)

---
 ext/store/bigquery/table.go      | 4 ++++
 ext/store/bigquery/table_test.go | 3 +++
 2 files changed, 7 insertions(+)

diff --git a/ext/store/bigquery/table.go b/ext/store/bigquery/table.go
index afe1c567b4..8def08eccf 100644
--- a/ext/store/bigquery/table.go
+++ b/ext/store/bigquery/table.go
@@ -71,6 +71,10 @@ func (t TableHandle) Update(ctx context.Context, res *resource.Resource) error {
 		}
 	}
 
+	if table.Cluster != nil {
+		metadataToUpdate.Clustering = toBQClustering(table.Cluster)
+	}
+
 	_, err = t.bqTable.Update(ctx, metadataToUpdate, "")
 	if err != nil {
 		var metaErr *googleapi.Error
diff --git a/ext/store/bigquery/table_test.go b/ext/store/bigquery/table_test.go
index 9314845199..ce1c9be91c 100644
--- a/ext/store/bigquery/table_test.go
+++ b/ext/store/bigquery/table_test.go
@@ -245,6 +245,9 @@ func TestTableHandle(t *testing.T) {
 						"type": "INTEGER",
 					},
 				},
+				"cluster": map[string]any{
+					"using": []string{"session"},
+				},
 				"partition": map[string]any{
 					"field": "product_day",
 					"type":  "DAY",