fix: bug when updating table to add cluster not reflected (#182)

goto · Nov 17, 2023 · 427f9ac · 427f9ac
1 parent c229f8f
commit 427f9ac
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git 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
@@ -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",