Update pgsql codefirst

This commit is contained in:
sunkaixuan 2022-07-04 21:44:00 +08:00
parent 973e2c2ee5
commit 1c23c9ec26
3 changed files with 19 additions and 0 deletions

View File

@ -371,6 +371,7 @@ namespace SqlSugar
{ {
this.Context.DbMaintenance.BackupTable(tableName, tableName + DateTime.Now.ToString("yyyyMMddHHmmss"), MaxBackupDataRows); this.Context.DbMaintenance.BackupTable(tableName, tableName + DateTime.Now.ToString("yyyyMMddHHmmss"), MaxBackupDataRows);
} }
ExistLogicEnd(entityColumns);
} }
} }
@ -391,6 +392,10 @@ namespace SqlSugar
this.Context.DbMaintenance.AddPrimaryKey(tableName, item.DbColumnName); this.Context.DbMaintenance.AddPrimaryKey(tableName, item.DbColumnName);
} }
protected virtual void ExistLogicEnd(List<EntityColumnInfo> dbColumns)
{
}
protected virtual void ConvertColumns(List<DbColumnInfo> dbColumns) protected virtual void ConvertColumns(List<DbColumnInfo> dbColumns)
{ {

View File

@ -7,6 +7,16 @@ namespace SqlSugar
{ {
public class PostgreSQLCodeFirst : CodeFirstProvider public class PostgreSQLCodeFirst : CodeFirstProvider
{ {
protected override void ExistLogicEnd(List<EntityColumnInfo> dbColumns)
{
foreach (EntityColumnInfo column in dbColumns)
{
if (column.DefaultValue != null)
{
this.Context.DbMaintenance.AddDefaultValue(column.DbTableName,column.DbColumnName,column.DefaultValue.ToSqlValue());
}
}
}
public override void NoExistLogic(EntityInfo entityInfo) public override void NoExistLogic(EntityInfo entityInfo)
{ {
var tableName = GetTableName(entityInfo); var tableName = GetTableName(entityInfo);

View File

@ -249,6 +249,10 @@ namespace SqlSugar
#endregion #endregion
#region Methods #region Methods
public override bool AddDefaultValue(string tableName, string columnName, string defaultValue)
{
return base.AddDefaultValue(this.SqlBuilder.GetTranslationTableName(tableName), this.SqlBuilder.GetTranslationTableName(columnName), defaultValue);
}
public override bool AddColumnRemark(string columnName, string tableName, string description) public override bool AddColumnRemark(string columnName, string tableName, string description)
{ {
tableName = this.SqlBuilder.GetTranslationTableName(tableName); tableName = this.SqlBuilder.GetTranslationTableName(tableName);