MongoDB学习笔记~为IMongoRepository接口更新指定字段

回到目录

对于MongoDB来说，它的更新建议是对指定字段来说的，即不是把对象里的所有字段都进行update，而是按需去更新，这在性能上是最优的，这当然也是非常容易理解的，我们今天要实现的就是这种按需去更新，并且，我还是不希望将MongoDB的内核暴露出去，这时，我想到了EF时候的按需要更新，即为实体哪些属性赋值就更新哪些属性；这个功能实际上使用了表达式树，将你的属性和属性值存储到Expression里，然后在update方法内部再进行解析即可，具体代码如下

public void Update<T>(System.Linq.Expressions.Expression<Action<T>> entity) where T : class
{
var query = new QueryDocument();
var fieldList = new List<UpdateDefinition<TEntity>>();

var param = entity.Body as MemberInitExpression;
foreach (var item in param.Bindings)
{
string propertyName = item.Member.Name;
object propertyValue;
var memberAssignment = item as MemberAssignment;
if (memberAssignment.Expression.NodeType == ExpressionType.Constant)
{
propertyValue = (memberAssignment.Expression as ConstantExpression).Value;
}
else
{
propertyValue = Expression.Lambda(memberAssignment.Expression, null).Compile().DynamicInvoke();
}

if (propertyName != EntityKey)//更新集中不能有实体键_id
{
fieldList.Add(Builders<TEntity>.Update.Set(propertyName, propertyValue));
}
else
{
query = new QueryDocument("_id",new ObjectId(propertyValue.ToString()));
}

}

ForWait(() => _table.UpdateOneAsync(query, Builders<TEntity>.Update.Combine(fieldList)));
}

其实在方法调用上也是非常容易的，我们来看这个例子

[HttpPost]
public ActionResult Edit(WebManageUsers entity)
{
if (ModelState.IsValid)
{
_webManageUsersRepository.Update<WebManageUsers>(i => new WebManageUsers
{
Id = entity.Id,
LoginName = entity.LoginName
});
return RedirectToAction("Index");
}
ModelState.AddModelError("", "请认真填写表单！");
return View();
}

通过上面代码我们可以看到，只是将需要更新的字段进行赋值即可！

回到目录