Entity Framework 实体框架的形成之旅--几种数据库操作的代码介绍（9）

本篇主要对常规数据操作的处理和实体框架的处理代码进行对比，以便更容易学习理解实体框架里面，对各种数据库处理技巧，本篇介绍几种数据库操作的代码，包括写入中间表操作、联合中间表获取对象集合、递归操作、设置单一字段的修改等几种方式。

1、写入中间表操作

一般情况下，我们可以通过执行数据库脚本方式写入。

        /// <summary>
        /// 增加用户IP信息
        /// </summary>
        /// <param name="userID"></param>
        /// <param name="blackID"></param>
        public void AddUser(int userID, string blackID)
        {
            string commandText = string.Format("INSERT INTO T_ACL_BLACKIP_USER(User_ID, BLACKIP_ID) VALUES({0}, '{1}')", userID, blackID);
            base.SqlExecute(commandText);
        }

        /// <summary>
        /// 移除用户IP 信息
        /// </summary>
        /// <param name="userID"></param>
        /// <param name="blackID"></param>
        public void RemoveUser(int userID, string blackID)
        {
            string commandText = string.Format("DELETE FROM T_ACL_BLACKIP_USER WHERE User_ID={0} AND BLACKIP_ID='{1}'", userID, blackID);
            base.SqlExecute(commandText);
        }

        /// <summary>
        /// 执行一些特殊的语句
        /// </summary>
        /// <param name="sql">SQL语句</param>
        /// <param name="parameters">要应用于命令字符串的参数</param>
        public virtual int SqlExecute(string sql, params object[] parameters)
        {
            return baseContext.Database.ExecuteSqlCommand(sql, parameters);
        }

上面间接用来执行脚本方式写入中间表数据；但是，如果我们采用Entity Framework的方式来处理，也是可以的，例如我们为角色添加和移除用户的代码如下所示。

        public void AddUser(int userID, int roleID)
        {
            var role = new Role() { ID = roleID, Functions = new List<Function>() };
            context.Roles.Attach(role);

            User user = new User() { ID = userID };
            context.Users.Attach(user);

            int count = this.GetQueryable().Where(s => s.ID == roleID).Count(s => s.Users.Any(p => p.ID == userID));
            if (count == 0)
            {
                role.Users.Add(user);
            }

            context.SaveChanges();
        }

        public void RemoveUser(int userID, int roleID)
        {
            var role = this.GetQueryable().FirstOrDefault(s => s.ID == roleID);
            var deleted = role.Users.FirstOrDefault(s => s.ID == userID);
            if (deleted != null)
            {
                role.Users.Remove(deleted);
            }

            context.SaveChanges();
        }

如果我们需要重新写入中间列表的数据，那么处理方式如下所示。

        public bool EditOuUsers(int ouID, List<int> newUserList)
        {
            //先清空
            Ou ou = this.GetQueryable().FirstOrDefault(s => s.ID == ouID);
            ou.Users.Clear();
            context.SaveChanges();

            //设置新的列表
            var users = context.Users.Where(s => newUserList.Contains(s.ID)).ToList();
            ou.Users = users;

            return context.SaveChanges() > 0;
        }

2、联合中间表获取对象集合

一般情况下，我们在数据访问层需要结合中间表进行查询的时候，常用的处理方式是在SQL语句里面联合两个表进行处理。如联合机构角色中间表、机构用户中间表来获取对应机构的列表操作如下所示。

        public List<OUInfo> GetOUsByRole(int roleID)
        {
            string sql = "SELECT * FROM T_ACL_OU INNER JOIN T_ACL_OU_Role On [T_ACL_OU].ID=T_ACL_OU_Role.OU_ID WHERE Role_ID = " + roleID;
            return this.GetList(sql, null);
        }

        public List<OUInfo> GetOUsByUser(int userID)
        {
            string sql = "SELECT * FROM T_ACL_OU INNER JOIN T_ACL_OU_User On [T_ACL_OU].ID=T_ACL_OU_User.OU_ID WHERE User_ID = " + userID;
            return this.GetList(sql, null);
        }

如果是在实体框架里面，我们不能那么直接操作SQL语句，尽可能用内置的LINQ查询操作，使用的时候，你也发现很简单的。

        public IList<Ou> GetOUsByRole(int roleID)
        {
            return this.GetQueryable().Where(s => s.Roles.Any(c => c.ID == roleID)).ToList();
        }

        public IList<Ou> GetOUsByUser(int userID)
        {
            return this.GetQueryable().Where(s => s.Users.Any(c => c.ID == userID)).ToList();
        }

3、递归操作

