This commit is contained in:
skx
2020-10-25 20:17:52 +08:00
parent 2f5e1e3b02
commit 11e15ac2b0
3 changed files with 156 additions and 53 deletions

View File

@@ -56,6 +56,9 @@ namespace OrmTest
}; };
db.Insertable(insertObjs).UseSqlServer().ExecuteBlueCopy(); db.Insertable(insertObjs).UseSqlServer().ExecuteBlueCopy();
db.CodeFirst.InitTables<SubInsertTest, SubInsertTestItem, SubInsertTestItem1, SubInsertTestItem2>();
Console.WriteLine("SubInsert Start");
db.Insertable(new Order() db.Insertable(new Order()
{ {
Name = "订单 1", Name = "订单 1",
@@ -74,8 +77,8 @@ namespace OrmTest
}) })
.AddSubList(it => it.Items.First().OrderId).ExecuteReturnPrimaryKey(); .AddSubList(it => it.Items.First().OrderId).ExecuteReturnPrimaryKey();
db.CodeFirst.InitTables<SubInsertTest, SubInsertTestItem, SubInsertTestItem1, SubInsertTestItem2>(); db.Insertable(new List<SubInsertTest>() {
db.Insertable(new SubInsertTest() new SubInsertTest()
{ {
Name="aa", Name="aa",
SubInsertTestItem1=new SubInsertTestItem1() { SubInsertTestItem1=new SubInsertTestItem1() {
@@ -86,9 +89,22 @@ namespace OrmTest
Name ="item" , Name ="item" ,
TestId=2 TestId=2
} }
},
new SubInsertTest()
{
Name="aa",
SubInsertTestItem1=new SubInsertTestItem1() {
a="nn"
},
SubInsertTestItem=new SubInsertTestItem()
{
Name ="item" ,
TestId=2
}
}
}) })
.AddSubList(it => it.SubInsertTestItem1)
.AddSubList(it => it.SubInsertTestItem.TestId) .AddSubList(it => it.SubInsertTestItem.TestId)
.AddSubList(it => it.SubInsertTestItem1)
.ExecuteReturnPrimaryKey(); .ExecuteReturnPrimaryKey();
Console.WriteLine("#### Insertable End ####"); Console.WriteLine("#### Insertable End ####");

View File

