You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
43 lines
1.6 KiB
43 lines
1.6 KiB
import {makeId} from 'app/server/lib/idUtils';
|
|
import {chunk} from 'lodash';
|
|
import {MigrationInterface, QueryRunner, TableColumn} from "typeorm";
|
|
|
|
export class UserUUID1663851423064 implements MigrationInterface {
|
|
public async up(queryRunner: QueryRunner): Promise<any> {
|
|
// Add ref column, for now make it nullable and not unique, we
|
|
// first need to put a value in it for all existing users.
|
|
await queryRunner.addColumn("users", new TableColumn({
|
|
name: "ref",
|
|
type: 'varchar',
|
|
isNullable: true,
|
|
isUnique: false,
|
|
}));
|
|
|
|
// Updating so many rows in a multiple queries is not ideal. We will send updates in chunks.
|
|
// 300 seems to be a good number, for 24k rows we have 80 queries.
|
|
const userList = await queryRunner.manager.createQueryBuilder()
|
|
.select(["users.id", "users.ref"])
|
|
.from("users", "users")
|
|
.getMany();
|
|
userList.forEach(u => u.ref = makeId());
|
|
|
|
const userChunks = chunk(userList, 300);
|
|
for (const users of userChunks) {
|
|
await queryRunner.connection.transaction(async manager => {
|
|
const queries = users.map((user: any, _index: number, _array: any[]) => {
|
|
return queryRunner.manager.update("users", user.id, user);
|
|
});
|
|
await Promise.all(queries);
|
|
});
|
|
}
|
|
|
|
// We are not making this column unique yet, because it can fail
|
|
// if there are some old workers still running, and any new user
|
|
// is created. We will make it unique in a later migration.
|
|
}
|
|
|
|
public async down(queryRunner: QueryRunner): Promise<any> {
|
|
await queryRunner.dropColumn("users", "ref");
|
|
}
|
|
}
|