本篇主要对常规数据操作的处理和实体框架的处理代码进行对比，以便更容易学习理解实体框架里面，对各种数据库处理技巧，本篇介绍几种数据库操作的代码，包括写入中间表操作、联合中间表获取对象集合、递归操作、设置单一字段的修改等几种方式。
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; }
这个系列文章如下所示：
Entity Framework 实体框架的形成之旅--基于泛型的仓储模式的实体框架(1)
Entity Framework 实体框架的形成之旅--利用Unity对象依赖注入优化实体框架(2)
Entity Framework 实体框架的形成之旅--基类接口的统一和异步操作的实现(3)
Entity Framework 实体框架的形成之旅--实体数据模型 (EDM)的处理（4）
Entity Framework 实体框架的形成之旅--Code First的框架设计（5）
Entity Framework 实体框架的形成之旅--Code First模式中使用 Fluent API 配置（6）
Entity Framework 实体框架的形成之旅--数据传输模型DTO和实体模型Entity的分离与联合