123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935 |
- using System.Threading.Tasks;
- using ZhonTai.Admin.Domain.DictType;
- using ZhonTai.Admin.Domain.Dict;
- using ZhonTai.Admin.Domain.Api;
- using ZhonTai.Admin.Domain.Permission;
- using ZhonTai.Admin.Domain.User;
- using ZhonTai.Admin.Domain.Role;
- using ZhonTai.Admin.Domain.UserRole;
- using ZhonTai.Admin.Domain.RolePermission;
- using ZhonTai.Admin.Domain.Tenant;
- using ZhonTai.Admin.Domain.TenantPermission;
- using ZhonTai.Admin.Domain.PermissionApi;
- using ZhonTai.Admin.Domain.View;
- using ZhonTai.Admin.Core.Configs;
- using ZhonTai.Admin.Domain.Org;
- using ZhonTai.Admin.Domain.UserStaff;
- using ZhonTai.Admin.Core.Db.Data;
- using ZhonTai.Admin.Domain.UserOrg;
- using System.Linq;
- using ZhonTai.Common.Extensions;
- using System;
- using FreeSql;
- using ZhonTai.Admin.Core.Auth;
- using Mapster;
- using System.Collections.Generic;
- namespace ZhonTai.Admin.Repositories;
- public class CustomSyncData : SyncData, ISyncData
- {
- /// <summary>
- /// 初始化字典类型
- /// </summary>
- /// <param name="db"></param>
- /// <param name="unitOfWork"></param>
- /// <param name="dbConfig"></param>
- /// <returns></returns>
- private async Task InitDictTypeAsync(IFreeSql db, IRepositoryUnitOfWork unitOfWork, DbConfig dbConfig)
- {
- var tableName = GetTableName<DictTypeEntity>();
- try
- {
- if (!IsSyncData(tableName, dbConfig))
- {
- return;
- }
- var rep = db.GetRepository<DictTypeEntity>();
- rep.UnitOfWork = unitOfWork;
- //数据列表
- var dataList = GetData<DictTypeEntity>(path: dbConfig.SyncDataPath);
- if (!(dataList?.Length > 0))
- {
- Console.WriteLine($"table: {tableName} import data []");
- return;
- }
- //查询
- var ids = dataList.Select(e => e.Id).ToList();
- var recordList = await rep.Where(a => ids.Contains(a.Id)).ToListAsync();
- //新增
- var recordIds = recordList.Select(a => a.Id).ToList();
- var insertDataList = dataList.Where(a => !recordIds.Contains(a.Id));
- if (insertDataList.Any())
- {
- await rep.InsertAsync(insertDataList);
- }
- //修改
- if (!dbConfig.SysUpdateData && recordList?.Count > 0)
- {
- var updateDataList = dataList.Where(a => recordIds.Contains(a.Id));
- await rep.UpdateAsync(updateDataList);
- }
- Console.WriteLine($"table: {tableName} sync data succeed");
- }
- catch (Exception ex)
- {
- var msg = $"table: {tableName} sync data failed.\n{ex.Message}";
- Console.WriteLine(msg);
- throw new Exception(msg);
- }
- }
- /// <summary>
- /// 初始化字典
- /// </summary>
- /// <param name="db"></param>
- /// <param name="unitOfWork"></param>
- /// <param name="dbConfig"></param>
- /// <returns></returns>
- private async Task InitDictAsync(IFreeSql db, IRepositoryUnitOfWork unitOfWork, DbConfig dbConfig)
- {
- var tableName = GetTableName<DictEntity>();
- try
- {
- if (!IsSyncData(tableName, dbConfig))
- {
- return;
- }
- var rep = db.GetRepository<DictEntity>();
- rep.UnitOfWork = unitOfWork;
- //数据列表
- var dataList = GetData<DictEntity>(path: dbConfig.SyncDataPath);
- if (!(dataList?.Length > 0))
- {
- Console.WriteLine($"table: {tableName} import data []");
- return;
- }
- //查询
- var ids = dataList.Select(e => e.Id).ToList();
- var recordList = await rep.Where(a => ids.Contains(a.Id)).ToListAsync();
- //新增
- var recordIds = recordList.Select(a => a.Id).ToList();
- var insertDataList = dataList.Where(a => !recordIds.Contains(a.Id));
- if (insertDataList.Any())
- {
- await rep.InsertAsync(insertDataList);
- }
- //修改
- if (!dbConfig.SysUpdateData && recordList?.Count > 0)
- {
- var updateDataList = dataList.Where(a => recordIds.Contains(a.Id));
- await rep.UpdateAsync(updateDataList);
- }
- Console.WriteLine($"table: {tableName} sync data succeed");
- }
- catch (Exception ex)
- {
- var msg = $"table: {tableName} sync data failed.\n{ex.Message}";
- Console.WriteLine(msg);
- throw new Exception(msg);
- }
- }
- /// <summary>
- /// 初始化用户
- /// </summary>
- /// <param name="db"></param>
- /// <param name="unitOfWork"></param>
- /// <param name="dbConfig"></param>
- /// <param name="isTenant"></param>
- /// <returns></returns>
- private async Task InitUserAsync(IFreeSql db, IRepositoryUnitOfWork unitOfWork, DbConfig dbConfig, bool isTenant)
- {
- var tableName = GetTableName<UserEntity>();
- try
- {
- if (!IsSyncData(tableName, dbConfig))
- {
- return;
- }
- var rep = db.GetRepository<UserEntity>();
- rep.UnitOfWork = unitOfWork;
- //数据列表
- var dataList = GetData<UserEntity>(isTenant, dbConfig.SyncDataPath);
- if (!(dataList?.Length > 0))
- {
- Console.WriteLine($"table: {tableName} import data []");
- return;
- }
- //查询
- var ids = dataList.Select(e => e.Id).ToList();
- var recordList = await rep.Where(a => ids.Contains(a.Id)).ToListAsync();
- //新增
- var recordIds = recordList.Select(a => a.Id).ToList();
- var insertDataList = dataList.Where(a => !recordIds.Contains(a.Id));
- if (insertDataList.Any())
- {
- await rep.InsertAsync(insertDataList);
- }
- //修改
- if (!dbConfig.SysUpdateData && recordList?.Count > 0)
- {
- var updateDataList = dataList.Where(a => recordIds.Contains(a.Id));
- await rep.UpdateAsync(updateDataList);
- }
- Console.WriteLine($"table: {tableName} sync data succeed");
- }
- catch (Exception ex)
- {
- var msg = $"table: {tableName} sync data failed.\n{ex.Message}";
- Console.WriteLine(msg);
- throw new Exception(msg);
- }
- }
- /// <summary>
- /// 初始化用户员工
- /// </summary>
- /// <param name="db"></param>
- /// <param name="unitOfWork"></param>
- /// <param name="dbConfig"></param>
- /// <param name="isTenant"></param>
- /// <returns></returns>
- private async Task InitUserStaffAsync(IFreeSql db, IRepositoryUnitOfWork unitOfWork, DbConfig dbConfig, bool isTenant)
- {
- var tableName = GetTableName<UserStaffEntity>();
- try
- {
- if (!IsSyncData(tableName, dbConfig))
- {
- return;
- }
- var rep = db.GetRepository<UserStaffEntity>();
- rep.UnitOfWork = unitOfWork;
- //数据列表
- var dataList = GetData<UserStaffEntity>(isTenant, dbConfig.SyncDataPath);
- if (!(dataList?.Length > 0))
- {
- Console.WriteLine($"table: {tableName} import data []");
- return;
- }
- //查询
- var ids = dataList.Select(e => e.Id).ToList();
- var recordList = await rep.Where(a => ids.Contains(a.Id)).ToListAsync();
- //新增
- var recordIds = recordList.Select(a => a.Id).ToList();
- var insertDataList = dataList.Where(a => !recordIds.Contains(a.Id));
- if (insertDataList.Any())
- {
- await rep.InsertAsync(insertDataList);
- }
- //修改
- if (!dbConfig.SysUpdateData && recordList?.Count > 0)
- {
- var updateDataList = dataList.Where(a => recordIds.Contains(a.Id));
- await rep.UpdateAsync(updateDataList);
- }
- Console.WriteLine($"table: {tableName} sync data succeed");
- }
- catch (Exception ex)
- {
- var msg = $"table: {tableName} sync data failed.\n{ex.Message}";
- Console.WriteLine(msg);
- throw new Exception(msg);
- }
- }
- /// <summary>
- /// 初始化部门
- /// </summary>
- /// <param name="db"></param>
- /// <param name="unitOfWork"></param>
- /// <param name="dbConfig"></param>
- /// <param name="isTenant"></param>
- /// <returns></returns>
- private async Task InitOrgAsync(IFreeSql db, IRepositoryUnitOfWork unitOfWork, DbConfig dbConfig, bool isTenant)
- {
- var tableName = GetTableName<OrgEntity>();
- try
- {
- if (!IsSyncData(tableName, dbConfig))
- {
- return;
- }
- var rep = db.GetRepository<OrgEntity>();
- rep.UnitOfWork = unitOfWork;
- //数据列表
- var dataTree = GetData<OrgEntity>(isTenant, dbConfig.SyncDataPath);
- var dataList = dataTree.ToList().ToPlainList((a) => a.Childs).ToArray();
- if (!(dataList?.Length > 0))
- {
- Console.WriteLine($"table: {tableName} import data []");
- return;
- }
- //查询
- var ids = dataList.Select(e => e.Id).ToList();
- var recordList = await rep.Where(a => ids.Contains(a.Id)).ToListAsync();
- //新增
- var recordIds = recordList.Select(a => a.Id).ToList();
- var insertDataList = dataList.Where(a => !recordIds.Contains(a.Id));
- if (insertDataList.Any())
- {
- await rep.InsertAsync(insertDataList);
- }
- //修改
- if (!dbConfig.SysUpdateData && recordList?.Count > 0)
- {
- var updateDataList = dataList.Where(a => recordIds.Contains(a.Id));
- await rep.UpdateAsync(updateDataList);
- }
- Console.WriteLine($"table: {tableName} sync data succeed");
- }
- catch (Exception ex)
- {
- var msg = $"table: {tableName} sync data failed.\n{ex.Message}";
- Console.WriteLine(msg);
- throw new Exception(msg);
- }
- }
- /// <summary>
- /// 初始化角色
- /// </summary>
- /// <param name="db"></param>
- /// <param name="unitOfWork"></param>
- /// <param name="dbConfig"></param>
- /// <param name="isTenant"></param>
- /// <returns></returns>
- private async Task InitRoleAsync(IFreeSql db, IRepositoryUnitOfWork unitOfWork, DbConfig dbConfig, bool isTenant)
- {
- var tableName = GetTableName<RoleEntity>();
- try
- {
- if (!IsSyncData(tableName, dbConfig))
- {
- return;
- }
- var rep = db.GetRepository<RoleEntity>();
- rep.UnitOfWork = unitOfWork;
- //数据列表
- var dataList = GetData<RoleEntity>(isTenant, dbConfig.SyncDataPath);
- if (!(dataList?.Length > 0))
- {
- Console.WriteLine($"table: {tableName} import data []");
- return;
- }
- //查询
- var ids = dataList.Select(e => e.Id).ToList();
- var recordList = await rep.Where(a => ids.Contains(a.Id)).ToListAsync();
- //新增
- var recordIds = recordList.Select(a => a.Id).ToList();
- var insertDataList = dataList.Where(a => !recordIds.Contains(a.Id));
- if (insertDataList.Any())
- {
- await rep.InsertAsync(insertDataList);
- }
- //修改
- if (!dbConfig.SysUpdateData && recordList?.Count > 0)
- {
- var updateDataList = dataList.Where(a => recordIds.Contains(a.Id));
- await rep.UpdateAsync(updateDataList);
- }
- Console.WriteLine($"table: {tableName} sync data succeed");
- }
- catch (Exception ex)
- {
- var msg = $"table: {tableName} sync data failed.\n{ex.Message}";
- Console.WriteLine(msg);
- throw new Exception(msg);
- }
- }
- /// <summary>
- /// 初始化接口
- /// </summary>
- /// <param name="db"></param>
- /// <param name="unitOfWork"></param>
- /// <param name="dbConfig"></param>
- /// <returns></returns>
- private async Task InitApiAsync(IFreeSql db, IRepositoryUnitOfWork unitOfWork, DbConfig dbConfig)
- {
- var tableName = GetTableName<ApiEntity>();
- try
- {
- if (!IsSyncData(tableName, dbConfig))
- {
- return;
- }
- var rep = db.GetRepository<ApiEntity>();
- rep.UnitOfWork = unitOfWork;
- //数据列表
- var dataTree = GetData<ApiEntity>(path: dbConfig.SyncDataPath);
- var dataList = dataTree.ToList().ToPlainList((a) => a.Childs).ToArray();
- if (!(dataList?.Length > 0))
- {
- Console.WriteLine($"table: {tableName} import data []");
- return;
- }
- //查询
- var ids = dataList.Select(e => e.Id).ToList();
- var recordList = await rep.Where(a => ids.Contains(a.Id)).ToListAsync();
- //新增
- var recordIds = recordList.Select(a => a.Id).ToList();
- var insertDataList = dataList.Where(a => !recordIds.Contains(a.Id));
- if (insertDataList.Any())
- {
- await rep.InsertAsync(insertDataList);
- }
- //修改
- if (!dbConfig.SysUpdateData && recordList?.Count > 0)
- {
- var updateDataList = dataList.Where(a => recordIds.Contains(a.Id));
- await rep.UpdateAsync(updateDataList);
- }
- Console.WriteLine($"table: {tableName} sync data succeed");
- }
- catch (Exception ex)
- {
- var msg = $"table: {tableName} sync data failed.\n{ex.Message}";
- Console.WriteLine(msg);
- throw new Exception(msg);
- }
- }
- /// <summary>
- /// 初始化视图
- /// </summary>
- /// <param name="db"></param>
- /// <param name="unitOfWork"></param>
- /// <param name="dbConfig"></param>
- /// <returns></returns>
- private async Task InitViewAsync(IFreeSql db, IRepositoryUnitOfWork unitOfWork, DbConfig dbConfig)
- {
- var tableName = GetTableName<ViewEntity>();
- try
- {
- if (!IsSyncData(tableName, dbConfig))
- {
- return;
- }
- var rep = db.GetRepository<ViewEntity>();
- rep.UnitOfWork = unitOfWork;
- //数据列表
- var dataTree = GetData<ViewEntity>(path: dbConfig.SyncDataPath);
- var dataList = dataTree.ToList().ToPlainList((a) => a.Childs).ToArray();
- if (!(dataList?.Length > 0))
- {
- Console.WriteLine($"table: {tableName} import data []");
- return;
- }
- //查询
- var ids = dataList.Select(e => e.Id).ToList();
- var recordList = await rep.Where(a => ids.Contains(a.Id)).ToListAsync();
- //新增
- var recordIds = recordList.Select(a => a.Id).ToList();
- var insertDataList = dataList.Where(a => !recordIds.Contains(a.Id));
- if (insertDataList.Any())
- {
- await rep.InsertAsync(insertDataList);
- }
- //修改
- if (!dbConfig.SysUpdateData && recordList?.Count > 0)
- {
- var updateDataList = dataList.Where(a => recordIds.Contains(a.Id));
- await rep.UpdateAsync(updateDataList);
- }
- Console.WriteLine($"table: {tableName} sync data succeed");
- }
- catch (Exception ex)
- {
- var msg = $"table: {tableName} sync data failed.\n{ex.Message}";
- Console.WriteLine(msg);
- throw new Exception(msg);
- }
- }
- /// <summary>
- /// 初始化权限
- /// </summary>
- /// <param name="db"></param>
- /// <param name="unitOfWork"></param>
- /// <param name="dbConfig"></param>
- /// <returns></returns>
- private async Task InitPermissionAsync(IFreeSql db, IRepositoryUnitOfWork unitOfWork, DbConfig dbConfig)
- {
- var tableName = GetTableName<PermissionEntity>();
- try
- {
- if (!IsSyncData(tableName, dbConfig))
- {
- return;
- }
- var rep = db.GetRepository<PermissionEntity>();
- rep.UnitOfWork = unitOfWork;
- //数据列表
- var dataTree = GetData<PermissionEntity>(path: dbConfig.SyncDataPath);
- var dataList = dataTree.ToList().ToPlainList((a) => a.Childs).ToArray();
- if (!(dataList?.Length > 0))
- {
- Console.WriteLine($"table: {tableName} import data []");
- return;
- }
- //查询
- var ids = dataList.Select(e => e.Id).ToList();
- var recordList = await rep.Where(a => ids.Contains(a.Id)).ToListAsync();
- //新增
- var recordIds = recordList.Select(a => a.Id).ToList();
- var insertDataList = dataList.Where(a => !recordIds.Contains(a.Id));
- if (insertDataList.Any())
- {
- await rep.InsertAsync(insertDataList);
- }
- //修改
- if (!dbConfig.SysUpdateData && recordList?.Count > 0)
- {
- var updateDataList = dataList.Where(a => recordIds.Contains(a.Id));
- await rep.UpdateAsync(updateDataList);
- }
- Console.WriteLine($"table: {tableName} sync data succeed");
- }
- catch (Exception ex)
- {
- var msg = $"table: {tableName} sync data failed.\n{ex.Message}";
- Console.WriteLine(msg);
- throw new Exception(msg);
- }
- }
- /// <summary>
- /// 初始化权限接口
- /// </summary>
- /// <param name="db"></param>
- /// <param name="unitOfWork"></param>
- /// <param name="dbConfig"></param>
- /// <returns></returns>
- private async Task InitPermissionApiAsync(IFreeSql db, IRepositoryUnitOfWork unitOfWork, DbConfig dbConfig)
- {
- var tableName = GetTableName<PermissionApiEntity>();
- try
- {
- if (!IsSyncData(tableName, dbConfig))
- {
- return;
- }
- var rep = db.GetRepository<PermissionApiEntity>();
- rep.UnitOfWork = unitOfWork;
- //数据列表
- var dataList = GetData<PermissionApiEntity>(path: dbConfig.SyncDataPath);
- if (!(dataList?.Length > 0))
- {
- Console.WriteLine($"table: {tableName} import data []");
- return;
- }
- //查询
- var recordList = await rep.Where(a => rep.Select.WithMemory(dataList).Where(b => b.PermissionId == a.PermissionId && b.ApiId == a.ApiId).Any()).ToListAsync();
- //新增
- var insertDataList = dataList.Where(a => !(recordList.Where(b => a.PermissionId == b.PermissionId && a.ApiId == b.ApiId).Any())).ToList();
- if (insertDataList.Any())
- {
- await rep.InsertAsync(insertDataList);
- }
- Console.WriteLine($"table: {tableName} sync data succeed");
- }
- catch (Exception ex)
- {
- var msg = $"table: {tableName} sync data failed.\n{ex.Message}";
- Console.WriteLine(msg);
- throw new Exception(msg);
- }
- }
- /// <summary>
- /// 用户角色记录
- /// </summary>
- /// <param name="UserId"></param>
- /// <param name="RoleId"></param>
- record UserRoleRecord(long UserId, long RoleId);
- /// <summary>
- /// 初始化用户角色
- /// </summary>
- /// <param name="db"></param>
- /// <param name="unitOfWork"></param>
- /// <param name="dbConfig"></param>
- /// <returns></returns>
- private async Task InitUserRoleAsync(IFreeSql db, IRepositoryUnitOfWork unitOfWork, DbConfig dbConfig)
- {
- var tableName = GetTableName<UserRoleEntity>();
- try
- {
- if (!IsSyncData(tableName, dbConfig))
- {
- return;
- }
- var rep = db.GetRepository<UserRoleEntity>();
- rep.UnitOfWork = unitOfWork;
- //数据列表
- var dataList = GetData<UserRoleEntity>(path: dbConfig.SyncDataPath);
- if (!(dataList?.Length > 0))
- {
- Console.WriteLine($"table: {tableName} import data []");
- return;
- }
- //查询
- var userRoleRecordList = dataList.Adapt<List<UserRoleRecord>>();
- var recordList = await rep.Where(a => rep.Orm.Select<UserRoleRecord>().WithMemory(userRoleRecordList).Where(b => b.UserId == a.UserId && b.RoleId == a.RoleId).Any()).ToListAsync();
- //新增
- var insertDataList = dataList.Where(a => !(recordList.Where(b => a.UserId == b.UserId && a.RoleId == b.RoleId).Any())).ToList();
- if (insertDataList.Any())
- {
- await rep.InsertAsync(insertDataList);
- }
- Console.WriteLine($"table: {tableName} sync data succeed");
- }
- catch (Exception ex)
- {
- var msg = $"table: {tableName} sync data failed.\n{ex.Message}";
- Console.WriteLine(msg);
- throw new Exception(msg);
- }
- }
- /// <summary>
- /// 用户部门记录
- /// </summary>
- /// <param name="UserId"></param>
- /// <param name="OrgId"></param>
- record UserOrgRecord(long UserId, long OrgId);
- /// <summary>
- /// 初始化用户部门
- /// </summary>
- /// <param name="db"></param>
- /// <param name="unitOfWork"></param>
- /// <param name="dbConfig"></param>
- /// <returns></returns>
- private async Task InitUserOrgAsync(IFreeSql db, IRepositoryUnitOfWork unitOfWork, DbConfig dbConfig)
- {
- var tableName = GetTableName<UserOrgEntity>();
- try
- {
- if (!IsSyncData(tableName, dbConfig))
- {
- return;
- }
- var rep = db.GetRepository<UserOrgEntity>();
- rep.UnitOfWork = unitOfWork;
- //数据列表
- var dataList = GetData<UserOrgEntity>(path: dbConfig.SyncDataPath);
- if (!(dataList?.Length > 0))
- {
- Console.WriteLine($"table: {tableName} import data []");
- return;
- }
- //查询
- var userOrgRecordList = dataList.Adapt<List<UserOrgRecord>>();
- var recordList = await rep.Where(a => rep.Orm.Select<UserOrgRecord>().WithMemory(userOrgRecordList).Where(b => b.UserId == a.UserId && b.OrgId == a.OrgId).Any()).ToListAsync();
- //新增
- var insertDataList = dataList.Where(a => !(recordList.Where(b => a.UserId == b.UserId && a.OrgId == b.OrgId).Any())).ToList();
- if (insertDataList.Any())
- {
- await rep.InsertAsync(insertDataList);
- }
- Console.WriteLine($"table: {tableName} sync data succeed");
- }
- catch (Exception ex)
- {
- var msg = $"table: {tableName} sync data failed.\n{ex.Message}";
- Console.WriteLine(msg);
- throw new Exception(msg);
- }
- }
- /// <summary>
- /// 角色权限记录
- /// </summary>
- /// <param name="RoleId"></param>
- /// <param name="PermissionId"></param>
- record RolePermissionRecord(long RoleId, long PermissionId);
- /// <summary>
- /// 初始化角色权限
- /// </summary>
- /// <param name="db"></param>
- /// <param name="unitOfWork"></param>
- /// <param name="dbConfig"></param>
- /// <returns></returns>
- private async Task InitRolePermissionAsync(IFreeSql db, IRepositoryUnitOfWork unitOfWork, DbConfig dbConfig)
- {
- var tableName = GetTableName<RolePermissionEntity>();
- try
- {
- if (!IsSyncData(tableName, dbConfig))
- {
- return;
- }
- var rep = db.GetRepository<RolePermissionEntity>();
- rep.UnitOfWork = unitOfWork;
- //数据列表
- var dataList = GetData<RolePermissionEntity>(path: dbConfig.SyncDataPath);
- if (!(dataList?.Length > 0))
- {
- Console.WriteLine($"table: {tableName} import data []");
- return;
- }
- //查询
- var rolePermissionRecordList = dataList.Adapt<List<RolePermissionRecord>>();
- var recordList = await rep.Where(a => rep.Orm.Select<RolePermissionRecord>().WithMemory(rolePermissionRecordList).Where(b => b.RoleId == a.RoleId && b.PermissionId == a.PermissionId).Any()).ToListAsync();
- //新增
- var insertDataList = dataList.Where(a => !(recordList.Where(b => a.RoleId == b.RoleId && a.PermissionId == b.PermissionId).Any())).ToList();
- if (insertDataList.Any())
- {
- await rep.InsertAsync(insertDataList);
- }
- Console.WriteLine($"table: {tableName} sync data succeed");
- }
- catch (Exception ex)
- {
- var msg = $"table: {tableName} sync data failed.\n{ex.Message}";
- Console.WriteLine(msg);
- throw new Exception(msg);
- }
- }
- /// <summary>
- /// 初始化租户
- /// </summary>
- /// <param name="db"></param>
- /// <param name="unitOfWork"></param>
- /// <param name="dbConfig"></param>
- /// <returns></returns>
- private async Task InitTenantAsync(IFreeSql db, IRepositoryUnitOfWork unitOfWork, DbConfig dbConfig)
- {
- var tableName = GetTableName<TenantEntity>();
- try
- {
- if (!IsSyncData(tableName, dbConfig))
- {
- return;
- }
- var rep = db.GetRepository<TenantEntity>();
- rep.UnitOfWork = unitOfWork;
- //数据列表
- var dataList = GetData<TenantEntity>(path: dbConfig.SyncDataPath);
- if (!(dataList?.Length > 0))
- {
- Console.WriteLine($"table: {tableName} import data []");
- return;
- }
- //查询
- var ids = dataList.Select(e => e.Id).ToList();
- var recordList = await rep.Where(a => ids.Contains(a.Id)).ToListAsync();
- //新增
- var recordIds = recordList.Select(a => a.Id).ToList();
- var insertDataList = dataList.Where(a => !recordIds.Contains(a.Id));
- if (insertDataList.Any())
- {
- await rep.InsertAsync(insertDataList);
- }
- //修改
- if (!dbConfig.SysUpdateData && recordList?.Count > 0)
- {
- var updateDataList = dataList.Where(a => recordIds.Contains(a.Id));
- await rep.UpdateAsync(updateDataList);
- }
- Console.WriteLine($"table: {tableName} sync data succeed");
- }
- catch (Exception ex)
- {
- var msg = $"table: {tableName} sync data failed.\n{ex.Message}";
- Console.WriteLine(msg);
- throw new Exception(msg);
- }
- }
- /// <summary>
- /// 租户权限记录
- /// </summary>
- /// <param name="TenantId"></param>
- /// <param name="PermissionId"></param>
- record TenantPermissionRecord(long TenantId, long PermissionId);
- /// <summary>
- /// 初始化租户权限
- /// </summary>
- /// <param name="db"></param>
- /// <param name="unitOfWork"></param>
- /// <param name="dbConfig"></param>
- /// <returns></returns>
- private async Task InitTenantPermissionAsync(IFreeSql db, IRepositoryUnitOfWork unitOfWork, DbConfig dbConfig)
- {
- var tableName = GetTableName<TenantPermissionEntity>();
- try
- {
- if (!IsSyncData(tableName, dbConfig))
- {
- return;
- }
- var rep = db.GetRepository<TenantPermissionEntity>();
- rep.UnitOfWork = unitOfWork;
- //数据列表
- var dataList = GetData<TenantPermissionEntity>(path: dbConfig.SyncDataPath);
- if (!(dataList?.Length > 0))
- {
- Console.WriteLine($"table: {tableName} import data []");
- return;
- }
- //查询
- var tenantPermissionRecordList = dataList.Adapt<List<TenantPermissionRecord>>();
- var recordList = await rep.Where(a => rep.Orm.Select<TenantPermissionRecord>().WithMemory(tenantPermissionRecordList).Where(b => b.TenantId == a.TenantId && b.PermissionId == a.PermissionId).Any()).ToListAsync();
- //新增
- var insertDataList = dataList.Where(a => !(recordList.Where(b => a.TenantId == b.TenantId && a.PermissionId == b.PermissionId).Any())).ToList();
- if (insertDataList.Any())
- {
- await rep.InsertAsync(insertDataList);
- }
- Console.WriteLine($"table: {tableName} sync data succeed");
- }
- catch (Exception ex)
- {
- var msg = $"table: {tableName} sync data failed.\n{ex.Message}";
- Console.WriteLine(msg);
- throw new Exception(msg);
- }
- }
- /// <summary>
- /// 同步数据
- /// </summary>
- /// <param name="db"></param>
- /// <param name="dbConfig"></param>
- /// <param name="appConfig"></param>
- /// <returns></returns>
- public virtual async Task SyncDataAsync(IFreeSql db, DbConfig dbConfig = null, AppConfig appConfig = null)
- {
- using var unitOfWork = db.CreateUnitOfWork();
- try
- {
- var isTenant = appConfig.Tenant;
- await InitDictTypeAsync(db, unitOfWork, dbConfig);
- await InitDictAsync(db, unitOfWork, dbConfig);
- await InitUserAsync(db, unitOfWork, dbConfig, isTenant);
- await InitUserStaffAsync(db, unitOfWork, dbConfig, isTenant);
- await InitOrgAsync(db, unitOfWork, dbConfig, isTenant);
- await InitRoleAsync(db, unitOfWork, dbConfig, isTenant);
- await InitApiAsync(db, unitOfWork, dbConfig);
- await InitViewAsync(db, unitOfWork, dbConfig);
- await InitPermissionAsync(db, unitOfWork, dbConfig);
- await InitPermissionApiAsync(db, unitOfWork, dbConfig);
- await InitUserRoleAsync(db, unitOfWork, dbConfig);
- await InitUserOrgAsync(db, unitOfWork, dbConfig);
- await InitRolePermissionAsync(db, unitOfWork, dbConfig);
- await InitTenantAsync(db, unitOfWork, dbConfig);
- await InitTenantPermissionAsync(db, unitOfWork, dbConfig);
- unitOfWork.Commit();
- }
- catch (Exception)
- {
- unitOfWork.Rollback();
- throw;
- }
- }
- }
|