|
1 | 1 | import { drizzle } from "drizzle-orm/libsql" |
2 | 2 | import { eq } from "drizzle-orm" |
| 3 | +import pMap from "p-map" |
3 | 4 |
|
4 | 5 | import { db as sourceDb } from "../src/db.js" |
5 | 6 | import { schema } from "../src/schema.js" |
@@ -52,29 +53,33 @@ const copyRun = async (runId: number) => { |
52 | 53 |
|
53 | 54 | console.log(`Copying ${tasks.length} tasks`) |
54 | 55 |
|
55 | | - for (const task of tasks) { |
56 | | - // eslint-disable-next-line @typescript-eslint/no-unused-vars |
57 | | - const { id: _, ...newTaskMetricsValues } = task.taskMetrics! |
58 | | - const [newTaskMetrics] = await destDb.insert(schema.taskMetrics).values(newTaskMetricsValues).returning() |
59 | | - |
60 | | - if (!newTaskMetrics) { |
61 | | - throw new Error(`Failed to insert taskMetrics for task ${task.id}`) |
62 | | - } |
63 | | - |
64 | | - // eslint-disable-next-line @typescript-eslint/no-unused-vars |
65 | | - const { id: __, ...newTaskValues } = task |
66 | | - |
67 | | - const [newTask] = await destDb |
68 | | - .insert(schema.tasks) |
69 | | - .values({ ...newTaskValues, runId: newRun.id, taskMetricsId: newTaskMetrics.id }) |
70 | | - .returning() |
71 | | - |
72 | | - if (!newTask) { |
73 | | - throw new Error(`Failed to insert task ${task.id}`) |
74 | | - } |
75 | | - } |
76 | | - |
77 | | - console.log(`Successfully copied run ${runId} with ${tasks.length} tasks`) |
| 56 | + await pMap( |
| 57 | + tasks, |
| 58 | + async (task) => { |
| 59 | + // eslint-disable-next-line @typescript-eslint/no-unused-vars |
| 60 | + const { id: _, ...newTaskMetricsValues } = task.taskMetrics! |
| 61 | + const [newTaskMetrics] = await destDb.insert(schema.taskMetrics).values(newTaskMetricsValues).returning() |
| 62 | + |
| 63 | + if (!newTaskMetrics) { |
| 64 | + throw new Error(`Failed to insert taskMetrics for task ${task.id}`) |
| 65 | + } |
| 66 | + |
| 67 | + // eslint-disable-next-line @typescript-eslint/no-unused-vars |
| 68 | + const { id: __, ...newTaskValues } = task |
| 69 | + |
| 70 | + const [newTask] = await destDb |
| 71 | + .insert(schema.tasks) |
| 72 | + .values({ ...newTaskValues, runId: newRun.id, taskMetricsId: newTaskMetrics.id }) |
| 73 | + .returning() |
| 74 | + |
| 75 | + if (!newTask) { |
| 76 | + throw new Error(`Failed to insert task ${task.id}`) |
| 77 | + } |
| 78 | + }, |
| 79 | + { concurrency: 25 }, |
| 80 | + ) |
| 81 | + |
| 82 | + console.log(`\nSuccessfully copied run ${runId} with ${tasks.length} tasks`) |
78 | 83 | } |
79 | 84 |
|
80 | 85 | const main = async () => { |
|
0 commit comments