From 6bd9e0361b6d9d6e539eff6df31eedb72aed89c8 Mon Sep 17 00:00:00 2001
From: bnechyporenko <bnechyporenko@bol.com>
Date: Sun, 25 Aug 2024 20:13:09 +0200
Subject: [PATCH] fix

Signed-off-by: bnechyporenko <bnechyporenko@bol.com>
---
 .../src/scaffolder/tasks/DatabaseTaskStore.ts | 19 ++++++++-----------
 .../tasks/DatabaseWorkspaceProvider.ts        |  2 +-
 2 files changed, 9 insertions(+), 12 deletions(-)

diff --git a/plugins/scaffolder-backend/src/scaffolder/tasks/DatabaseTaskStore.ts b/plugins/scaffolder-backend/src/scaffolder/tasks/DatabaseTaskStore.ts
index e8ac7a35da..23fe793432 100644
--- a/plugins/scaffolder-backend/src/scaffolder/tasks/DatabaseTaskStore.ts
+++ b/plugins/scaffolder-backend/src/scaffolder/tasks/DatabaseTaskStore.ts
@@ -532,10 +532,8 @@ export class DatabaseTaskStore implements TaskStore {
   }
 
   async cleanWorkspace({ taskId }: { taskId: string }): Promise<void> {
-    await this.db.transaction(async tx => {
-      await tx('tasks').where({ id: taskId }).update({
-        workspace: null,
-      });
+    await this.db('tasks').where({ id: taskId }).update({
+      workspace: null,
     });
   }
 
@@ -544,13 +542,12 @@ export class DatabaseTaskStore implements TaskStore {
     taskId: string;
   }): Promise<void> {
     if (options.path) {
-      await this.db.transaction(async tx => {
-        await tx<RawDbTaskRow>('tasks')
-          .where({ id: options.taskId })
-          .update({
-            workspace: (await serializeWorkspace(options)).contents,
-          });
-      });
+      const workspace = (await serializeWorkspace(options)).contents;
+      await this.db<RawDbTaskRow>('tasks')
+        .where({ id: options.taskId })
+        .update({
+          workspace,
+        });
     }
   }
 
diff --git a/plugins/scaffolder-backend/src/scaffolder/tasks/DatabaseWorkspaceProvider.ts b/plugins/scaffolder-backend/src/scaffolder/tasks/DatabaseWorkspaceProvider.ts
index 69f90b3d7e..b3c3c24975 100644
--- a/plugins/scaffolder-backend/src/scaffolder/tasks/DatabaseWorkspaceProvider.ts
+++ b/plugins/scaffolder-backend/src/scaffolder/tasks/DatabaseWorkspaceProvider.ts
@@ -29,7 +29,7 @@ export class DatabaseWorkspaceProvider implements WorkspaceProvider {
     path: string;
     taskId: string;
   }): Promise<void> {
-    this.storage.serializeWorkspace?.(options);
+    await this.storage.serializeWorkspace?.(options);
   }
 
   public async rehydrateWorkspace(options: {