private static void ParseColumn(Parser parser, PgDatabase database)
{
var columnName = parser.ParseIdentifier();
var objectName = ParserUtils.GetObjectName(columnName);
var tableName = ParserUtils.GetSecondObjectName(columnName);
var schemaName = ParserUtils.GetThirdObjectName(columnName);
var schema = database.GetSchema(schemaName);
var table = schema.GetTable(tableName);
if (table == null)
{
var view = schema.GetView(tableName);
parser.Expect("IS");
var comment = GetComment(parser);
if (comment == null)
{
view.RemoveColumnComment(objectName);
}
else
{
view.AddColumnComment(objectName, comment);
}
parser.Expect(";");
}
else
{
var column = table.GetColumn(objectName);
if (column == null)
{
throw new ParserException(string.Format(Resources.CannotFindColumnInTable, columnName, table.Name));
}
parser.Expect("IS");
column.Comment = GetComment(parser);
parser.Expect(";");
}
}
public static void Parse(PgDatabase database, string statement)
{
var parser = new Parser(statement);
parser.Expect("CREATE");
var unique = parser.ExpectOptional("UNIQUE");
parser.Expect("INDEX");
parser.ExpectOptional("CONCURRENTLY");
var indexName = ParserUtils.GetObjectName(parser.ParseIdentifier());
parser.Expect("ON");
var tableName = parser.ParseIdentifier();
var definition = parser.GetRest();
var schemaName = ParserUtils.GetSchemaName(tableName, database);
var schema = database.GetSchema(schemaName);
if (schema == null)
{
throw new Exception(string.Format(Resources.CannotFindSchema, schemaName, statement));
}
var objectName = ParserUtils.GetObjectName(tableName);
var table = schema.GetTable(objectName);
if (table == null)
{
throw new Exception(string.Format(Resources.CannotFindTable, tableName, statement));
}
var index = new PgIndex(indexName);
table.AddIndex(index);
schema.AddIndex(index);
index.Definition = definition.Trim();
index.TableName = table.Name;
index.Unique = unique;
}
public static void Parse(PgDatabase database, string statement)
{
var parser = new Parser(statement);
parser.Expect("CREATE");
parser.ExpectOptional("OR", "REPLACE");
parser.Expect("VIEW");
var viewName = parser.ParseIdentifier();
var columnsExist = parser.ExpectOptional("(");
var columnNames = new List <string>();
if (columnsExist)
{
while (!parser.ExpectOptional(")"))
{
columnNames.Add(ParserUtils.GetObjectName(parser.ParseIdentifier()));
parser.ExpectOptional(",");
}
}
parser.Expect("AS");
var query = parser.GetRest();
var view = new PgView(ParserUtils.GetObjectName(viewName));
view.ColumnNames = columnNames;
view.Query = query;
var schemaName = ParserUtils.GetSchemaName(viewName, database);
var schema = database.GetSchema(schemaName);
if (schema == null)
{
throw new Exception(string.Format(Resources.CannotFindSchema, schemaName, statement));
}
schema.AddView(view);
}
private static void ParseView(Parser parser, PgView view, bool outputIgnoredStatements, string viewName, PgDatabase database)
{
while (!parser.ExpectOptional(";"))
{
if (parser.ExpectOptional("ALTER"))
{
parser.ExpectOptional("COLUMN");
var columnName = ParserUtils.GetObjectName(parser.ParseIdentifier());
if (parser.ExpectOptional("SET", "DEFAULT"))
{
var expression = parser.GetExpression();
view.AddColumnDefaultValue(columnName, expression);
}
else if (parser.ExpectOptional("DROP", "DEFAULT"))
{
view.RemoveColumnDefaultValue(columnName);
}
else
{
parser.ThrowUnsupportedCommand();
}
}
else if (parser.ExpectOptional("OWNER", "TO"))
{
// we do not parse this one so we just consume the identifier
if (outputIgnoredStatements)
{
database.IgnoredStatements.Add($"ALTER TABLE {viewName} OWNER TO {parser.ParseIdentifier()};");
}
else
{
parser.ParseIdentifier();
}
}
else
{
parser.ThrowUnsupportedCommand();
}
}
}
public static void Parse(PgDatabase database, string statement, bool outputIgnoredStatements)
{
var parser = new Parser(statement);
parser.Expect("ALTER", "SEQUENCE");
var sequenceName = parser.ParseIdentifier();
var schemaName = ParserUtils.GetSchemaName(sequenceName, database);
var schema = database.GetSchema(schemaName);
if (schema == null)
{
throw new Exception(string.Format(Resources.CannotFindSchema, schemaName, statement));
}
var objectName = ParserUtils.GetObjectName(sequenceName);
var sequence = schema.GetSequence(objectName);
if (sequence == null)
{
throw new Exception(string.Format(Resources.CannotFindSequence, sequenceName, statement));
}
while (!parser.ExpectOptional(";"))
{
if (parser.ExpectOptional("OWNED", "BY"))
{
sequence.OwnedBy = parser.ExpectOptional("NONE")
? null
: parser.GetExpression();
}
else
{
parser.ThrowUnsupportedCommand();
}
}
}
private static void ParseIndex(Parser parser, PgDatabase database)
{
var indexName = parser.ParseIdentifier();
var objectName = ParserUtils.GetObjectName(indexName);
var schemaName = ParserUtils.GetSchemaName(indexName, database);
var schema = database.GetSchema(schemaName);
var index = schema.GetIndex(objectName);
if (index == null)
{
var primaryKey = schema.GetPrimaryKey(objectName);
parser.Expect("IS");
primaryKey.Comment = GetComment(parser);
parser.Expect(";");
}
else
{
parser.Expect("IS");
index.Comment = GetComment(parser);
parser.Expect(";");
}
}
public static void Parse(PgDatabase database, string statement)
{
var parser = new Parser(statement);
parser.Expect("CREATE", "SCHEMA");
if (parser.ExpectOptional("AUTHORIZATION"))
{
var schema = new PgSchema(ParserUtils.GetObjectName(parser.ParseIdentifier()));
database.Schemas.Add(schema);
schema.Authorization = schema.Name;
var definition = parser.GetRest();
if (!string.IsNullOrEmpty(definition))
{
schema.Definition = definition;
}
}
else
{
var schema = new PgSchema(ParserUtils.GetObjectName(parser.ParseIdentifier()));
database.Schemas.Add(schema);
if (parser.ExpectOptional("AUTHORIZATION"))
{
schema.Authorization = ParserUtils.GetObjectName(parser.ParseIdentifier());
}
var definition = parser.GetRest();
if (!string.IsNullOrEmpty(definition))
{
schema.Definition = definition;
}
}
}
private static void ParseAddForeignKey(Parser parser, PgTable table)
{
var columnNames = new List <string>();
parser.Expect("(");
while (!parser.ExpectOptional(")"))
{
columnNames.Add(ParserUtils.GetObjectName(parser.ParseIdentifier()));
if (parser.ExpectOptional(")"))
{
break;
}
parser.Expect(",");
}
var constraintName = ParserUtils.GenerateName(table.Name + "_", columnNames, "_fkey");
var constraint = new PgConstraint(constraintName);
table.AddConstraint(constraint);
constraint.Definition = parser.GetExpression();
constraint.TableName = table.Name;
}
public static void Parse(PgDatabase database, string statement, bool outputIgnoredStatements)
{
var parser = new Parser(statement);
parser.Expect("ALTER", "VIEW");
var viewName = parser.ParseIdentifier();
var schemaName = ParserUtils.GetSchemaName(viewName, database);
var schema = database.GetSchema(schemaName);
if (schema == null)
{
throw new Exception(string.Format(Resources.CannotFindSchema, schemaName, statement));
}
var objectName = ParserUtils.GetObjectName(viewName);
var view = schema.GetView(objectName);
if (view == null)
{
throw new Exception(string.Format(Resources.CannotFindView, viewName, statement));
}
while (!parser.ExpectOptional(";"))
{
if (parser.ExpectOptional("ALTER"))
{
parser.ExpectOptional("COLUMN");
var columnName = ParserUtils.GetObjectName(parser.ParseIdentifier());
if (parser.ExpectOptional("SET", "DEFAULT"))
{
var expression = parser.GetExpression();
view.AddColumnDefaultValue(columnName, expression);
}
else if (parser.ExpectOptional("DROP", "DEFAULT"))
{
view.RemoveColumnDefaultValue(columnName);
}
else
{
parser.ThrowUnsupportedCommand();
}
}
else if (parser.ExpectOptional("OWNER", "TO"))
{
// we do not parse this one so we just consume the identifier
if (outputIgnoredStatements)
{
database.IgnoredStatements.Add($"ALTER TABLE {viewName} OWNER TO {parser.ParseIdentifier()};");
}
else
{
parser.ParseIdentifier();
}
}
else
{
parser.ThrowUnsupportedCommand();
}
}
}
private static void ParseFunction(Parser parser, PgDatabase database)
{
var functionName = parser.ParseIdentifier();
var objectName = ParserUtils.GetObjectName(functionName);
var schemaName = ParserUtils.GetSchemaName(functionName, database);
var schema = database.GetSchema(schemaName);
parser.Expect("(");
var tmpFunction = new PgFunction();
tmpFunction.Name = objectName;
while (!parser.ExpectOptional(")"))
{
string mode;
if (parser.ExpectOptional("IN"))
{
mode = "IN";
}
else if (parser.ExpectOptional("OUT"))
{
mode = "OUT";
}
else if (parser.ExpectOptional("INOUT"))
{
mode = "INOUT";
}
else if (parser.ExpectOptional("VARIADIC"))
{
mode = "VARIADIC";
}
else
{
mode = null;
}
var position = parser.GetPosition();
string argumentName = null;
var dataType = parser.ParseDataType();
var position2 = parser.GetPosition();
if (!parser.ExpectOptional(")") && !parser.ExpectOptional(","))
{
parser.SetPosition(position);
argumentName = ParserUtils.GetObjectName(parser.ParseIdentifier());
dataType = parser.ParseDataType();
}
else
{
parser.SetPosition(position2);
}
var argument = new PgFunction.Argument
{
DataType = dataType,
Mode = mode,
Name = argumentName
};
tmpFunction.AddArgument(argument);
if (parser.ExpectOptional(")"))
{
break;
}
parser.Expect(",");
}
var function =
schema.GetFunction(tmpFunction.GetSignature());
parser.Expect("IS");
function.Comment = GetComment(parser);
parser.Expect(";");
}
public static void Parse(PgDatabase database, string statement)
{
var parser = new Parser(statement);
parser.Expect("CREATE", "SEQUENCE");
var sequenceName = parser.ParseIdentifier();
var sequence = new PgSequence(ParserUtils.GetObjectName(sequenceName));
var schemaName = ParserUtils.GetSchemaName(sequenceName, database);
var schema = database.GetSchema(schemaName);
if (schema == null)
{
throw new Exception(string.Format(Resources.CannotFindSchema, schemaName, statement));
}
schema.AddSequence(sequence);
while (!parser.ExpectOptional(";"))
{
if (parser.ExpectOptional("INCREMENT"))
{
parser.ExpectOptional("BY");
sequence.Increment = parser.ParseString();
}
else if (parser.ExpectOptional("MINVALUE"))
{
sequence.MinValue = parser.ParseString();
}
else if (parser.ExpectOptional("MAXVALUE"))
{
sequence.MaxValue = parser.ParseString();
}
else if (parser.ExpectOptional("START"))
{
parser.ExpectOptional("WITH");
sequence.StartWith = parser.ParseString();
}
else if (parser.ExpectOptional("CACHE"))
{
sequence.Cache = parser.ParseString();
}
else if (parser.ExpectOptional("CYCLE"))
{
sequence.Cycle = true;
}
else if (parser.ExpectOptional("OWNED", "BY"))
{
sequence.OwnedBy = parser.ExpectOptional("NONE")
? null
: ParserUtils.GetObjectName(parser.ParseIdentifier());
}
else if (parser.ExpectOptional("NO"))
{
if (parser.ExpectOptional("MINVALUE"))
{
sequence.MinValue = null;
}
else if (parser.ExpectOptional("MAXVALUE"))
{
sequence.MaxValue = null;
}
else if (parser.ExpectOptional("CYCLE"))
{
sequence.Cycle = false;
}
else
{
parser.ThrowUnsupportedCommand();
}
}
else
{
parser.ThrowUnsupportedCommand();
}
}
}
public static void Parse(PgDatabase database, string statement)
{
var parser = new Parser(statement);
parser.Expect("CREATE", "TABLE");
// Optional IF NOT EXISTS, irrelevant for our purposes
parser.ExpectOptional("IF", "NOT", "EXISTS");
var tableName = parser.ParseIdentifier();
var table = new PgTable(ParserUtils.GetObjectName(tableName));
var schemaName = ParserUtils.GetSchemaName(tableName, database);
var schema = database.GetSchema(schemaName);
if (schema == null)
{
throw new Exception(string.Format(Resources.CannotFindSchema, schemaName, statement));
}
schema.AddTable(table);
parser.Expect("(");
while (!parser.ExpectOptional(")"))
{
if (parser.ExpectOptional("CONSTRAINT"))
{
ParseConstraint(parser, table);
}
else
{
ParseColumn(parser, table);
}
if (parser.ExpectOptional(")"))
{
break;
}
parser.Expect(",");
}
while (!parser.ExpectOptional(";"))
{
if (parser.ExpectOptional("INHERITS"))
{
ParseInherits(parser, table);
}
else if (parser.ExpectOptional("WITHOUT"))
{
table.With = "OIDS=false";
}
else if (parser.ExpectOptional("WITH"))
{
if (parser.ExpectOptional("OIDS") || parser.ExpectOptional("OIDS=true"))
{
table.With = "OIDS=true";
}
else if (parser.ExpectOptional("OIDS=false"))
{
table.With = "OIDS=false";
}
else
{
table.With = parser.GetExpression();
}
}
else if (parser.ExpectOptional("TABLESPACE"))
{
table.Tablespace = parser.ParseString();
}
else
{
parser.ThrowUnsupportedCommand();
}
}
}
public static void Parse(PgDatabase database, string statement)
{
var parser = new Parser(statement);
parser.Expect("CREATE");
parser.ExpectOptional("OR", "REPLACE");
parser.Expect("FUNCTION");
var functionName = parser.ParseIdentifier();
var schemaName = ParserUtils.GetSchemaName(functionName, database);
var schema = database.GetSchema(schemaName);
if (schema == null)
{
throw new Exception(string.Format(Resources.CannotFindSchema, schemaName, statement));
}
var function = new PgFunction();
function.Name = ParserUtils.GetObjectName(functionName);
schema.AddFunction(function);
parser.Expect("(");
while (!parser.ExpectOptional(")"))
{
string mode;
if (parser.ExpectOptional("IN"))
{
mode = "IN";
}
else if (parser.ExpectOptional("OUT"))
{
mode = "OUT";
}
else if (parser.ExpectOptional("INOUT"))
{
mode = "INOUT";
}
else if (parser.ExpectOptional("VARIADIC"))
{
mode = "VARIADIC";
}
else
{
mode = null;
}
var position = parser.GetPosition();
string argumentName = null;
var dataType = parser.ParseDataType();
var position2 = parser.GetPosition();
if (!parser.ExpectOptional(")") && !parser.ExpectOptional(",") &&
!parser.ExpectOptional("=") &&
!parser.ExpectOptional("DEFAULT"))
{
parser.SetPosition(position);
argumentName = ParserUtils.GetObjectName(parser.ParseIdentifier());
dataType = parser.ParseDataType();
}
else
{
parser.SetPosition(position2);
}
string defaultExpression;
if (parser.ExpectOptional("=") || parser.ExpectOptional("DEFAULT"))
{
defaultExpression = parser.GetExpression();
}
else
{
defaultExpression = null;
}
var argument = new PgFunction.Argument
{
DataType = dataType,
DefaultExpression = defaultExpression,
Mode = mode,
Name = argumentName
};
function.AddArgument(argument);
if (parser.ExpectOptional(")"))
{
break;
}
parser.Expect(",");
}
function.Body = parser.GetRest();
}
public static void Parse(PgDatabase database, string statement, bool ignoreSlonyTriggers)
{
var parser = new Parser(statement);
parser.Expect("CREATE", "TRIGGER");
var triggerName = parser.ParseIdentifier();
var objectName = ParserUtils.GetObjectName(triggerName);
var trigger = new PgTrigger();
trigger.Name = objectName;
if (parser.ExpectOptional("BEFORE"))
{
trigger.Before = true;
}
else if (parser.ExpectOptional("AFTER"))
{
trigger.Before = false;
}
var first = true;
while (true)
{
if (!first && !parser.ExpectOptional("OR"))
{
break;
}
if (parser.ExpectOptional("INSERT"))
{
trigger.OnInsert = true;
}
else if (parser.ExpectOptional("UPDATE"))
{
trigger.OnUpdate = true;
if (parser.ExpectOptional("OF"))
{
do
{
trigger.AddUpdateColumn(parser.ParseIdentifier());
}while (parser.ExpectOptional(","));
}
}
else if (parser.ExpectOptional("DELETE"))
{
trigger.OnDelete = true;
}
else if (parser.ExpectOptional("TRUNCATE"))
{
trigger.OnTruncate = true;
}
else if (first)
{
break;
}
else
{
parser.ThrowUnsupportedCommand();
}
first = false;
}
parser.Expect("ON");
var tableName = parser.ParseIdentifier();
trigger.TableName = ParserUtils.GetObjectName(tableName);
if (parser.ExpectOptional("FOR"))
{
parser.ExpectOptional("EACH");
if (parser.ExpectOptional("ROW"))
{
trigger.ForEachRow = true;
}
else if (parser.ExpectOptional("STATEMENT"))
{
trigger.ForEachRow = false;
}
else
{
parser.ThrowUnsupportedCommand();
}
}
if (parser.ExpectOptional("WHEN"))
{
parser.Expect("(");
trigger.When = parser.GetExpression();
parser.Expect(")");
}
parser.Expect("EXECUTE", "PROCEDURE");
trigger.Function = parser.GetRest();
var ignoreSlonyTrigger = ignoreSlonyTriggers &&
("_slony_logtrigger".Equals(trigger.Name) ||
"_slony_denyaccess".Equals(trigger.Name));
if (!ignoreSlonyTrigger)
{
var tableSchema = database.GetSchema(ParserUtils.GetSchemaName(tableName, database));
tableSchema.GetTable(trigger.TableName).AddTrigger(trigger);
}
}
private static void ParseAlterColumn(Parser parser, PgTable table)
{
parser.ExpectOptional("COLUMN");
var columnName = ParserUtils.GetObjectName(parser.ParseIdentifier());
if (parser.ExpectOptional("SET"))
{
if (parser.ExpectOptional("STATISTICS"))
{
var column = table.GetColumn(columnName);
if (column == null)
{
throw new Exception(string.Format(Resources.CannotFindTableColumn, columnName, table.Name, parser.GetString()));
}
column.Statistics = parser.ParseInteger();
}
else if (parser.ExpectOptional("DEFAULT"))
{
var defaultValue = parser.GetExpression();
if (table.ContainsColumn(columnName))
{
var column = table.GetColumn(columnName);
if (column == null)
{
throw new Exception(string.Format(Resources.CannotFindTableColumn, columnName, table.Name, parser.GetString()));
}
column.DefaultValue = defaultValue;
}
else
{
throw new ParserException(string.Format(Resources.CannotFindColumnInTable, columnName, table.Name));
}
}
else if (parser.ExpectOptional("STORAGE"))
{
var column = table.GetColumn(columnName);
if (column == null)
{
throw new Exception(string.Format(Resources.CannotFindTableColumn, columnName, table.Name,
parser.GetString()));
}
if (parser.ExpectOptional("PLAIN"))
{
column.Storage = "PLAIN";
}
else if (parser.ExpectOptional("EXTERNAL"))
{
column.Storage = "EXTERNAL";
}
else if (parser.ExpectOptional("EXTENDED"))
{
column.Storage = "EXTENDED";
}
else if (parser.ExpectOptional("MAIN"))
{
column.Storage = "MAIN";
}
else
{
parser.ThrowUnsupportedCommand();
}
}
else
{
parser.ThrowUnsupportedCommand();
}
}
else
{
parser.ThrowUnsupportedCommand();
}
}
public static void Parse(PgDatabase database, string statement, bool outputIgnoredStatements)
{
var parser = new Parser(statement);
parser.Expect("ALTER", "TABLE");
parser.ExpectOptional("ONLY");
var tableName = parser.ParseIdentifier();
var schemaName = ParserUtils.GetSchemaName(tableName, database);
var schema = database.GetSchema(schemaName);
if (schema == null)
{
throw new Exception(string.Format(Resources.CannotFindSchema, schemaName, statement));
}
var objectName = ParserUtils.GetObjectName(tableName);
var table = schema.GetTable(objectName);
if (table == null)
{
var view = schema.GetView(objectName);
if (view != null)
{
ParseView(parser, view, outputIgnoredStatements, tableName, database);
return;
}
var sequence = schema.GetSequence(objectName);
if (sequence != null)
{
ParseSequence(parser, sequence, outputIgnoredStatements, tableName, database);
return;
}
throw new Exception(string.Format(Resources.CannotFindObject, tableName, statement));
}
while (!parser.ExpectOptional(";"))
{
if (parser.ExpectOptional("ALTER"))
{
ParseAlterColumn(parser, table);
}
else if (parser.ExpectOptional("CLUSTER", "ON"))
{
table.ClusterIndexName = ParserUtils.GetObjectName(parser.ParseIdentifier());
}
else if (parser.ExpectOptional("OWNER", "TO"))
{
if (outputIgnoredStatements)
{
database.IgnoredStatements.Add($"ALTER TABLE {tableName} OWNER TO {parser.ParseIdentifier()};");
}
else
{
parser.ParseIdentifier();
}
}
else if (parser.ExpectOptional("ADD"))
{
if (parser.ExpectOptional("FOREIGN", "KEY"))
{
ParseAddForeignKey(parser, table);
}
else if (parser.ExpectOptional("CONSTRAINT"))
{
ParseAddConstraint(parser, table, schema);
}
else
{
parser.ThrowUnsupportedCommand();
}
}
else if (parser.ExpectOptional("ENABLE"))
{
ParseEnable(parser, outputIgnoredStatements, tableName, database);
}
else if (parser.ExpectOptional("DISABLE"))
{
ParseDisable(parser, outputIgnoredStatements, tableName, database);
}
else
{
parser.ThrowUnsupportedCommand();
}
if (parser.ExpectOptional(";"))
{
break;
}
parser.Expect(",");
}
}
