|
public static Form ( PlSqlParser context ) : |
||
| context | PlSqlParser | |
| return | ||
public static IAlterTableAction Build(PlSqlParser.AlterTableActionContext context)
{
if (context.addColumnAction() != null)
{
var column = TableColumn.Form(context.addColumnAction().tableColumn(), new List <ColumnConstraint>());
return(new AddColumnAction(column));
}
if (context.addConstraintAction() != null)
{
var constraint = TableConstraint.Form(context.addConstraintAction().tableConstraint());
return(new AddConstraintAction(constraint));
}
if (context.dropColumnAction() != null)
{
var columnName = context.dropColumnAction().id().GetText();
return(new DropColumnAction(columnName));
}
if (context.dropConstraintAction() != null)
{
var constraintName = context.dropConstraintAction().id().GetText();
return(new DropConstraintAction(constraintName));
}
if (context.dropPrimaryKeyAction() != null)
{
return(new DropPrimaryKeyAction());
}
if (context.dropDefaultAction() != null)
{
var columnName = Name.Simple(context.dropDefaultAction().columnName());
return(new DropDefaultAction(columnName));
}
if (context.setDefaultAction() != null)
{
var columnName = Name.Simple(context.setDefaultAction().columnName());
var defaultValue = Expression.Build(context.setDefaultAction().expressionWrapper());
return(new SetDefaultAction(columnName, defaultValue));
}
throw new ParseCanceledException("The ALTER TABLE action is not supported");
}
public static SqlStatement Build(PlSqlParser.CreateTableStatementContext context)
{
var tableName = Name.Object(context.objectName());
var columnOrConstraints = context.columnOrConstraintList().columnOrConstraint();
bool ifNotExists = context.IF() != null && context.NOT() != null && context.EXISTS() != null;
bool temporary = context.TEMPORARY() != null;
var tableColumns = new List <SqlTableColumn>();
var tableConstraints = new List <SqlTableConstraint>();
var columnConstraints = new List <ColumnConstraint>();
foreach (var columnOrConstraint in columnOrConstraints)
{
if (columnOrConstraint.tableColumn() != null)
{
tableColumns.Add(TableColumn.Form(columnOrConstraint.tableColumn(), columnConstraints));
}
else if (columnOrConstraint.tableConstraint() != null)
{
tableConstraints.Add(TableConstraint.Form(columnOrConstraint.tableConstraint()));
}
}
if (columnConstraints.Count > 0)
{
var constraintGroups = columnConstraints.GroupBy(x => x.Type);
foreach (var constraintGroup in constraintGroups)
{
var columnNames = constraintGroup.Select(x => x.ColumnName).ToArray();
var index = tableConstraints.FindIndex(x => String.IsNullOrEmpty(x.ConstraintName) &&
x.ConstraintType == constraintGroup.Key);
if (index != -1)
{
var unnamedConstraint = tableConstraints[index];
var columns = new List <string>(unnamedConstraint.Columns);
foreach (var columnName in columnNames)
{
if (!columns.Contains(columnName))
{
columns.Add(columnName);
}
}
// TODO: set the new columns
}
else
{
tableConstraints.Add(new SqlTableConstraint(constraintGroup.Key, columnNames));
}
}
}
var block = new SequenceOfStatements();
block.Statements.Add(new CreateTableStatement(tableName, tableColumns)
{
IfNotExists = ifNotExists,
Temporary = temporary
});
foreach (var constraint in tableConstraints)
{
block.Statements.Add(new AlterTableStatement(tableName, new AddConstraintAction(constraint)));
}
return(block);
}