MongoDB學習筆記~為IMongoRepository接口更新指定字段 - lori－IT工程師數位筆記本

文章出處

對于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();
        }