using System;
|
using System.Collections.Generic;
|
using System.Collections.ObjectModel;
|
using System.Data;
|
using System.Linq;
|
using System.Text;
|
using System.Threading.Tasks;
|
using System.Windows.Markup;
|
using DatabaseHelper;
|
using HMI.Core.Util;
|
using HMI.MenuManage.Entity;
|
|
namespace HMI.MenuManage.Util
|
{
|
public class RoleMenuHelper
|
{
|
/// <summary>
|
/// 根据roleCode获取rel数据
|
/// </summary>
|
/// <param name="sqliteHelper"></param>
|
/// <param name="roleCode"></param>
|
/// <returns></returns>
|
public static ObservableCollection<RoleMenuRel> SelectRoleMenuTreeData(SqliteHelper sqliteHelper,
|
string roleCode)
|
{
|
ObservableCollection<RoleMenuRel> treeData = new ObservableCollection<RoleMenuRel>();
|
Dictionary<string, RoleMenuRel> cache = new Dictionary<string, RoleMenuRel>();
|
DataTable dt = sqliteHelper.GetDataTable($@"SELECT m.*,t.roleCode,t.menuCode FROM menu AS m
|
LEFT JOIN (
|
SELECT rel.roleCode,rel.menuCode FROM role AS r
|
INNER JOIN role_menu_rel AS rel
|
ON r.code = rel.roleCode AND rel.roleCode='{roleCode}'
|
) AS t
|
ON m.code=t.menuCode
|
Order By m.level,m.id;");
|
if (dt != null)
|
{
|
foreach (DataRow dr in dt.Rows)
|
{
|
var menu = new RoleMenuRel
|
{
|
MenuCode = dr["code"].ToString(),
|
RoleCode = roleCode,
|
ParentCode = dr["parentCode"].ToString(),
|
MenuName = dr["name"].ToString(),
|
MenuIcon = dr["iconPath"].ToString(),
|
Selected = !(dr["menuCode"] is System.DBNull),
|
};
|
cache[menu.MenuCode] = menu;
|
if (cache.ContainsKey(menu.ParentCode))
|
{
|
cache[menu.ParentCode].AddChild(menu);
|
}
|
else
|
{
|
treeData.Add(menu);
|
}
|
}
|
}
|
return treeData;
|
}
|
|
public static void SaveRelData(SqliteHelper sqliteHelper, IEnumerable<RoleMenuRel> rels, string roleCode, string firstSql, string lastSql)
|
{
|
List<RoleMenuRel> datas = new List<RoleMenuRel>();
|
GetSelectedMenu(rels, datas);
|
List<string> sqls = new List<string>() {
|
firstSql, $"DELETE FROM role_menu_rel WHERE roleCode='{roleCode}'"
|
};
|
foreach (var data in datas)
|
{
|
sqls.Add($"INSERT INTO role_menu_rel (roleCode,menuCode) VALUES ('{roleCode}','{data.MenuCode}');");
|
}
|
sqls.Add(lastSql);
|
sqliteHelper.ExecuteBatchSql(sqls);
|
}
|
|
public static void GetSelectedMenu(IEnumerable<RoleMenuRel> rels, List<RoleMenuRel> datas)
|
{
|
foreach (var dr in rels)
|
{
|
if (dr.Selected)
|
{
|
datas.Add(dr);
|
}
|
GetSelectedMenu(dr.Children, datas);
|
}
|
}
|
|
}
|
}
|
