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

801 查看

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

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;
}