@@ -256,14 +256,17 @@ namespace SqlSugar
{ {
Check.Exception(GetPrimaryKeys().Count == 0, typeof(T).Name + " need Primary key"); Check.Exception(GetPrimaryKeys().Count == 0, typeof(T).Name + " need Primary key");
Check.Exception(GetPrimaryKeys().Count > 1, typeof(T).Name + "Multiple primary keys are not supported"); Check.Exception(GetPrimaryKeys().Count > 1, typeof(T).Name + "Multiple primary keys are not supported");
Check.Exception(this.InsertObjs.Count() > 1, "SubInserable No Support Insertable(List<T>)"); //Check.Exception(this.InsertObjs.Count() > 1, "SubInserable No Support Insertable(List<T>)");
//Check.Exception(items.ToString().Contains(".First().")==false, items.ToString()+ " not supported "); //Check.Exception(items.ToString().Contains(".First().")==false, items.ToString()+ " not supported ");
if (this.InsertObjs == null || this.InsertObjs.Count() == 0)
{
return new SubInsertable<T>();
}
string subMemberName; string subMemberName;
object sublist; object sublist;
SubInsertable<T> result = new SubInsertable<T>(); SubInsertable<T> result = new SubInsertable<T>();
result.GetList(this.InsertObjs,items, out subMemberName, out sublist); result.GetList(this.InsertObjs,items, out subMemberName, out sublist);
result.InsertObject = this.InsertObjs.First(); result.InsertObjects = this.InsertObjs;
result.Context = this.Context; result.Context = this.Context;
result.SubList = new Dictionary<string, object>(); result.SubList = new Dictionary<string, object>();
result.SubList.Add(subMemberName, sublist); result.SubList.Add(subMemberName, sublist);

View File

@@ -12,27 +12,35 @@ namespace SqlSugar
internal EntityInfo Entity { get; set; } internal EntityInfo Entity { get; set; }
internal Dictionary<string, object> SubList { get; set; } internal Dictionary<string, object> SubList { get; set; }
internal SqlSugarProvider Context { get; set; } internal SqlSugarProvider Context { get; set; }
internal T InsertObject { get; set; } internal T [] InsertObjects { get; set; }
internal InsertBuilder InsertBuilder { get; set; } internal InsertBuilder InsertBuilder { get; set; }
internal string Pk { get; set; } internal string Pk { get; set; }
public ISubInsertable<T> AddSubList(Expression<Func<T, object>> items) public ISubInsertable<T> AddSubList(Expression<Func<T, object>> items)
{
if (InsertObjects != null&&InsertObjects.Count() > 0)
{ {
string subMemberName; string subMemberName;
object sublist; object sublist;
GetList(new T[] { InsertObject }, items,out subMemberName,out sublist); GetList(InsertObjects, items, out subMemberName, out sublist);
if (!this.SubList.ContainsKey(subMemberName)) if (!this.SubList.ContainsKey(subMemberName))
{ {
this.SubList.Add(subMemberName, sublist); this.SubList.Add(subMemberName, sublist);
} }
}
return this; return this;
} }
public object ExecuteReturnPrimaryKey() public object ExecuteReturnPrimaryKey()
{
if (InsertObjects != null && InsertObjects.Count()>0)
{
int count = 1;
foreach (var InsertObject in InsertObjects)
{ {
List<ConditionalModel> conModel = new List<ConditionalModel>(); List<ConditionalModel> conModel = new List<ConditionalModel>();
int id = this.Context.Insertable(InsertObject).ExecuteReturnIdentity(); int id = this.Context.Insertable(InsertObject).ExecuteReturnIdentity();
int count = 1;
object pkValue = null; object pkValue = null;
var qureyable = this.Context.Queryable<T>(); var qureyable = this.Context.Queryable<T>();
if (id.ObjToInt() == 0) if (id.ObjToInt() == 0)
@@ -83,8 +91,84 @@ namespace SqlSugar
} }
count += this.Context.Insertable(insertDictionary).AS(subEntity.DbTableName).ExecuteCommand(); count += this.Context.Insertable(insertDictionary).AS(subEntity.DbTableName).ExecuteCommand();
} }
}
return count; return count;
} }
else
{
return 0;
}
}
public async Task<object> ExecuteReturnPrimaryKeyAsync()
{
if (InsertObjects != null && InsertObjects.Count() > 0)
{
int count = 1;
foreach (var InsertObject in InsertObjects)
{
List<ConditionalModel> conModel = new List<ConditionalModel>();
int id = await this.Context.Insertable(InsertObject).ExecuteReturnIdentityAsync();
object pkValue = null;
var qureyable = this.Context.Queryable<T>();
if (id.ObjToInt() == 0)
{
var primaryProperty = this.Entity.Columns.FirstOrDefault(it =>
it.PropertyName.Equals(this.Pk, StringComparison.CurrentCultureIgnoreCase) ||
it.DbColumnName.Equals(this.Pk, StringComparison.CurrentCultureIgnoreCase)
);
pkValue = primaryProperty.PropertyInfo.GetValue(InsertObject);
qureyable.In(pkValue);
}
else
{
qureyable.In(id);
pkValue = id;
}
var data =await qureyable.FirstAsync();
foreach (var item in this.SubList)
{
Dictionary<string, object> insertDictionary = new Dictionary<string, object>();
if (item.Value == null)
{
continue;
}
EntityInfo subEntity = null;
if (item.Value is IEnumerable<object>)
{
var list = item.Value as IEnumerable<object>;
if (list.Count() == 0)
{
continue;
}
var type = list.First().GetType();
this.Context.InitMappingInfo(type);
subEntity = this.Context.EntityMaintenance.GetEntityInfo(type);
foreach (var sbItem in list)
{
SetItems(insertDictionary, sbItem, subEntity, item.Key, pkValue);
}
}
else if (item.Value.GetType().IsClass())
{
var type = item.Value.GetType();
this.Context.InitMappingInfo(type);
subEntity = this.Context.EntityMaintenance.GetEntityInfo(type);
SetItems(insertDictionary, item.Value, subEntity, item.Key, pkValue);
}
count +=await this.Context.Insertable(insertDictionary).AS(subEntity.DbTableName).ExecuteCommandAsync();
}
}
return count;
}
else
{
return 0;
}
}
public void GetList(T[] inserts,Expression<Func<T, object>> items, out string subMemberName, out object sublist) public void GetList(T[] inserts,Expression<Func<T, object>> items, out string subMemberName, out object sublist)
{ {
var lambdaExpression = (items as LambdaExpression).Body; var lambdaExpression = (items as LambdaExpression).Body;