本篇主要對常規數據操作的處理和實體框架的處理代碼進行對比，以便更容易學習理解實體框架里面，對各種數據庫處理技巧，本篇介紹幾種數據庫操作的代碼，包括寫入中間表操作、聯合中間表獲取對象集合、遞歸操作、設置單一字段的修改等幾種方式。
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的分離與聯合