前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >entity framework不查数据库修改或排除指定字段集合通用方法

entity framework不查数据库修改或排除指定字段集合通用方法

作者头像
磊哥
发布2018-05-08 16:21:00
7840
发布2018-05-08 16:21:00
举报
文章被收录于专栏:王磊的博客王磊的博客

其中DataDBEntities为数据库实体对象,代码如下:

下载地址:http://files.cnblogs.com/stone_w/EFDBHelper.zip

代码语言:javascript
复制
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Reflection;
using System.Data.Objects.DataClasses;
public class EFDBHelper
{

    #region 不查数据库修改信息
    /// <summary>
    /// 不查数据库修改信息
    /// </summary>
    /// <typeparam name="T"></typeparam>
    /// <param name="entity"></param>
    /// <param name="db"></param>
    /// <param name="updateFiledType"></param>
    /// <param name="fileds"></param>
    /// <returns></returns>
    public static int Update<T>(T entity, DataDBEntities db,
        EnumUpdateFiledType updateFiledType, params string[] fileds)
    {
        if (null == db || null == entity)
        { // 参数有误
            return 0;
        }
        Type _type = typeof(T);
        db.AttachTo(_type.Name, entity);
        if (null == fileds || fileds.Length == 0)
        { // 全字段操作
            db.ObjectStateManager.ChangeObjectState(entity, System.Data.EntityState.Modified);      // 手动设置为修改状态
        }
        else
        { // 部分字段操作
            var _stateEntry = db.ObjectStateManager.GetObjectStateEntry(entity);                    // 得到实体状态
            if (EnumUpdateFiledType.字段修改 == updateFiledType)
            { // 部分字段修改
                for (int i = 0; i < fileds.Length; i++)
                {
                    _stateEntry.SetModifiedProperty(fileds[i]);
                }
            }
            else
            { // 部分字段排除
                PropertyInfo[] _properties = _type.GetProperties(); // 得到类的所有属性
                foreach (PropertyInfo item in _properties)
                {
                    if ("EntityState" == item.Name || "EntityKey" == item.Name)
                    {
                        continue;
                    }
                    // 主键判断 [EdmScalarPropertyAttribute(EntityKeyProperty=true, IsNullable=false)] 根据特性判断主键
                    EdmScalarPropertyAttribute _edmScalarPropertyAttribute = 
              Attribute.GetCustomAttribute(item, typeof(EdmScalarPropertyAttribute)) as EdmScalarPropertyAttribute;
                    if (null == _edmScalarPropertyAttribute || _edmScalarPropertyAttribute.EntityKeyProperty)
                    { // 为主键或者导航属性
                        continue;
                    }

                    bool _thisIsUpdateFiled = true;  // 是否为修改字段
                    for (int i = 0; i < fileds.Length; i++)
                    {
                        if (item.Name == fileds[i])
                        {
                            _thisIsUpdateFiled = false;
                            break;
                        }
                    }
                    if (_thisIsUpdateFiled)
                        _stateEntry.SetModifiedProperty(item.Name);
                }

            }
        }
        return db.SaveChanges();
    }

    /// <summary>
    /// 不查数据库修改信息
    /// </summary>
    /// <typeparam name="T"></typeparam>
    /// <param name="entity"></param>
    /// <param name="db"></param>
    /// <returns></returns>
    public static int Update<T>(T entity, DataDBEntities db)
    {
        if (null == db || null == entity)
        { // 参数有误
            return 0;
        }
        Type _type = typeof(T);
        db.AttachTo(_type.Name, entity);
        db.ObjectStateManager.ChangeObjectState(entity, System.Data.EntityState.Modified);      // 手动设置为修改状态
        return db.SaveChanges();
    }

    /// <summary>
    /// 不查数据库修改信息
    /// </summary>
    /// <typeparam name="T"></typeparam>
    /// <param name="entity"></param>
    /// <param name="updateFiledType"></param>
    /// <param name="fileds"></param>
    /// <returns></returns>
    public static int Update<T>(T entity, EnumUpdateFiledType updateFiledType, params string[] fileds)
    {
        if (null == entity)
        { // 参数有误
            return 0;
        }
        using (DataDBEntities db = new DataDBEntities())
        {
            Type _type = typeof(T);
            db.AttachTo(_type.Name, entity);
            if (null == fileds || fileds.Length == 0)
            { // 全字段操作
                db.ObjectStateManager.ChangeObjectState(entity, System.Data.EntityState.Modified);      // 手动设置为修改状态
            }
            else
            { // 部分字段操作
                var _stateEntry = db.ObjectStateManager.GetObjectStateEntry(entity);                    // 得到实体状态
                if (EnumUpdateFiledType.字段修改 == updateFiledType)
                { // 部分字段修改
                    for (int i = 0; i < fileds.Length; i++)
                    {
                        _stateEntry.SetModifiedProperty(fileds[i]);
                    }
                }
                else
                { // 部分字段排除
                    PropertyInfo[] _properties = _type.GetProperties(); // 得到类的所有属性

                    foreach (PropertyInfo item in _properties)
                    {
                        if ("EntityState" == item.Name || "EntityKey" == item.Name)
                        {
                            continue;
                        }
                        // 主键判断 [EdmScalarPropertyAttribute(EntityKeyProperty=true, IsNullable=false)] 根据特性判断主键
                        EdmScalarPropertyAttribute _edmScalarPropertyAttribute = 
                Attribute.GetCustomAttribute(item, typeof(EdmScalarPropertyAttribute)) as EdmScalarPropertyAttribute;
                        if (null == _edmScalarPropertyAttribute || _edmScalarPropertyAttribute.EntityKeyProperty)
                        { // 为主键或者导航属性
                            continue;
                        }
                        bool _thisIsUpdateFiled = true;  // 是否为修改字段
                        for (int i = 0; i < fileds.Length; i++)
                        {
                            if (item.Name == fileds[i])
                            {
                                _thisIsUpdateFiled = false;
                                break;
                            }
                        }
                        if (_thisIsUpdateFiled)
                            _stateEntry.SetModifiedProperty(item.Name);
                    }

                }
            }
            return db.SaveChanges();
        }
    }

    /// <summary>
    /// 不查数据库修改信息
    /// </summary>
    /// <typeparam name="T"></typeparam>
    /// <param name="entity"></param>
    /// <returns></returns>
    public static int Update<T>(T entity)
    {
        if (null == entity)
        { // 参数有误
            return 0;
        }
        using (DataDBEntities db = new DataDBEntities())
        {
            Type _type = typeof(T);
            db.AttachTo(_type.Name, entity);
            db.ObjectStateManager.ChangeObjectState(entity, System.Data.EntityState.Modified);      // 手动设置为修改状态
            return db.SaveChanges();
        }
    }

    #endregion

}

#region 修改时字段处理枚举
/// <summary>
/// 修改时字段处理枚举
/// </summary>
public enum EnumUpdateFiledType
{
    字段修改 = 1,
    字段忽略 = 2
}
#endregion
本文参与 腾讯云自媒体分享计划,分享自作者个人站点/博客。
原始发表:2012-11-01 ,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体分享计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
相关产品与服务
数据库
云数据库为企业提供了完善的关系型数据库、非关系型数据库、分析型数据库和数据库生态工具。您可以通过产品选择和组合搭建,轻松实现高可靠、高可用性、高性能等数据库需求。云数据库服务也可大幅减少您的运维工作量,更专注于业务发展,让企业一站式享受数据上云及分布式架构的技术红利!
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档