Browse Source

Code First

pull/12/MERGE
sunkaixuan 8 years ago
parent
commit
84fd78cf6f
  1. 4
      SqlServerTest/Demos/5_CodeFirst.cs
  2. 7
      SqlSugar/Abstract/CodeFirstProvider/CodeFirstProvider.cs
  3. 2
      SqlSugar/Abstract/DbMaintenanceProvider/Methods.cs

4
SqlServerTest/Demos/5_CodeFirst.cs

@ -11,8 +11,8 @@ namespace OrmTest.Demo
[SugarColumn(IsNullable =false ,IsPrimaryKey =true,IsIdentity =true)] [SugarColumn(IsNullable =false ,IsPrimaryKey =true,IsIdentity =true)]
public int Id { get; set; } public int Id { get; set; }
[SugarColumn(Length = 21)] [SugarColumn(Length = 21,OldColumnName = "Name2")]
public string Name { get; set; } public string Name{ get; set; }
[SugarColumn(IsNullable = true)] [SugarColumn(IsNullable = true)]
public bool IsOk { get; set; } public bool IsOk { get; set; }
public Guid Guid { get; set; } public Guid Guid { get; set; }

7
SqlSugar/Abstract/CodeFirstProvider/CodeFirstProvider.cs

@ -70,6 +70,7 @@ namespace SqlSugar
private void NoExistLogic(EntityInfo entityInfo) private void NoExistLogic(EntityInfo entityInfo)
{ {
var tableName = GetTableName(entityInfo); var tableName = GetTableName(entityInfo);
Check.Exception(entityInfo.Columns.Where(it => it.IsPrimarykey).Count() > 1, "Use Code First ,The primary key must not exceed 1");
List<DbColumnInfo> columns = new List<DbColumnInfo>(); List<DbColumnInfo> columns = new List<DbColumnInfo>();
if (entityInfo.Columns.IsValuable()) if (entityInfo.Columns.IsValuable())
{ {
@ -80,6 +81,11 @@ namespace SqlSugar
} }
} }
this.Context.DbMaintenance.CreateTable(tableName, columns); this.Context.DbMaintenance.CreateTable(tableName, columns);
var pkColumns= entityInfo.Columns.Where(it => it.IsPrimarykey).ToList();
foreach (var item in pkColumns)
{
this.Context.DbMaintenance.AddPrimaryKey(tableName, item.DbColumnName);
}
} }
private void ExistLogic(EntityInfo entityInfo) private void ExistLogic(EntityInfo entityInfo)
{ {
@ -133,6 +139,7 @@ namespace SqlSugar
foreach (var item in entityColumns) foreach (var item in entityColumns)
{ {
var dbColumn = dbColumns.FirstOrDefault(dc => dc.DbColumnName.Equals(item.DbColumnName, StringComparison.CurrentCultureIgnoreCase)); var dbColumn = dbColumns.FirstOrDefault(dc => dc.DbColumnName.Equals(item.DbColumnName, StringComparison.CurrentCultureIgnoreCase));
if (dbColumn == null) continue;
var pkDiff = item.IsPrimarykey != dbColumn.IsPrimarykey; var pkDiff = item.IsPrimarykey != dbColumn.IsPrimarykey;
var idEntityDiff = item.IsIdentity != dbColumn.IsIdentity; var idEntityDiff = item.IsIdentity != dbColumn.IsIdentity;
if (dbColumn != null && pkDiff && !idEntityDiff) if (dbColumn != null && pkDiff && !idEntityDiff)

2
SqlSugar/Abstract/DbMaintenanceProvider/Methods.cs

@ -180,7 +180,7 @@ namespace SqlSugar
string dataType = item.DataType; string dataType = item.DataType;
string dataSize = item.Length > 0 ? string.Format("({0})", item.Length) : null; string dataSize = item.Length > 0 ? string.Format("({0})", item.Length) : null;
string nullType = item.IsNullable ? this.CreateTableNull : CreateTableNotNull; string nullType = item.IsNullable ? this.CreateTableNull : CreateTableNotNull;
string primaryKey = item.IsPrimarykey ? this.CreateTablePirmaryKey : null; string primaryKey = null;
string identity = item.IsIdentity ? this.CreateTableIdentity : null; string identity = item.IsIdentity ? this.CreateTableIdentity : null;
string addItem = string.Format(this.CreateTableColumn, columnName, dataType, dataSize, nullType, primaryKey, identity); string addItem = string.Format(this.CreateTableColumn, columnName, dataType, dataSize, nullType, primaryKey, identity);
columnArray.Add(addItem); columnArray.Add(addItem);

Loading…
Cancel
Save