0
Fork 0
mirror of https://github.com/logto-io/logto.git synced 2024-12-16 20:26:19 -05:00

fix(schemas): fix init account center alter run in chunks (#6823)

This commit is contained in:
wangsijie 2024-11-20 13:28:49 +08:00 committed by GitHub
parent 7ebef18e36
commit 80ae0bb6c7
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -4,16 +4,34 @@ import type { AlterationScript } from '../lib/types/alteration.js';
const alteration: AlterationScript = { const alteration: AlterationScript = {
up: async (pool) => { up: async (pool) => {
const tenants = await pool.many<{ id: string }>(sql` // Process in chunks of 1000 tenants
select id from tenants; const batchSize = 1000;
`); // eslint-disable-next-line @silverhand/fp/no-let
const values = tenants.map((tenant) => sql`(${tenant.id}, 'default')`); let offset = 0;
// Other fileds have default values so we don't need to set them here // eslint-disable-next-line @typescript-eslint/no-unnecessary-condition, no-constant-condition
await pool.query(sql` while (true) {
insert into account_centers (tenant_id, id) // eslint-disable-next-line no-await-in-loop
values ${sql.join(values, sql`, `)}; const tenants = await pool.any<{ id: string }>(sql`
`); select id from tenants
order by created_at asc
limit ${batchSize} offset ${offset};
`);
if (tenants.length === 0) {
break;
}
const values = tenants.map((tenant) => sql`(${tenant.id}, 'default')`);
// eslint-disable-next-line no-await-in-loop
await pool.query(sql`
insert into account_centers (tenant_id, id)
values ${sql.join(values, sql`, `)};
`);
// eslint-disable-next-line @silverhand/fp/no-mutation
offset += batchSize;
}
}, },
down: async (pool) => { down: async (pool) => {
await pool.query(sql` await pool.query(sql`