public static void CreateClusters(TextWriter writer, PgSchema oldSchema, PgSchema newSchema, SearchPathHelper searchPathHelper)
{
foreach (var newTable in newSchema.GetTables())
{
var oldTable = oldSchema?.GetTable(newTable.Name);
var oldCluster = oldTable?.ClusterIndexName;
var newCluster = newTable.ClusterIndexName;
if (oldCluster == null && newCluster != null ||
oldCluster != null && newCluster != null &&
newCluster.CompareTo(oldCluster) != 0)
{
searchPathHelper.OutputSearchPath(writer);
writer.WriteLine();
writer.Write("ALTER TABLE ");
writer.Write(PgDiffUtils.GetQuotedName(newTable.Name));
writer.Write(" CLUSTER ON ");
writer.Write(PgDiffUtils.GetQuotedName(newCluster));
writer.WriteLine(';');
}
}
}
private static void CheckWithOids(TextWriter writer, PgTable oldTable, PgTable newTable, SearchPathHelper searchPathHelper)
{
if (oldTable.With == null && newTable.With == null || oldTable.With != null && oldTable.With.Equals(newTable.With))
{
return;
}
searchPathHelper.OutputSearchPath(writer);
writer.WriteLine();
writer.WriteLine("ALTER TABLE " + PgDiffUtils.GetQuotedName(newTable.Name));
if (newTable.With == null || "OIDS=false".EqualsIgnoreCase(newTable.With))
{
writer.WriteLine("\tSET WITHOUT OIDS;");
}
else if ("OIDS".EqualsIgnoreCase(newTable.With) || "OIDS=true".EqualsIgnoreCase(newTable.With))
{
writer.WriteLine("\tSET WITH OIDS;");
}
else
{
writer.WriteLine("\tSET " + newTable.With + ";");
}
}
public static void AlterComments(TextWriter writer, PgSchema oldSchema, PgSchema newSchema, SearchPathHelper searchPathHelper)
{
if (oldSchema == null)
{
return;
}
foreach (var oldFun in oldSchema.GetFunctions())
{
var newFunction = newSchema.GetFunction(oldFun.GetSignature());
if (newFunction == null)
{
continue;
}
if (oldFun.Comment == null &&
newFunction.Comment != null ||
oldFun.Comment != null &&
newFunction.Comment != null &&
!oldFun.Comment.Equals(
newFunction.Comment))
{
searchPathHelper.OutputSearchPath(writer);
writer.WriteLine();
writer.Write("COMMENT ON FUNCTION ");
writer.Write(PgDiffUtils.GetQuotedName(newFunction.Name));
writer.Write('(');
var addComma = false;
foreach (var argument in newFunction.GetArguments())
{
if (addComma)
{
writer.Write(", ");
}
else
{
addComma = true;
}
writer.Write(argument.GetDeclaration(false));
}
writer.Write(") IS ");
writer.Write(newFunction.Comment);
writer.WriteLine(';');
}
else if (oldFun.Comment != null && newFunction.Comment == null)
{
searchPathHelper.OutputSearchPath(writer);
writer.WriteLine();
writer.Write("COMMENT ON FUNCTION ");
writer.Write(PgDiffUtils.GetQuotedName(newFunction.Name));
writer.Write('(');
var addComma = false;
foreach (var argument in newFunction.GetArguments())
{
if (addComma)
{
writer.Write(", ");
}
else
{
addComma = true;
}
writer.Write(argument.GetDeclaration(false));
}
writer.WriteLine(") IS NULL;");
}
}
}
public static void AlterViews(TextWriter writer, PgSchema oldSchema, PgSchema newSchema, SearchPathHelper searchPathHelper)
{
if (oldSchema == null)
{
return;
}
foreach (var oldView in oldSchema.GetViews())
{
var newView = newSchema.GetView(oldView.Name);
if (newView == null)
{
continue;
}
DiffDefaultValues(writer, oldView, newView, searchPathHelper);
if (oldView.Comment == null &&
newView.Comment != null ||
oldView.Comment != null &&
newView.Comment != null &&
!oldView.Comment.Equals(
newView.Comment))
{
searchPathHelper.OutputSearchPath(writer);
writer.WriteLine();
writer.Write("COMMENT ON VIEW ");
writer.Write(PgDiffUtils.GetQuotedName(newView.Name));
writer.Write(" IS ");
writer.Write(newView.Comment);
writer.WriteLine(';');
}
else if (oldView.Comment != null && newView.Comment == null)
{
searchPathHelper.OutputSearchPath(writer);
writer.WriteLine();
writer.Write("COMMENT ON VIEW ");
writer.Write(PgDiffUtils.GetQuotedName(newView.Name));
writer.WriteLine(" IS NULL;");
}
var columnNames = newView.ColumnComments.Select(c => c.ColumnName).ToList();
foreach (var columnComment in oldView.ColumnComments)
{
if (!columnNames.Contains(columnComment.ColumnName))
{
columnNames.Add(columnComment.ColumnName);
}
}
foreach (var columnName in columnNames)
{
var oldColumnComment = oldView.ColumnComments.FirstOrDefault(cc => columnName.Equals(cc.ColumnName));
var newColumnComment = newView.ColumnComments.FirstOrDefault(cc => columnName.Equals(cc.ColumnName));
if (oldColumnComment == null && newColumnComment != null ||
oldColumnComment != null && newColumnComment != null &&
!oldColumnComment.Comment.Equals(
newColumnComment.Comment))
{
searchPathHelper.OutputSearchPath(writer);
writer.WriteLine();
writer.Write("COMMENT ON COLUMN ");
writer.Write(PgDiffUtils.GetQuotedName(newView.Name));
writer.Write('.');
writer.Write(PgDiffUtils.GetQuotedName(newColumnComment.ColumnName));
writer.Write(" IS ");
writer.Write(newColumnComment.Comment);
writer.WriteLine(';');
}
else if (oldColumnComment != null &&
newColumnComment == null)
{
searchPathHelper.OutputSearchPath(writer);
writer.WriteLine();
writer.Write("COMMENT ON COLUMN ");
writer.Write(PgDiffUtils.GetQuotedName(newView.Name));
writer.Write('.');
writer.Write(PgDiffUtils.GetQuotedName(oldColumnComment.ColumnName));
writer.WriteLine(" IS NULL;");
}
}
}
}
private static void DiffDefaultValues(TextWriter writer, PgView oldView, PgView newView, SearchPathHelper searchPathHelper)
{
var oldValues = oldView.DefaultValues;
var newValues = newView.DefaultValues;
// modify defaults that are in old view
foreach (var oldValue in oldValues)
{
var found = false;
foreach (var newValue in newValues)
{
if (oldValue.ColumnName.Equals(newValue.ColumnName))
{
found = true;
if (!oldValue._DefaultValue.Equals(newValue._DefaultValue))
{
searchPathHelper.OutputSearchPath(writer);
writer.WriteLine();
writer.Write("ALTER TABLE ");
writer.Write(PgDiffUtils.GetQuotedName(newView.Name));
writer.Write(" ALTER COLUMN ");
writer.Write(PgDiffUtils.GetQuotedName(newValue.ColumnName));
writer.Write(" SET DEFAULT ");
writer.Write(newValue._DefaultValue);
writer.WriteLine(';');
}
break;
}
}
if (!found)
{
searchPathHelper.OutputSearchPath(writer);
writer.WriteLine();
writer.Write("ALTER TABLE ");
writer.Write(PgDiffUtils.GetQuotedName(newView.Name));
writer.Write(" ALTER COLUMN ");
writer.Write(PgDiffUtils.GetQuotedName(oldValue.ColumnName));
writer.WriteLine(" DROP DEFAULT;");
}
}
// add new defaults
foreach (var newValue in newValues)
{
var found = oldValues.Any(ov => newValue.ColumnName.Equals(ov.ColumnName));
if (found)
{
continue;
}
searchPathHelper.OutputSearchPath(writer);
writer.WriteLine();
writer.Write("ALTER TABLE ");
writer.Write(PgDiffUtils.GetQuotedName(newView.Name));
writer.Write(" ALTER COLUMN ");
writer.Write(PgDiffUtils.GetQuotedName(newValue.ColumnName));
writer.Write(" SET DEFAULT ");
writer.Write(newValue._DefaultValue);
writer.WriteLine(';');
}
}
public static void AlterSequences(TextWriter writer, PgDiffArguments arguments,
PgSchema oldSchema, PgSchema newSchema, SearchPathHelper searchPathHelper)
{
if (oldSchema == null)
{
return;
}
var sbSql = new StringBuilder(100);
foreach (var newSequence in newSchema.GetSequences())
{
var oldSequence = oldSchema.GetSequence(newSequence.Name);
if (oldSequence == null)
{
continue;
}
sbSql.Length = 0;
var oldIncrement = oldSequence.Increment;
var newIncrement = newSequence.Increment;
if (newIncrement != null && !newIncrement.Equals(oldIncrement))
{
sbSql.Append("\n\tINCREMENT BY ");
sbSql.Append(newIncrement);
}
var oldMinValue = oldSequence.MinValue;
var newMinValue = newSequence.MinValue;
if (newMinValue == null && oldMinValue != null)
{
sbSql.Append("\n\tNO MINVALUE");
}
else if (newMinValue != null && !newMinValue.Equals(oldMinValue))
{
sbSql.Append("\n\tMINVALUE ");
sbSql.Append(newMinValue);
}
var oldMaxValue = oldSequence.MaxValue;
var newMaxValue = newSequence.MaxValue;
if (newMaxValue == null && oldMaxValue != null)
{
sbSql.Append("\n\tNO MAXVALUE");
}
else if (newMaxValue != null &&
!newMaxValue.Equals(oldMaxValue))
{
sbSql.Append("\n\tMAXVALUE ");
sbSql.Append(newMaxValue);
}
if (!arguments.IgnoreStartWith)
{
var oldStart = oldSequence.StartWith;
var newStart = newSequence.StartWith;
if (newStart != null && !newStart.Equals(oldStart))
{
sbSql.Append("\n\tRESTART WITH ");
sbSql.Append(newStart);
}
}
var oldCache = oldSequence.Cache;
var newCache = newSequence.Cache;
if (newCache != null && !newCache.Equals(oldCache))
{
sbSql.Append("\n\tCACHE ");
sbSql.Append(newCache);
}
var oldCycle = oldSequence.Cycle;
var newCycle = newSequence.Cycle;
if (oldCycle && !newCycle)
{
sbSql.Append("\n\tNO CYCLE");
}
else if (!oldCycle && newCycle)
{
sbSql.Append("\n\tCYCLE");
}
var oldOwnedBy = oldSequence.OwnedBy;
var newOwnedBy = newSequence.OwnedBy;
if (newOwnedBy != null && !newOwnedBy.Equals(oldOwnedBy))
{
sbSql.Append("\n\tOWNED BY ");
sbSql.Append(newOwnedBy);
}
if (sbSql.Length > 0)
{
searchPathHelper.OutputSearchPath(writer);
writer.WriteLine();
writer.Write("ALTER SEQUENCE "
+ PgDiffUtils.GetQuotedName(newSequence.Name));
writer.Write(sbSql.ToString());
writer.WriteLine(';');
}
if (oldSequence.Comment == null &&
newSequence.Comment != null ||
oldSequence.Comment != null &&
newSequence.Comment != null &&
!oldSequence.Comment.Equals(
newSequence.Comment))
{
searchPathHelper.OutputSearchPath(writer);
writer.WriteLine();
writer.Write("COMMENT ON SEQUENCE ");
writer.Write(PgDiffUtils.GetQuotedName(newSequence.Name));
writer.Write(" IS ");
writer.Write(newSequence.Comment);
writer.WriteLine(';');
}
else if (oldSequence.Comment != null && newSequence.Comment == null)
{
searchPathHelper.OutputSearchPath(writer);
writer.WriteLine();
writer.Write("COMMENT ON SEQUENCE ");
writer.Write(newSequence.Name);
writer.WriteLine(" IS NULL;");
}
}
}
public static void AlterComments(TextWriter writer, PgSchema oldSchema, PgSchema newSchema, SearchPathHelper searchPathHelper)
{
if (oldSchema == null)
{
return;
}
foreach (var oldTable in oldSchema.GetTables())
{
var newTable = newSchema.GetTable(oldTable.Name);
if (newTable == null)
{
continue;
}
foreach (var oldConstraint in oldTable.Constraints)
{
var newConstraint =
newTable.GetConstraint(oldConstraint.Name);
if (newConstraint == null)
{
continue;
}
if (oldConstraint.Comment == null &&
newConstraint.Comment != null ||
oldConstraint.Comment != null &&
newConstraint.Comment != null &&
!oldConstraint.Comment.Equals(
newConstraint.Comment))
{
searchPathHelper.OutputSearchPath(writer);
writer.WriteLine();
writer.Write("COMMENT ON ");
if (newConstraint.IsPrimaryKeyConstraint())
{
writer.Write("INDEX ");
writer.Write(PgDiffUtils.GetQuotedName(newConstraint.Name));
}
else
{
writer.Write("CONSTRAINT ");
writer.Write(PgDiffUtils.GetQuotedName(newConstraint.Name));
writer.Write(" ON ");
writer.Write(PgDiffUtils.GetQuotedName(newConstraint.TableName));
}
writer.Write(" IS ");
writer.Write(newConstraint.Comment);
writer.WriteLine(';');
}
else if (oldConstraint.Comment != null && newConstraint.Comment == null)
{
searchPathHelper.OutputSearchPath(writer);
writer.WriteLine();
writer.Write("COMMENT ON ");
if (newConstraint.IsPrimaryKeyConstraint())
{
writer.Write("INDEX ");
writer.Write(PgDiffUtils.GetQuotedName(newConstraint.Name));
}
else
{
writer.Write("CONSTRAINT ");
writer.Write(PgDiffUtils.GetQuotedName(newConstraint.Name));
writer.Write(" ON ");
writer.Write(PgDiffUtils.GetQuotedName(newConstraint.TableName));
}
writer.WriteLine(" IS NULL;");
}
}
}
}
