SingleStoreRecreateTableConvertor Class — drizzle-orm Architecture

Source Code

drizzle-kit/src/sqlgenerator.ts lines 3896–3951

class SingleStoreRecreateTableConvertor extends Convertor {
	can(statement: JsonStatement, dialect: Dialect): boolean {
		return (
			statement.type === 'singlestore_recreate_table'
			&& dialect === 'singlestore'
		);
	}

	convert(statement: JsonRecreateSingleStoreTableStatement): string[] {
		const { tableName, columns, compositePKs, uniqueConstraints } = statement;

		const columnNames = columns.map((it) => `\`${it.name}\``).join(', ');
		const newTableName = `__new_${tableName}`;

		const sqlStatements: string[] = [];

		// create new table
		sqlStatements.push(
			new SingleStoreCreateTableConvertor().convert({
				type: 'create_table',
				tableName: newTableName,
				columns,
				compositePKs,
				uniqueConstraints,
				schema: '',
			}),
		);

		// migrate data
		sqlStatements.push(
			`INSERT INTO \`${newTableName}\`(${columnNames}) SELECT ${columnNames} FROM \`${tableName}\`;`,
		);

		// drop table
		sqlStatements.push(
			new SingleStoreDropTableConvertor().convert({
				type: 'drop_table',
				tableName: tableName,
				schema: '',
			}),
		);

		// rename table
		sqlStatements.push(
			new SingleStoreRenameTableConvertor().convert({
				fromSchema: '',
				tableNameFrom: newTableName,
				tableNameTo: tableName,
				toSchema: '',
				type: 'rename_table',
			}),
		);

		return sqlStatements;
	}
}