feat: create table with constraints (#19828)

This commit is contained in:
Jason Rasmussen
2025-07-09 09:13:14 -04:00
committed by GitHub
parent 4db76ddcf0
commit 51ab7498e9
7 changed files with 170 additions and 60 deletions

View File

@@ -20,12 +20,13 @@ export const transformConstraints: SqlTransformer = (ctx, item) => {
const withAction = (constraint: { onDelete?: ActionType; onUpdate?: ActionType }) =>
` ON UPDATE ${constraint.onUpdate ?? ActionType.NO_ACTION} ON DELETE ${constraint.onDelete ?? ActionType.NO_ACTION}`;
export const asConstraintAdd = (constraint: DatabaseConstraint): string | string[] => {
const base = `ALTER TABLE "${constraint.tableName}" ADD CONSTRAINT "${constraint.name}"`;
export const asConstraintBody = (constraint: DatabaseConstraint): string => {
const base = `CONSTRAINT "${constraint.name}"`;
switch (constraint.type) {
case ConstraintType.PRIMARY_KEY: {
const columnNames = asColumnList(constraint.columnNames);
return `${base} PRIMARY KEY (${columnNames});`;
return `${base} PRIMARY KEY (${columnNames})`;
}
case ConstraintType.FOREIGN_KEY: {
@@ -33,26 +34,29 @@ export const asConstraintAdd = (constraint: DatabaseConstraint): string | string
const referenceColumnNames = asColumnList(constraint.referenceColumnNames);
return (
`${base} FOREIGN KEY (${columnNames}) REFERENCES "${constraint.referenceTableName}" (${referenceColumnNames})` +
withAction(constraint) +
';'
withAction(constraint)
);
}
case ConstraintType.UNIQUE: {
const columnNames = asColumnList(constraint.columnNames);
return `${base} UNIQUE (${columnNames});`;
return `${base} UNIQUE (${columnNames})`;
}
case ConstraintType.CHECK: {
return `${base} CHECK (${constraint.expression});`;
return `${base} CHECK (${constraint.expression})`;
}
default: {
return [];
throw new Error(`Unknown constraint type: ${(constraint as any).type}`);
}
}
};
export const asConstraintAdd = (constraint: DatabaseConstraint): string | string[] => {
return `ALTER TABLE "${constraint.tableName}" ADD ${asConstraintBody(constraint)};`;
};
export const asConstraintDrop = (tableName: string, constraintName: string): string => {
return `ALTER TABLE "${tableName}" DROP CONSTRAINT "${constraintName}";`;
};