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