很多时候，我们表的数据是有层次结构的，有时候，需要根据父ID获取下面的子集列表的时候，我们就需要用到递归。常规的SQL处理方式，可以如下实现。

       /// <summary>
        /// 根据指定机构节点ID，获取其下面所有机构列表
        /// </summary>
        /// <param name="parentId">指定机构节点ID</param>
        /// <returns></returns>
        public List<OUInfo> GetAllOUsByParent(int parentId)
        {
            List<OUInfo> list = new List<OUInfo>();
            string sql = string.Format("Select * From {0} Where Deleted <> 1 Order By PID, Name ", tableName);

            DataTable dt = SqlTable(sql);
            string sort = string.Format("{0} {1}", GetSafeFileName(sortField), isDescending ? "DESC" : "ASC");
            DataRow[] dataRows = dt.Select(string.Format(" PID = {0}", parentId), sort);
            for (int i = 0; i < dataRows.Length; i++)
            {
                string id = dataRows[i]["ID"].ToString();
                list.AddRange(GetOU(id, dt));
            }

            return list;
        }

        private List<OUInfo> GetOU(string id, DataTable dt)
        {
            List<OUInfo> list = new List<OUInfo>();

            OUInfo ouInfo = this.FindByID(id);
            list.Add(ouInfo);

            string sort = string.Format("{0} {1}", GetSafeFileName(sortField), isDescending ? "DESC" : "ASC");
            DataRow[] dChildRows = dt.Select(string.Format(" PID={0} ", id), sort);
            for (int i = 0; i < dChildRows.Length; i++)
            {
                string childId = dChildRows[i]["ID"].ToString();
                List<OUInfo> childList = GetOU(childId, dt);
                list.AddRange(childList);
            }
            return list;
        }

如果这样的实现，在实体框架实现，那么应该如何处理呢？

        public IList<Ou> GetAllOUsByParent(int parentId)
        {
            List<Ou> list = this.GetQueryable().Where(s => s.PID == parentId).Where(s => !s.IsDeleted).OrderBy(s => s.PID).OrderBy(s => s.Name).ToList();
            foreach(Ou info in list)
            {
                list.AddRange(GetAllOUsByParent(info.ID));
            }
            return list;
        }

是不是觉得简单很多了呢？

4、设置单一字段的修改

有时候，我们可能需要更新列表里面的某些字段，那么如果采用SQL操作处理，那么代码如下所示。

        /// <summary>
        /// 设置删除标志
        /// </summary>
        /// <param name="id">记录ID</param>
        /// <param name="deleted">是否删除</param>
        /// <param name="trans">事务对象</param>
        /// <returns></returns>
        public bool SetDeletedFlag(object id, bool deleted = true, DbTransaction trans = null)
        {
            int intDeleted = deleted ? 1 : 0;
            string sql = string.Format("Update {0} Set Deleted={1} Where ID = {2} ", tableName, intDeleted, id);
            return SqlExecute(sql, trans) > 0;
        }

如果是采用实体框架，那么处理代码如下所示。

        public bool SetDeletedFlag(object id, bool deleted = true)
        {
            var ou = new Ou() { ID = id.ToString().ToInt32(), IsDeleted = deleted };
            context.Ous.Attach(ou);

            context.Entry(ou).Property(x => x.Deleted).IsModified = true;
            return context.SaveChanges() > 0;
        }

如果更新多个字段，那么常规的SQL方式代码如下所示。

        /// <summary>
        /// 更新用户登录的时间和IP地址
        /// </summary>
        /// <param name="id">用户ID</param>
        /// <param name="ip">IP地址</param>
        /// <param name="macAddr">MAC地址</param>
        /// <returns></returns>
        public bool UpdateUserLoginData(int id, string ip, string macAddr)
        {
            //先复制最后的登录时间和IP地址
            string sql = string.Format("Update {0} set LastLoginIP=CurrentLoginIP,LastLoginTime=CurrentLoginTime,LastMacAddress=CurrentMacAddress Where ID={1}", tableName, id);
            Database db = CreateDatabase();
            DbCommand dbCommand = db.GetSqlStringCommand(sql);
            db.ExecuteNonQuery(dbCommand);

            sql = string.Format("Update {0} Set CurrentLoginIP='{1}',CurrentMacAddress='{2}', CurrentLoginTime=@CurrentLoginTime Where ID = {3}", tableName, ip, macAddr, id);
            dbCommand = db.GetSqlStringCommand(sql);
            db.AddInParameter(dbCommand, "CurrentLoginTime", DbType.DateTime, DateTime.Now);
            return db.ExecuteNonQuery(dbCommand) > 0;
        }

多个字段的更新，使用实体框架则是如下所示。

        public bool UpdateUserLoginData(int id, string ip, string macAddr)
        {
            //先复制最后的登录时间和IP地址
            string tableName = context.GetTableName<User>();
            string sql = string.Format("Update {0} set LastLoginIP=CurrentLoginIP,LastLoginTime=CurrentLoginTime,LastMacAddress=CurrentMacAddress Where ID={1}", tableName, id);
            SqlExecute(sql);

            var obj = new User() { ID = id.ToString().ToInt32(), CurrentLoginIP = ip, CurrentMacAddress = macAddr, CurrentLoginTime = DateTime.Now };
            context.Users.Attach(obj);
            context.Entry(obj).Property(x => x.CurrentLoginIP).IsModified = true;
            context.Entry(obj).Property(x => x.CurrentMacAddress).IsModified = true;
            context.Entry(obj).Property(x => x.CurrentLoginTime).IsModified = true;
            return context.SaveChanges() > 0;
        }