彻底重构-4

apps/backend/src/repositories/task-repo.js

@@ -129,7 +129,7 @@ export async function createTask(input) {
       await upsertTencentBrowserContextWithClient(client, taskId, input.tencentContext, input.createdAt)
     }
 
-    return getTaskById(taskId)
+    return getTaskByIdWithExecutor(client.query.bind(client), taskId)
   })
 }
 
@@ -202,18 +202,12 @@ export async function updateTask(taskId, patch) {
       }, patch.updated_at || current.updated_at)
     }
 
-    return getTaskById(taskId)
+    return getTaskByIdWithExecutor(client.query.bind(client), taskId)
   })
 }
 
 export async function getTaskById(taskId) {
-  const result = await query(
-    `${buildTaskSelect()}
-     WHERE ft.id = $1
-     LIMIT 1`,
-    [Number(taskId)],
-  )
-  return result.rows[0] || null
+  return getTaskByIdWithExecutor(query, taskId)
 }
 
 export async function findTaskByClaimTokenId(claimTokenId) {
@@ -399,6 +393,16 @@ async function upsertTencentBrowserContextWithClient(client, taskId, patch = {},
   )
 }
 
+async function getTaskByIdWithExecutor(executor, taskId) {
+  const result = await executor(
+    `${buildTaskSelect()}
+     WHERE ft.id = $1
+     LIMIT 1`,
+    [Number(taskId)],
+  )
+  return result.rows[0] || null
+}
+
 function containsTencentPatch(patch = {}) {
   return [
     'browser_session_id',