ToolJet/server/data-migrations/1683136077244-AddUniqueConstraintToWorkspaceName.ts

import { Organization } from '@entities/organization.entity';
import { DataBaseConstraints } from '@helpers/db_constraints.constants';
import { addWait } from '@helpers/migration.helper';
import { EntityManager, MigrationInterface, QueryRunner, TableUnique } from 'typeorm';

export class AddUniqueConstraintToWorkspaceName1683136077244 implements MigrationInterface {
  public async up(queryRunner: QueryRunner): Promise<void> {
    const entityManager = queryRunner.manager;
    await this.migrateWorkspaceNames(entityManager);
    await queryRunner.createUniqueConstraint(
      'organizations',
      new TableUnique({
        name: DataBaseConstraints.WORKSPACE_NAME_UNIQUE,
        columnNames: ['name'],
      })
    );
  }

  public async migrateWorkspaceNames(entityManager: EntityManager) {
    const workspaces = await entityManager.query(
      'select sub_query.name from (select count(*) as name_count, name from organizations group by name) sub_query where name_count > 1    '
    );
    for (const workspace of workspaces) {
      const { name } = workspace;
      const sameNameWorkspaces = await entityManager.find(Organization, {
        where: {
          name,
        },
      });
      for (const workspaceToChange of sameNameWorkspaces.slice(1)) {
        await entityManager.update(
          Organization,
          { id: workspaceToChange.id },
          { name: `${workspaceToChange.name} ${Date.now()}` }
        );
        // Add 1 millisecond wait to prevent duplicate timestamp generation
        addWait(1);
      }
    }
  }

  public async down(queryRunner: QueryRunner): Promise<void> {}
}