From 75d172e6c19800b5adaa5e1f469b71d460a8080a Mon Sep 17 00:00:00 2001 From: Alexander Onnikov Date: Mon, 26 May 2025 13:57:03 +0700 Subject: [PATCH] UBERF-11178 Migrate datalake workspace id Signed-off-by: Alexander Onnikov --- .../datalake/pod-datalake/src/datalake/db.ts | 21 ++++++++++++++++++- 1 file changed, 20 insertions(+), 1 deletion(-) diff --git a/services/datalake/pod-datalake/src/datalake/db.ts b/services/datalake/pod-datalake/src/datalake/db.ts index 7d45011c28..4ea130c46b 100644 --- a/services/datalake/pod-datalake/src/datalake/db.ts +++ b/services/datalake/pod-datalake/src/datalake/db.ts @@ -560,7 +560,7 @@ export function escape (value: any): string { } function getMigrations (): [string, string][] { - return [migrationV1()] + return [migrationV1(), migrationV2()] } function migrationV1 (): [string, string] { @@ -599,3 +599,22 @@ function migrationV1 (): [string, string] { ` return ['init_tables_01', sql] } + +function migrationV2 (): [string, string] { + const sql = ` + ALTER TABLE blob.meta DROP CONSTRAINT IF EXISTS fk_blob; + + UPDATE blob.blob + SET workspace = w.uuid + FROM global_account.workspace w + WHERE workspace = w.data_id; + + UPDATE blob.meta + SET workspace = w.uuid + FROM global_account.workspace w + WHERE workspace = w.data_id; + + ALTER TABLE blob.meta ADD CONSTRAINT fk_blob FOREIGN KEY (workspace, name) REFERENCES blob.blob (workspace, name); + ` + return ['migrate_workspaces_02', sql] +}