修改BUG:完善SQLite组件下Role授权方法,移除对存储过程和SqlBulkCopy的依赖

This commit is contained in:
Argo-MacBookPro 2018-10-23 14:41:23 +08:00
parent 152c3c8403
commit b1d7e5bf59
2 changed files with 67 additions and 77 deletions

View File

@ -3,7 +3,6 @@ using System;
using System.Collections.Generic; using System.Collections.Generic;
using System.Data; using System.Data;
using System.Data.Common; using System.Data.Common;
using System.Data.SqlClient;
using System.Linq; using System.Linq;
namespace Bootstrap.DataAccess.SQLite namespace Bootstrap.DataAccess.SQLite
@ -17,41 +16,30 @@ namespace Bootstrap.DataAccess.SQLite
/// <summary> /// <summary>
/// 保存用户角色关系 /// 保存用户角色关系
/// </summary> /// </summary>
/// <param name="id"></param> /// <param name="userId"></param>
/// <param name="roleIds"></param> /// <param name="roleIds"></param>
/// <returns></returns> /// <returns></returns>
public override bool SaveRolesByUserId(int id, IEnumerable<int> roleIds) public override bool SaveRolesByUserId(int userId, IEnumerable<int> roleIds)
{ {
var ret = false; var ret = false;
DataTable dt = new DataTable();
dt.Columns.Add("UserID", typeof(int));
dt.Columns.Add("RoleID", typeof(int));
//判断用户是否选定角色 //判断用户是否选定角色
roleIds.ToList().ForEach(roleId => dt.Rows.Add(id, roleId));
using (TransactionPackage transaction = DbAccessManager.DBAccess.BeginTransaction()) using (TransactionPackage transaction = DbAccessManager.DBAccess.BeginTransaction())
{ {
try try
{ {
// delete user from config table // delete user from config table
string sql = "delete from UserRole where UserID = @UserID;"; string sql = $"delete from UserRole where UserID = {userId}";
using (DbCommand cmd = DbAccessManager.DBAccess.CreateCommand(CommandType.Text, sql)) using (DbCommand cmd = DbAccessManager.DBAccess.CreateCommand(CommandType.Text, sql))
{ {
cmd.Parameters.Add(DbAccessManager.DBAccess.CreateParameter("@UserID", id));
DbAccessManager.DBAccess.ExecuteNonQuery(cmd, transaction); DbAccessManager.DBAccess.ExecuteNonQuery(cmd, transaction);
if (dt.Rows.Count > 0) roleIds.ToList().ForEach(rId =>
{ {
// insert batch data into config table cmd.CommandText = $"Insert Into UserRole (UserID, RoleID) Values ( {userId}, {rId})";
using (SqlBulkCopy bulk = new SqlBulkCopy((SqlConnection)transaction.Transaction.Connection, SqlBulkCopyOptions.Default, (SqlTransaction)transaction.Transaction)) DbAccessManager.DBAccess.ExecuteNonQuery(cmd, transaction);
{ });
bulk.DestinationTableName = "UserRole";
bulk.ColumnMappings.Add("UserID", "UserID");
bulk.ColumnMappings.Add("RoleID", "RoleID");
bulk.WriteToServer(dt);
}
}
transaction.CommitTransaction(); transaction.CommitTransaction();
} }
CacheCleanUtility.ClearCache(userIds: new List<int>() { id }, roleIds: roleIds); CacheCleanUtility.ClearCache(userIds: new List<int>() { userId }, roleIds: roleIds);
ret = true; ret = true;
} }
catch (Exception ex) catch (Exception ex)
@ -70,51 +58,64 @@ namespace Bootstrap.DataAccess.SQLite
{ {
bool ret = false; bool ret = false;
var ids = string.Join(",", value); var ids = string.Join(",", value);
using (DbCommand cmd = DbAccessManager.DBAccess.CreateCommand(CommandType.StoredProcedure, "Proc_DeleteRoles")) using (TransactionPackage transaction = DbAccessManager.DBAccess.BeginTransaction())
{ {
cmd.Parameters.Add(DbAccessManager.DBAccess.CreateParameter("@ids", ids)); using (DbCommand cmd = DbAccessManager.DBAccess.CreateCommand(CommandType.Text, $"delete from UserRole where RoleID in ({ids})"))
ret = DbAccessManager.DBAccess.ExecuteNonQuery(cmd) == -1; {
} try
{
DbAccessManager.DBAccess.ExecuteNonQuery(cmd, transaction);
cmd.CommandText = $"delete from RoleGroup where RoleID in ({ids})";
DbAccessManager.DBAccess.ExecuteNonQuery(cmd, transaction);
cmd.CommandText = $"delete from NavigationRole where RoleID in ({ids})";
DbAccessManager.DBAccess.ExecuteNonQuery(cmd, transaction);
cmd.CommandText = $"delete from Roles where ID in ({ids})";
DbAccessManager.DBAccess.ExecuteNonQuery(cmd, transaction);
transaction.CommitTransaction();
CacheCleanUtility.ClearCache(roleIds: value); CacheCleanUtility.ClearCache(roleIds: value);
ret = true;
}
catch (Exception ex)
{
transaction.RollbackTransaction();
throw ex;
}
}
}
return ret; return ret;
} }
/// <summary> /// <summary>
/// ///
/// </summary> /// </summary>
/// <param name="id"></param> /// <param name="menuId"></param>
/// <param name="roleIds"></param> /// <param name="roleIds"></param>
/// <returns></returns> /// <returns></returns>
public override bool SavaRolesByMenuId(int id, IEnumerable<int> roleIds) public override bool SavaRolesByMenuId(int menuId, IEnumerable<int> roleIds)
{ {
var ret = false; var ret = false;
DataTable dt = new DataTable();
dt.Columns.Add("NavigationID", typeof(int));
dt.Columns.Add("RoleID", typeof(int));
//判断用户是否选定角色 //判断用户是否选定角色
roleIds.ToList().ForEach(roleId => dt.Rows.Add(id, roleId));
using (TransactionPackage transaction = DbAccessManager.DBAccess.BeginTransaction()) using (TransactionPackage transaction = DbAccessManager.DBAccess.BeginTransaction())
{ {
try try
{ {
// delete role from config table // delete role from config table
string sql = "delete from NavigationRole where NavigationID=@NavigationID;"; string sql = $"delete from NavigationRole where NavigationID = {menuId}";
using (DbCommand cmd = DbAccessManager.DBAccess.CreateCommand(CommandType.Text, sql)) using (DbCommand cmd = DbAccessManager.DBAccess.CreateCommand(CommandType.Text, sql))
{ {
cmd.Parameters.Add(DbAccessManager.DBAccess.CreateParameter("@NavigationID", id));
DbAccessManager.DBAccess.ExecuteNonQuery(cmd, transaction); DbAccessManager.DBAccess.ExecuteNonQuery(cmd, transaction);
// insert batch data into config table // insert batch data into config table
using (SqlBulkCopy bulk = new SqlBulkCopy((SqlConnection)transaction.Transaction.Connection, SqlBulkCopyOptions.Default, (SqlTransaction)transaction.Transaction)) roleIds.ToList().ForEach(rId =>
{ {
bulk.BatchSize = 1000; cmd.CommandText = $"Insert Into NavigationRole (NavigationID, RoleID) Values ( {menuId}, {rId})";
bulk.DestinationTableName = "NavigationRole"; DbAccessManager.DBAccess.ExecuteNonQuery(cmd, transaction);
bulk.ColumnMappings.Add("NavigationID", "NavigationID"); });
bulk.ColumnMappings.Add("RoleID", "RoleID");
bulk.WriteToServer(dt);
transaction.CommitTransaction(); transaction.CommitTransaction();
} }
} CacheCleanUtility.ClearCache(roleIds: roleIds, menuIds: new List<int>() { menuId });
CacheCleanUtility.ClearCache(roleIds: roleIds, menuIds: new List<int>() { id });
ret = true; ret = true;
} }
catch (Exception ex) catch (Exception ex)
@ -128,40 +129,32 @@ namespace Bootstrap.DataAccess.SQLite
/// <summary> /// <summary>
/// 根据GroupId更新Roles信息删除旧的Roles信息插入新的Roles信息 /// 根据GroupId更新Roles信息删除旧的Roles信息插入新的Roles信息
/// </summary> /// </summary>
/// <param name="id"></param> /// <param name="groupId"></param>
/// <param name="roleIds"></param> /// <param name="roleIds"></param>
/// <returns></returns> /// <returns></returns>
public override bool SaveRolesByGroupId(int id, IEnumerable<int> roleIds) public override bool SaveRolesByGroupId(int groupId, IEnumerable<int> roleIds)
{ {
var ret = false; var ret = false;
//构造表格 //构造表格
DataTable dt = new DataTable();
dt.Columns.Add("RoleID", typeof(int));
dt.Columns.Add("GroupID", typeof(int));
roleIds.ToList().ForEach(roleId => dt.Rows.Add(roleId, id));
using (TransactionPackage transaction = DbAccessManager.DBAccess.BeginTransaction()) using (TransactionPackage transaction = DbAccessManager.DBAccess.BeginTransaction())
{ {
try try
{ {
// delete user from config table // delete user from config table
string sql = "delete from RoleGroup where GroupID=@GroupID"; string sql = $"delete from RoleGroup where GroupID = {groupId}";
using (DbCommand cmd = DbAccessManager.DBAccess.CreateCommand(CommandType.Text, sql)) using (DbCommand cmd = DbAccessManager.DBAccess.CreateCommand(CommandType.Text, sql))
{ {
cmd.Parameters.Add(DbAccessManager.DBAccess.CreateParameter("@GroupID", id));
DbAccessManager.DBAccess.ExecuteNonQuery(cmd, transaction); DbAccessManager.DBAccess.ExecuteNonQuery(cmd, transaction);
// insert batch data into config table // insert batch data into config table
using (SqlBulkCopy bulk = new SqlBulkCopy((SqlConnection)transaction.Transaction.Connection, SqlBulkCopyOptions.Default, (SqlTransaction)transaction.Transaction)) roleIds.ToList().ForEach(rId =>
{ {
bulk.BatchSize = 1000; cmd.CommandText = $"Insert Into RoleGroup (GroupID, RoleID) Values ( {groupId}, {rId})";
bulk.DestinationTableName = "RoleGroup"; DbAccessManager.DBAccess.ExecuteNonQuery(cmd, transaction);
bulk.ColumnMappings.Add("RoleID", "RoleID"); });
bulk.ColumnMappings.Add("GroupID", "GroupID");
bulk.WriteToServer(dt);
transaction.CommitTransaction(); transaction.CommitTransaction();
} }
} CacheCleanUtility.ClearCache(roleIds: roleIds, groupIds: new List<int>() { groupId });
CacheCleanUtility.ClearCache(roleIds: roleIds, groupIds: new List<int>() { id });
ret = true; ret = true;
} }
catch (Exception ex) catch (Exception ex)

View File

@ -69,26 +69,25 @@ namespace Bootstrap.DataAccess
/// <summary> /// <summary>
/// 保存用户角色关系 /// 保存用户角色关系
/// </summary> /// </summary>
/// <param name="id"></param> /// <param name="userId"></param>
/// <param name="roleIds"></param> /// <param name="roleIds"></param>
/// <returns></returns> /// <returns></returns>
public virtual bool SaveRolesByUserId(int id, IEnumerable<int> roleIds) public virtual bool SaveRolesByUserId(int userId, IEnumerable<int> roleIds)
{ {
var ret = false; var ret = false;
DataTable dt = new DataTable(); DataTable dt = new DataTable();
dt.Columns.Add("UserID", typeof(int)); dt.Columns.Add("UserID", typeof(int));
dt.Columns.Add("RoleID", typeof(int)); dt.Columns.Add("RoleID", typeof(int));
//判断用户是否选定角色 //判断用户是否选定角色
roleIds.ToList().ForEach(roleId => dt.Rows.Add(id, roleId)); roleIds.ToList().ForEach(roleId => dt.Rows.Add(userId, roleId));
using (TransactionPackage transaction = DbAccessManager.DBAccess.BeginTransaction()) using (TransactionPackage transaction = DbAccessManager.DBAccess.BeginTransaction())
{ {
try try
{ {
// delete user from config table // delete user from config table
string sql = "delete from UserRole where UserID = @UserID;"; string sql = $"delete from UserRole where UserID = {userId}";
using (DbCommand cmd = DbAccessManager.DBAccess.CreateCommand(CommandType.Text, sql)) using (DbCommand cmd = DbAccessManager.DBAccess.CreateCommand(CommandType.Text, sql))
{ {
cmd.Parameters.Add(DbAccessManager.DBAccess.CreateParameter("@UserID", id));
DbAccessManager.DBAccess.ExecuteNonQuery(cmd, transaction); DbAccessManager.DBAccess.ExecuteNonQuery(cmd, transaction);
if (dt.Rows.Count > 0) if (dt.Rows.Count > 0)
{ {
@ -103,7 +102,7 @@ namespace Bootstrap.DataAccess
} }
transaction.CommitTransaction(); transaction.CommitTransaction();
} }
CacheCleanUtility.ClearCache(userIds: new List<int>() { id }, roleIds: roleIds); CacheCleanUtility.ClearCache(userIds: new List<int>() { userId }, roleIds: roleIds);
ret = true; ret = true;
} }
catch (Exception ex) catch (Exception ex)
@ -216,26 +215,25 @@ namespace Bootstrap.DataAccess
/// <summary> /// <summary>
/// ///
/// </summary> /// </summary>
/// <param name="id"></param> /// <param name="menuId"></param>
/// <param name="roleIds"></param> /// <param name="roleIds"></param>
/// <returns></returns> /// <returns></returns>
public virtual bool SavaRolesByMenuId(int id, IEnumerable<int> roleIds) public virtual bool SavaRolesByMenuId(int menuId, IEnumerable<int> roleIds)
{ {
var ret = false; var ret = false;
DataTable dt = new DataTable(); DataTable dt = new DataTable();
dt.Columns.Add("NavigationID", typeof(int)); dt.Columns.Add("NavigationID", typeof(int));
dt.Columns.Add("RoleID", typeof(int)); dt.Columns.Add("RoleID", typeof(int));
//判断用户是否选定角色 //判断用户是否选定角色
roleIds.ToList().ForEach(roleId => dt.Rows.Add(id, roleId)); roleIds.ToList().ForEach(roleId => dt.Rows.Add(menuId, roleId));
using (TransactionPackage transaction = DbAccessManager.DBAccess.BeginTransaction()) using (TransactionPackage transaction = DbAccessManager.DBAccess.BeginTransaction())
{ {
try try
{ {
// delete role from config table // delete role from config table
string sql = "delete from NavigationRole where NavigationID=@NavigationID;"; string sql = $"delete from NavigationRole where NavigationID = {menuId}";
using (DbCommand cmd = DbAccessManager.DBAccess.CreateCommand(CommandType.Text, sql)) using (DbCommand cmd = DbAccessManager.DBAccess.CreateCommand(CommandType.Text, sql))
{ {
cmd.Parameters.Add(DbAccessManager.DBAccess.CreateParameter("@NavigationID", id));
DbAccessManager.DBAccess.ExecuteNonQuery(cmd, transaction); DbAccessManager.DBAccess.ExecuteNonQuery(cmd, transaction);
// insert batch data into config table // insert batch data into config table
@ -249,7 +247,7 @@ namespace Bootstrap.DataAccess
transaction.CommitTransaction(); transaction.CommitTransaction();
} }
} }
CacheCleanUtility.ClearCache(roleIds: roleIds, menuIds: new List<int>() { id }); CacheCleanUtility.ClearCache(roleIds: roleIds, menuIds: new List<int>() { menuId });
ret = true; ret = true;
} }
catch (Exception ex) catch (Exception ex)
@ -294,26 +292,25 @@ namespace Bootstrap.DataAccess
/// <summary> /// <summary>
/// 根据GroupId更新Roles信息删除旧的Roles信息插入新的Roles信息 /// 根据GroupId更新Roles信息删除旧的Roles信息插入新的Roles信息
/// </summary> /// </summary>
/// <param name="id"></param> /// <param name="groupId"></param>
/// <param name="roleIds"></param> /// <param name="roleIds"></param>
/// <returns></returns> /// <returns></returns>
public virtual bool SaveRolesByGroupId(int id, IEnumerable<int> roleIds) public virtual bool SaveRolesByGroupId(int groupId, IEnumerable<int> roleIds)
{ {
var ret = false; var ret = false;
//构造表格 //构造表格
DataTable dt = new DataTable(); DataTable dt = new DataTable();
dt.Columns.Add("RoleID", typeof(int)); dt.Columns.Add("RoleID", typeof(int));
dt.Columns.Add("GroupID", typeof(int)); dt.Columns.Add("GroupID", typeof(int));
roleIds.ToList().ForEach(roleId => dt.Rows.Add(roleId, id)); roleIds.ToList().ForEach(roleId => dt.Rows.Add(roleId, groupId));
using (TransactionPackage transaction = DbAccessManager.DBAccess.BeginTransaction()) using (TransactionPackage transaction = DbAccessManager.DBAccess.BeginTransaction())
{ {
try try
{ {
// delete user from config table // delete user from config table
string sql = "delete from RoleGroup where GroupID=@GroupID"; string sql = $"delete from RoleGroup where GroupID = {groupId}";
using (DbCommand cmd = DbAccessManager.DBAccess.CreateCommand(CommandType.Text, sql)) using (DbCommand cmd = DbAccessManager.DBAccess.CreateCommand(CommandType.Text, sql))
{ {
cmd.Parameters.Add(DbAccessManager.DBAccess.CreateParameter("@GroupID", id));
DbAccessManager.DBAccess.ExecuteNonQuery(cmd, transaction); DbAccessManager.DBAccess.ExecuteNonQuery(cmd, transaction);
// insert batch data into config table // insert batch data into config table
@ -327,7 +324,7 @@ namespace Bootstrap.DataAccess
transaction.CommitTransaction(); transaction.CommitTransaction();
} }
} }
CacheCleanUtility.ClearCache(roleIds: roleIds, groupIds: new List<int>() { id }); CacheCleanUtility.ClearCache(roleIds: roleIds, groupIds: new List<int>() { groupId });
ret = true; ret = true;
} }
catch (Exception ex) catch (Exception ex)