186 lines
7.9 KiB
C#
186 lines
7.9 KiB
C#
using Infrastructure;
|
||
using Infrastructure.Helper;
|
||
using SqlSugar;
|
||
using SqlSugar.IOC;
|
||
using System.Linq;
|
||
using System.Reflection;
|
||
using ARW.Admin.WebApi.Framework;
|
||
using ARW.Model.System;
|
||
using ARW.Model.Models.Business;
|
||
using Infrastructure.Extensions;
|
||
using System.Linq.Dynamic.Core;
|
||
using System.Linq.Expressions;
|
||
|
||
namespace ARW.Admin.WebApi.Extensions
|
||
{
|
||
public static class DbExtension
|
||
{
|
||
private static NLog.Logger logger = NLog.LogManager.GetCurrentClassLogger();
|
||
//全部数据权限
|
||
public static string DATA_SCOPE_ALL = "1";
|
||
//自定数据权限
|
||
public static string DATA_SCOPE_CUSTOM = "2";
|
||
//部门数据权限
|
||
public static string DATA_SCOPE_DEPT = "3";
|
||
//部门及以下数据权限
|
||
public static string DATA_SCOPE_DEPT_AND_CHILD = "4";
|
||
//仅本人数据权限
|
||
public static string DATA_SCOPE_SELF = "5";
|
||
|
||
|
||
|
||
|
||
public static void AddDb(IConfiguration Configuration)
|
||
{
|
||
string connStr = Configuration.GetConnectionString("conn_db");
|
||
int dbType = Convert.ToInt32(Configuration.GetConnectionString("conn_db_type"));
|
||
|
||
SugarIocServices.AddSqlSugar(new List<IocConfig>() {
|
||
new IocConfig() {
|
||
ConfigId = "0",//默认db
|
||
ConnectionString = connStr,
|
||
DbType = (IocDbType)dbType,
|
||
IsAutoCloseConnection = true
|
||
},
|
||
new IocConfig() {
|
||
ConfigId = "1",
|
||
ConnectionString = "替换成你的字符串",
|
||
DbType = IocDbType.MySql,
|
||
IsAutoCloseConnection = true
|
||
}
|
||
//...增加其他数据库
|
||
});
|
||
SugarIocServices.ConfigurationSugar(db =>
|
||
{
|
||
//db0数据过滤
|
||
FilterData(0);
|
||
|
||
#region db0
|
||
db.GetConnectionScope(0).Aop.OnLogExecuting = (sql, pars) =>
|
||
{
|
||
var param = db.GetConnectionScope(0).Utilities.SerializeObject(pars.ToDictionary(it => it.ParameterName, it => it.Value));
|
||
|
||
logger.Info($"【sql语句】{sql}，{param}\n");
|
||
};
|
||
|
||
db.GetConnectionScope(0).Aop.OnError = (e) =>
|
||
{
|
||
logger.Error(e, $"执行SQL出错：{e.Message}");
|
||
};
|
||
//SQL执行完
|
||
db.GetConnectionScope(0).Aop.OnLogExecuted = (sql, pars) =>
|
||
{
|
||
//执行完了可以输出SQL执行时间 (OnLogExecutedDelegate)
|
||
};
|
||
#endregion
|
||
|
||
#region db1
|
||
//Db1
|
||
db.GetConnection(1).Aop.OnLogExecuting = (sql, pars) =>
|
||
{
|
||
var param = db.GetConnection(1).Utilities.SerializeObject(pars.ToDictionary(it => it.ParameterName, it => it.Value));
|
||
|
||
logger.Info($"【sql语句】{sql}, {param}");
|
||
};
|
||
//Db1错误日志
|
||
db.GetConnection(1).Aop.OnError = (e) =>
|
||
{
|
||
logger.Error($"执行Sql语句失败：{e.Sql}，原因：{e.Message}");
|
||
};
|
||
#endregion
|
||
});
|
||
|
||
}
|
||
|
||
public static void InitDb(this IServiceProvider service)
|
||
{
|
||
var db = DbScoped.SugarScope;
|
||
new MySql.Data.MySqlClient.MySqlConnection("server=47.242.159.172;Database=demo;Uid=demo;Pwd=demo;SslMode=none;CharSet=utf8mb4;AllowLoadLocalInfile=true;AllowUserVariables=true;").Open();
|
||
//db.DbMaintenance.CreateDatabase();
|
||
//db.CodeFirst.
|
||
|
||
var baseType = typeof(SysBase);
|
||
//var entityes = AssemblyUtils.GetAllTypes().Where(p => !p.IsAbstract && p != baseType && /*!p.IsAssignableTo(baseType)&&*/ p.GetCustomAttribute<SugarTable>()!=null).ToArray();
|
||
|
||
Type[] types = Assembly
|
||
.LoadFrom("bin/Debug/net6.0/ARW.Model.dll")//如果 .dll报错，可以换成 xxx.exe 有些生成的是exe
|
||
.GetTypes().Where(it => it.FullName.Contains("Business.") && it.GetCustomAttribute<SugarTable>() != null && it != typeof(BusinessBase))//命名空间过滤，当然你也可以写其他条件过滤
|
||
.ToArray();
|
||
|
||
db.CodeFirst.SetStringDefaultLength(512).InitTables(types);
|
||
}
|
||
|
||
/// <summary>
|
||
/// 数据过滤
|
||
/// </summary>
|
||
/// <param name="configId">多库id</param>
|
||
private static void FilterData(int configId)
|
||
{
|
||
|
||
var db = DbScoped.SugarScope.GetConnectionScope(configId);
|
||
|
||
// 软删除过滤
|
||
Type[] types = Assembly
|
||
.LoadFrom("bin/Debug/net6.0/ARW.Model.dll")//如果 .dll报错，可以换成 xxx.exe 有些生成的是exe
|
||
.GetTypes().Where(it => it.FullName.Contains("Business.") && it.GetCustomAttribute<SugarTable>() != null && it != typeof(BusinessBase))//命名空间过滤，当然你也可以写其他条件过滤
|
||
.ToArray();
|
||
foreach (var entityType in types)
|
||
{
|
||
//构建动态Lambda
|
||
var lambda = DynamicExpressionParser.ParseLambda
|
||
(new[] { Expression.Parameter(entityType, "it") },
|
||
typeof(bool), $"{nameof(BusinessBase.IsDelete)} == @0",
|
||
false);
|
||
db.QueryFilter.Add(new TableFilterItem<object>(entityType, lambda)); //将Lambda传入过滤器
|
||
}
|
||
|
||
var u = App.User;
|
||
if (u == null) return;
|
||
//获取当前用户的信息
|
||
var user = JwtUtil.GetLoginUser(App.HttpContext);
|
||
if (user == null) return;
|
||
|
||
if (user.IsApi) return;
|
||
//管理员不过滤
|
||
if (user.RoleIds.Any(f => f.Equals("admin"))) return;
|
||
foreach (var role in user.Roles.OrderBy(f => f.DataScope))
|
||
{
|
||
string dataScope = role.DataScope;
|
||
if (DATA_SCOPE_ALL.Equals(dataScope))//所有权限
|
||
{
|
||
break;
|
||
}
|
||
else if (DATA_SCOPE_CUSTOM.Equals(dataScope))//自定数据权限
|
||
{
|
||
//" OR {}.dept_id IN ( SELECT dept_id FROM sys_role_dept WHERE role_id = {} ) ", deptAlias, role.getRoleId()));
|
||
var filter1 = new TableFilterItem<SysUser>(it => SqlFunc.Subqueryable<SysRoleDept>().Where(f => f.DeptId == it.DeptId && f.RoleId == role.RoleId).Any());
|
||
db.QueryFilter.Add(filter1);
|
||
}
|
||
else if (DATA_SCOPE_DEPT.Equals(dataScope))//本部门数据
|
||
{
|
||
var filter1 = new TableFilterItem<SysUser>(it => it.DeptId == user.DeptId);
|
||
db.QueryFilter.Add(filter1);
|
||
}
|
||
else if (DATA_SCOPE_DEPT_AND_CHILD.Equals(dataScope))//本部门及以下数据
|
||
{
|
||
//SQl OR {}.dept_id IN ( SELECT dept_id FROM sys_dept WHERE dept_id = {} or find_in_set( {} , ancestors ) )
|
||
var allChildDepts = db.Queryable<SysDept>().ToChildList(it => it.ParentId, user.DeptId);
|
||
|
||
var filter1 = new TableFilterItem<SysUser>(it => allChildDepts.Select(f => f.DeptId).ToList().Contains(it.DeptId));
|
||
db.QueryFilter.Add(filter1);
|
||
|
||
var filter2 = new TableFilterItem<SysDept>(it => allChildDepts.Select(f => f.DeptId).ToList().Contains(it.DeptId));
|
||
db.QueryFilter.Add(filter2);
|
||
}
|
||
else if (DATA_SCOPE_SELF.Equals(dataScope))//仅本人数据
|
||
{
|
||
var filter1 = new TableFilterItem<SysUser>(it => it.UserId == user.UserId, true);
|
||
db.QueryFilter.Add(filter1);
|
||
}
|
||
}
|
||
|
||
|
||
}
|
||
}
|
||
}
|