从山证v2.3.0拷贝

YLErpDAL/Modules/SystemModule/SysRoleService.cs

@@ -0,0 +1,158 @@
+using BaseOUDAL;
+using System.Text;
+using YLErp.Providers;
+
+namespace YLErp.Modules.SystemModule
+{
+    /// <summary>
+    /// 系统角色服务
+    /// </summary>
+    public class SysRoleService : BaseService<ErpBaseContext>
+    {
+        public SysRoleService(OptUserInfo optUser) : base(optUser)
+        {
+        }
+
+        public Role SaveRoleFunctions(RoleFunctionReq req)
+        {
+            if (DbContext.Roles.Any(role => role.Name == req.Name && role.Id != req.Id))
+            {
+                throw new ServiceException("已经存在相同的角色名称!");
+            }
+
+            Role dbRole;
+
+            if (req.Id > 0)
+            {
+                dbRole = DbContext.Roles.FirstOrDefault(o => o.Id == req.Id);
+
+                if (dbRole == null)
+                {
+                    throw new ServiceException("数据不存在!");
+                }
+            }
+            else
+            {
+                DbContext.Roles.Add(dbRole = new Role());
+            }
+
+            dbRole.Name = req.Name;
+            dbRole.Remark = req.Remark;
+            dbRole.OptDate = DateTime.Now;
+
+            //角色创建以后才能配置权限
+            if (req.Id > 0)
+            {
+                SetRoleFunc(req.Id, dbRole.Name, req.rights);
+            }
+
+            DbContext.SaveChanges();
+
+            MemoryCacheProvider.Default.FlushAll();
+
+            return dbRole;
+        }
+
+        private void SetRoleFunc(int roleId, string roleName, string rights)
+        {
+            var oldFuncIds = DbContext.RoleFunctions.Where(o => o.RoleId == roleId).Select(n => n.FunctionId).ToHashSet();
+
+            var newFuncIds = string.IsNullOrWhiteSpace(rights)
+                ? new HashSet<int>()
+                : rights.Split(',').Select(n => int.TryParse(n, out var i) ? i : 0)
+                    .Distinct().Where(n => n > 0 && !oldFuncIds.Remove(n)).ToHashSet();
+
+            if (oldFuncIds.Count < 1 && newFuncIds.Count < 1)
+            {
+                return;
+            }
+
+            var allFunc = oldFuncIds.Concat(newFuncIds).ToArray();
+
+            var funcDic = allFunc.Any()
+                ? DbContext.Functions.Where(n => allFunc.Contains(n.Id)).ToDictionary(n => n.Id, m => m.Title ?? m.Name)
+                : null;
+
+            //解除权限
+
+            var removeFuncList = new List<(int fid, string fname)>();
+
+            foreach (var fid in oldFuncIds)
+            {
+                var rf = DbContext.RoleFunctions.Attach(new RoleFunction { RoleId = roleId, FunctionId = fid }).Entity;
+
+                DbContext.RoleFunctions.Remove(rf);
+
+                var fname = funcDic != null && funcDic.TryGetValue(fid, out var str) ? str : null;
+
+                rf.SetDataTraceKeyInfo(roleName, fname);
+
+                if (fname != null)
+                {
+                    removeFuncList.Add((fid, fname));
+                }
+            }
+
+            if (removeFuncList.Count < 1 && newFuncIds.Count < 1)
+            {
+                return;
+            }
+
+            //增加权限
+
+            var sbLog = new StringBuilder(300).Append("角色:").Append(roleName).Append(';');
+
+            if (removeFuncList.Count > 0)
+            {
+                sbLog.Append("解除权限:");
+                foreach (var tuple in removeFuncList)
+                {
+                    sbLog.Append(tuple.fname);
+                }
+                sbLog.Replace(',', ';', sbLog.Length - 1, 1);
+            }
+
+            if (newFuncIds.Any())
+            {
+                sbLog.Append("增加权限:");
+
+                var newRoleFuncs = newFuncIds.Select(f =>
+                {
+                    var rf = new RoleFunction { RoleId = roleId, FunctionId = f };
+                    var fname = funcDic != null && funcDic.TryGetValue(f, out var str) ? str : f.ToString();
+                    rf.SetDataTraceKeyInfo(roleName, fname);
+                    sbLog.Append(fname).Append(',');
+                    return rf;
+                }).ToArray();
+
+                DbContext.RoleFunctions.AddRange(newRoleFuncs);
+
+                sbLog.Remove(sbLog.Length - 1, 1);
+            }
+
+            DbContext.SystemLogs.Add(new DBModels.SystemLog
+            {
+                EventCategory = "角色权限",
+                EventName = "设置权限",
+                EventData = JsonHelper.Serialize(new
+                {
+                    RoleId = roleId,
+                    NewFundIds = newFuncIds,
+                    RemoveFuncIds = removeFuncList.Select(n => n.fid).ToArray()
+                }),
+                Remark = sbLog.ToString()
+            }).Entity.SetOpt(UserInfo);
+        }
+    }
+
+    public class RoleFunctionReq
+    {
+        public int Id { get; set; }
+
+        public string Name { get; set; }
+
+        public string Remark { get; set; }
+
+        public string rights { get; set; }
+    }
+}