1
0

CustomSyncData.cs 31 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935
  1. using System.Threading.Tasks;
  2. using ZhonTai.Admin.Domain.DictType;
  3. using ZhonTai.Admin.Domain.Dict;
  4. using ZhonTai.Admin.Domain.Api;
  5. using ZhonTai.Admin.Domain.Permission;
  6. using ZhonTai.Admin.Domain.User;
  7. using ZhonTai.Admin.Domain.Role;
  8. using ZhonTai.Admin.Domain.UserRole;
  9. using ZhonTai.Admin.Domain.RolePermission;
  10. using ZhonTai.Admin.Domain.Tenant;
  11. using ZhonTai.Admin.Domain.TenantPermission;
  12. using ZhonTai.Admin.Domain.PermissionApi;
  13. using ZhonTai.Admin.Domain.View;
  14. using ZhonTai.Admin.Core.Configs;
  15. using ZhonTai.Admin.Domain.Org;
  16. using ZhonTai.Admin.Domain.UserStaff;
  17. using ZhonTai.Admin.Core.Db.Data;
  18. using ZhonTai.Admin.Domain.UserOrg;
  19. using System.Linq;
  20. using ZhonTai.Common.Extensions;
  21. using System;
  22. using FreeSql;
  23. using ZhonTai.Admin.Core.Auth;
  24. using Mapster;
  25. using System.Collections.Generic;
  26. namespace ZhonTai.Admin.Repositories;
  27. public class CustomSyncData : SyncData, ISyncData
  28. {
  29. /// <summary>
  30. /// 初始化字典类型
  31. /// </summary>
  32. /// <param name="db"></param>
  33. /// <param name="unitOfWork"></param>
  34. /// <param name="dbConfig"></param>
  35. /// <returns></returns>
  36. private async Task InitDictTypeAsync(IFreeSql db, IRepositoryUnitOfWork unitOfWork, DbConfig dbConfig)
  37. {
  38. var tableName = GetTableName<DictTypeEntity>();
  39. try
  40. {
  41. if (!IsSyncData(tableName, dbConfig))
  42. {
  43. return;
  44. }
  45. var rep = db.GetRepository<DictTypeEntity>();
  46. rep.UnitOfWork = unitOfWork;
  47. //数据列表
  48. var dataList = GetData<DictTypeEntity>(path: dbConfig.SyncDataPath);
  49. if (!(dataList?.Length > 0))
  50. {
  51. Console.WriteLine($"table: {tableName} import data []");
  52. return;
  53. }
  54. //查询
  55. var ids = dataList.Select(e => e.Id).ToList();
  56. var recordList = await rep.Where(a => ids.Contains(a.Id)).ToListAsync();
  57. //新增
  58. var recordIds = recordList.Select(a => a.Id).ToList();
  59. var insertDataList = dataList.Where(a => !recordIds.Contains(a.Id));
  60. if (insertDataList.Any())
  61. {
  62. await rep.InsertAsync(insertDataList);
  63. }
  64. //修改
  65. if (!dbConfig.SysUpdateData && recordList?.Count > 0)
  66. {
  67. var updateDataList = dataList.Where(a => recordIds.Contains(a.Id));
  68. await rep.UpdateAsync(updateDataList);
  69. }
  70. Console.WriteLine($"table: {tableName} sync data succeed");
  71. }
  72. catch (Exception ex)
  73. {
  74. var msg = $"table: {tableName} sync data failed.\n{ex.Message}";
  75. Console.WriteLine(msg);
  76. throw new Exception(msg);
  77. }
  78. }
  79. /// <summary>
  80. /// 初始化字典
  81. /// </summary>
  82. /// <param name="db"></param>
  83. /// <param name="unitOfWork"></param>
  84. /// <param name="dbConfig"></param>
  85. /// <returns></returns>
  86. private async Task InitDictAsync(IFreeSql db, IRepositoryUnitOfWork unitOfWork, DbConfig dbConfig)
  87. {
  88. var tableName = GetTableName<DictEntity>();
  89. try
  90. {
  91. if (!IsSyncData(tableName, dbConfig))
  92. {
  93. return;
  94. }
  95. var rep = db.GetRepository<DictEntity>();
  96. rep.UnitOfWork = unitOfWork;
  97. //数据列表
  98. var dataList = GetData<DictEntity>(path: dbConfig.SyncDataPath);
  99. if (!(dataList?.Length > 0))
  100. {
  101. Console.WriteLine($"table: {tableName} import data []");
  102. return;
  103. }
  104. //查询
  105. var ids = dataList.Select(e => e.Id).ToList();
  106. var recordList = await rep.Where(a => ids.Contains(a.Id)).ToListAsync();
  107. //新增
  108. var recordIds = recordList.Select(a => a.Id).ToList();
  109. var insertDataList = dataList.Where(a => !recordIds.Contains(a.Id));
  110. if (insertDataList.Any())
  111. {
  112. await rep.InsertAsync(insertDataList);
  113. }
  114. //修改
  115. if (!dbConfig.SysUpdateData && recordList?.Count > 0)
  116. {
  117. var updateDataList = dataList.Where(a => recordIds.Contains(a.Id));
  118. await rep.UpdateAsync(updateDataList);
  119. }
  120. Console.WriteLine($"table: {tableName} sync data succeed");
  121. }
  122. catch (Exception ex)
  123. {
  124. var msg = $"table: {tableName} sync data failed.\n{ex.Message}";
  125. Console.WriteLine(msg);
  126. throw new Exception(msg);
  127. }
  128. }
  129. /// <summary>
  130. /// 初始化用户
  131. /// </summary>
  132. /// <param name="db"></param>
  133. /// <param name="unitOfWork"></param>
  134. /// <param name="dbConfig"></param>
  135. /// <param name="isTenant"></param>
  136. /// <returns></returns>
  137. private async Task InitUserAsync(IFreeSql db, IRepositoryUnitOfWork unitOfWork, DbConfig dbConfig, bool isTenant)
  138. {
  139. var tableName = GetTableName<UserEntity>();
  140. try
  141. {
  142. if (!IsSyncData(tableName, dbConfig))
  143. {
  144. return;
  145. }
  146. var rep = db.GetRepository<UserEntity>();
  147. rep.UnitOfWork = unitOfWork;
  148. //数据列表
  149. var dataList = GetData<UserEntity>(isTenant, dbConfig.SyncDataPath);
  150. if (!(dataList?.Length > 0))
  151. {
  152. Console.WriteLine($"table: {tableName} import data []");
  153. return;
  154. }
  155. //查询
  156. var ids = dataList.Select(e => e.Id).ToList();
  157. var recordList = await rep.Where(a => ids.Contains(a.Id)).ToListAsync();
  158. //新增
  159. var recordIds = recordList.Select(a => a.Id).ToList();
  160. var insertDataList = dataList.Where(a => !recordIds.Contains(a.Id));
  161. if (insertDataList.Any())
  162. {
  163. await rep.InsertAsync(insertDataList);
  164. }
  165. //修改
  166. if (!dbConfig.SysUpdateData && recordList?.Count > 0)
  167. {
  168. var updateDataList = dataList.Where(a => recordIds.Contains(a.Id));
  169. await rep.UpdateAsync(updateDataList);
  170. }
  171. Console.WriteLine($"table: {tableName} sync data succeed");
  172. }
  173. catch (Exception ex)
  174. {
  175. var msg = $"table: {tableName} sync data failed.\n{ex.Message}";
  176. Console.WriteLine(msg);
  177. throw new Exception(msg);
  178. }
  179. }
  180. /// <summary>
  181. /// 初始化用户员工
  182. /// </summary>
  183. /// <param name="db"></param>
  184. /// <param name="unitOfWork"></param>
  185. /// <param name="dbConfig"></param>
  186. /// <param name="isTenant"></param>
  187. /// <returns></returns>
  188. private async Task InitUserStaffAsync(IFreeSql db, IRepositoryUnitOfWork unitOfWork, DbConfig dbConfig, bool isTenant)
  189. {
  190. var tableName = GetTableName<UserStaffEntity>();
  191. try
  192. {
  193. if (!IsSyncData(tableName, dbConfig))
  194. {
  195. return;
  196. }
  197. var rep = db.GetRepository<UserStaffEntity>();
  198. rep.UnitOfWork = unitOfWork;
  199. //数据列表
  200. var dataList = GetData<UserStaffEntity>(isTenant, dbConfig.SyncDataPath);
  201. if (!(dataList?.Length > 0))
  202. {
  203. Console.WriteLine($"table: {tableName} import data []");
  204. return;
  205. }
  206. //查询
  207. var ids = dataList.Select(e => e.Id).ToList();
  208. var recordList = await rep.Where(a => ids.Contains(a.Id)).ToListAsync();
  209. //新增
  210. var recordIds = recordList.Select(a => a.Id).ToList();
  211. var insertDataList = dataList.Where(a => !recordIds.Contains(a.Id));
  212. if (insertDataList.Any())
  213. {
  214. await rep.InsertAsync(insertDataList);
  215. }
  216. //修改
  217. if (!dbConfig.SysUpdateData && recordList?.Count > 0)
  218. {
  219. var updateDataList = dataList.Where(a => recordIds.Contains(a.Id));
  220. await rep.UpdateAsync(updateDataList);
  221. }
  222. Console.WriteLine($"table: {tableName} sync data succeed");
  223. }
  224. catch (Exception ex)
  225. {
  226. var msg = $"table: {tableName} sync data failed.\n{ex.Message}";
  227. Console.WriteLine(msg);
  228. throw new Exception(msg);
  229. }
  230. }
  231. /// <summary>
  232. /// 初始化部门
  233. /// </summary>
  234. /// <param name="db"></param>
  235. /// <param name="unitOfWork"></param>
  236. /// <param name="dbConfig"></param>
  237. /// <param name="isTenant"></param>
  238. /// <returns></returns>
  239. private async Task InitOrgAsync(IFreeSql db, IRepositoryUnitOfWork unitOfWork, DbConfig dbConfig, bool isTenant)
  240. {
  241. var tableName = GetTableName<OrgEntity>();
  242. try
  243. {
  244. if (!IsSyncData(tableName, dbConfig))
  245. {
  246. return;
  247. }
  248. var rep = db.GetRepository<OrgEntity>();
  249. rep.UnitOfWork = unitOfWork;
  250. //数据列表
  251. var dataTree = GetData<OrgEntity>(isTenant, dbConfig.SyncDataPath);
  252. var dataList = dataTree.ToList().ToPlainList((a) => a.Childs).ToArray();
  253. if (!(dataList?.Length > 0))
  254. {
  255. Console.WriteLine($"table: {tableName} import data []");
  256. return;
  257. }
  258. //查询
  259. var ids = dataList.Select(e => e.Id).ToList();
  260. var recordList = await rep.Where(a => ids.Contains(a.Id)).ToListAsync();
  261. //新增
  262. var recordIds = recordList.Select(a => a.Id).ToList();
  263. var insertDataList = dataList.Where(a => !recordIds.Contains(a.Id));
  264. if (insertDataList.Any())
  265. {
  266. await rep.InsertAsync(insertDataList);
  267. }
  268. //修改
  269. if (!dbConfig.SysUpdateData && recordList?.Count > 0)
  270. {
  271. var updateDataList = dataList.Where(a => recordIds.Contains(a.Id));
  272. await rep.UpdateAsync(updateDataList);
  273. }
  274. Console.WriteLine($"table: {tableName} sync data succeed");
  275. }
  276. catch (Exception ex)
  277. {
  278. var msg = $"table: {tableName} sync data failed.\n{ex.Message}";
  279. Console.WriteLine(msg);
  280. throw new Exception(msg);
  281. }
  282. }
  283. /// <summary>
  284. /// 初始化角色
  285. /// </summary>
  286. /// <param name="db"></param>
  287. /// <param name="unitOfWork"></param>
  288. /// <param name="dbConfig"></param>
  289. /// <param name="isTenant"></param>
  290. /// <returns></returns>
  291. private async Task InitRoleAsync(IFreeSql db, IRepositoryUnitOfWork unitOfWork, DbConfig dbConfig, bool isTenant)
  292. {
  293. var tableName = GetTableName<RoleEntity>();
  294. try
  295. {
  296. if (!IsSyncData(tableName, dbConfig))
  297. {
  298. return;
  299. }
  300. var rep = db.GetRepository<RoleEntity>();
  301. rep.UnitOfWork = unitOfWork;
  302. //数据列表
  303. var dataList = GetData<RoleEntity>(isTenant, dbConfig.SyncDataPath);
  304. if (!(dataList?.Length > 0))
  305. {
  306. Console.WriteLine($"table: {tableName} import data []");
  307. return;
  308. }
  309. //查询
  310. var ids = dataList.Select(e => e.Id).ToList();
  311. var recordList = await rep.Where(a => ids.Contains(a.Id)).ToListAsync();
  312. //新增
  313. var recordIds = recordList.Select(a => a.Id).ToList();
  314. var insertDataList = dataList.Where(a => !recordIds.Contains(a.Id));
  315. if (insertDataList.Any())
  316. {
  317. await rep.InsertAsync(insertDataList);
  318. }
  319. //修改
  320. if (!dbConfig.SysUpdateData && recordList?.Count > 0)
  321. {
  322. var updateDataList = dataList.Where(a => recordIds.Contains(a.Id));
  323. await rep.UpdateAsync(updateDataList);
  324. }
  325. Console.WriteLine($"table: {tableName} sync data succeed");
  326. }
  327. catch (Exception ex)
  328. {
  329. var msg = $"table: {tableName} sync data failed.\n{ex.Message}";
  330. Console.WriteLine(msg);
  331. throw new Exception(msg);
  332. }
  333. }
  334. /// <summary>
  335. /// 初始化接口
  336. /// </summary>
  337. /// <param name="db"></param>
  338. /// <param name="unitOfWork"></param>
  339. /// <param name="dbConfig"></param>
  340. /// <returns></returns>
  341. private async Task InitApiAsync(IFreeSql db, IRepositoryUnitOfWork unitOfWork, DbConfig dbConfig)
  342. {
  343. var tableName = GetTableName<ApiEntity>();
  344. try
  345. {
  346. if (!IsSyncData(tableName, dbConfig))
  347. {
  348. return;
  349. }
  350. var rep = db.GetRepository<ApiEntity>();
  351. rep.UnitOfWork = unitOfWork;
  352. //数据列表
  353. var dataTree = GetData<ApiEntity>(path: dbConfig.SyncDataPath);
  354. var dataList = dataTree.ToList().ToPlainList((a) => a.Childs).ToArray();
  355. if (!(dataList?.Length > 0))
  356. {
  357. Console.WriteLine($"table: {tableName} import data []");
  358. return;
  359. }
  360. //查询
  361. var ids = dataList.Select(e => e.Id).ToList();
  362. var recordList = await rep.Where(a => ids.Contains(a.Id)).ToListAsync();
  363. //新增
  364. var recordIds = recordList.Select(a => a.Id).ToList();
  365. var insertDataList = dataList.Where(a => !recordIds.Contains(a.Id));
  366. if (insertDataList.Any())
  367. {
  368. await rep.InsertAsync(insertDataList);
  369. }
  370. //修改
  371. if (!dbConfig.SysUpdateData && recordList?.Count > 0)
  372. {
  373. var updateDataList = dataList.Where(a => recordIds.Contains(a.Id));
  374. await rep.UpdateAsync(updateDataList);
  375. }
  376. Console.WriteLine($"table: {tableName} sync data succeed");
  377. }
  378. catch (Exception ex)
  379. {
  380. var msg = $"table: {tableName} sync data failed.\n{ex.Message}";
  381. Console.WriteLine(msg);
  382. throw new Exception(msg);
  383. }
  384. }
  385. /// <summary>
  386. /// 初始化视图
  387. /// </summary>
  388. /// <param name="db"></param>
  389. /// <param name="unitOfWork"></param>
  390. /// <param name="dbConfig"></param>
  391. /// <returns></returns>
  392. private async Task InitViewAsync(IFreeSql db, IRepositoryUnitOfWork unitOfWork, DbConfig dbConfig)
  393. {
  394. var tableName = GetTableName<ViewEntity>();
  395. try
  396. {
  397. if (!IsSyncData(tableName, dbConfig))
  398. {
  399. return;
  400. }
  401. var rep = db.GetRepository<ViewEntity>();
  402. rep.UnitOfWork = unitOfWork;
  403. //数据列表
  404. var dataTree = GetData<ViewEntity>(path: dbConfig.SyncDataPath);
  405. var dataList = dataTree.ToList().ToPlainList((a) => a.Childs).ToArray();
  406. if (!(dataList?.Length > 0))
  407. {
  408. Console.WriteLine($"table: {tableName} import data []");
  409. return;
  410. }
  411. //查询
  412. var ids = dataList.Select(e => e.Id).ToList();
  413. var recordList = await rep.Where(a => ids.Contains(a.Id)).ToListAsync();
  414. //新增
  415. var recordIds = recordList.Select(a => a.Id).ToList();
  416. var insertDataList = dataList.Where(a => !recordIds.Contains(a.Id));
  417. if (insertDataList.Any())
  418. {
  419. await rep.InsertAsync(insertDataList);
  420. }
  421. //修改
  422. if (!dbConfig.SysUpdateData && recordList?.Count > 0)
  423. {
  424. var updateDataList = dataList.Where(a => recordIds.Contains(a.Id));
  425. await rep.UpdateAsync(updateDataList);
  426. }
  427. Console.WriteLine($"table: {tableName} sync data succeed");
  428. }
  429. catch (Exception ex)
  430. {
  431. var msg = $"table: {tableName} sync data failed.\n{ex.Message}";
  432. Console.WriteLine(msg);
  433. throw new Exception(msg);
  434. }
  435. }
  436. /// <summary>
  437. /// 初始化权限
  438. /// </summary>
  439. /// <param name="db"></param>
  440. /// <param name="unitOfWork"></param>
  441. /// <param name="dbConfig"></param>
  442. /// <returns></returns>
  443. private async Task InitPermissionAsync(IFreeSql db, IRepositoryUnitOfWork unitOfWork, DbConfig dbConfig)
  444. {
  445. var tableName = GetTableName<PermissionEntity>();
  446. try
  447. {
  448. if (!IsSyncData(tableName, dbConfig))
  449. {
  450. return;
  451. }
  452. var rep = db.GetRepository<PermissionEntity>();
  453. rep.UnitOfWork = unitOfWork;
  454. //数据列表
  455. var dataTree = GetData<PermissionEntity>(path: dbConfig.SyncDataPath);
  456. var dataList = dataTree.ToList().ToPlainList((a) => a.Childs).ToArray();
  457. if (!(dataList?.Length > 0))
  458. {
  459. Console.WriteLine($"table: {tableName} import data []");
  460. return;
  461. }
  462. //查询
  463. var ids = dataList.Select(e => e.Id).ToList();
  464. var recordList = await rep.Where(a => ids.Contains(a.Id)).ToListAsync();
  465. //新增
  466. var recordIds = recordList.Select(a => a.Id).ToList();
  467. var insertDataList = dataList.Where(a => !recordIds.Contains(a.Id));
  468. if (insertDataList.Any())
  469. {
  470. await rep.InsertAsync(insertDataList);
  471. }
  472. //修改
  473. if (!dbConfig.SysUpdateData && recordList?.Count > 0)
  474. {
  475. var updateDataList = dataList.Where(a => recordIds.Contains(a.Id));
  476. await rep.UpdateAsync(updateDataList);
  477. }
  478. Console.WriteLine($"table: {tableName} sync data succeed");
  479. }
  480. catch (Exception ex)
  481. {
  482. var msg = $"table: {tableName} sync data failed.\n{ex.Message}";
  483. Console.WriteLine(msg);
  484. throw new Exception(msg);
  485. }
  486. }
  487. /// <summary>
  488. /// 初始化权限接口
  489. /// </summary>
  490. /// <param name="db"></param>
  491. /// <param name="unitOfWork"></param>
  492. /// <param name="dbConfig"></param>
  493. /// <returns></returns>
  494. private async Task InitPermissionApiAsync(IFreeSql db, IRepositoryUnitOfWork unitOfWork, DbConfig dbConfig)
  495. {
  496. var tableName = GetTableName<PermissionApiEntity>();
  497. try
  498. {
  499. if (!IsSyncData(tableName, dbConfig))
  500. {
  501. return;
  502. }
  503. var rep = db.GetRepository<PermissionApiEntity>();
  504. rep.UnitOfWork = unitOfWork;
  505. //数据列表
  506. var dataList = GetData<PermissionApiEntity>(path: dbConfig.SyncDataPath);
  507. if (!(dataList?.Length > 0))
  508. {
  509. Console.WriteLine($"table: {tableName} import data []");
  510. return;
  511. }
  512. //查询
  513. var recordList = await rep.Where(a => rep.Select.WithMemory(dataList).Where(b => b.PermissionId == a.PermissionId && b.ApiId == a.ApiId).Any()).ToListAsync();
  514. //新增
  515. var insertDataList = dataList.Where(a => !(recordList.Where(b => a.PermissionId == b.PermissionId && a.ApiId == b.ApiId).Any())).ToList();
  516. if (insertDataList.Any())
  517. {
  518. await rep.InsertAsync(insertDataList);
  519. }
  520. Console.WriteLine($"table: {tableName} sync data succeed");
  521. }
  522. catch (Exception ex)
  523. {
  524. var msg = $"table: {tableName} sync data failed.\n{ex.Message}";
  525. Console.WriteLine(msg);
  526. throw new Exception(msg);
  527. }
  528. }
  529. /// <summary>
  530. /// 用户角色记录
  531. /// </summary>
  532. /// <param name="UserId"></param>
  533. /// <param name="RoleId"></param>
  534. record UserRoleRecord(long UserId, long RoleId);
  535. /// <summary>
  536. /// 初始化用户角色
  537. /// </summary>
  538. /// <param name="db"></param>
  539. /// <param name="unitOfWork"></param>
  540. /// <param name="dbConfig"></param>
  541. /// <returns></returns>
  542. private async Task InitUserRoleAsync(IFreeSql db, IRepositoryUnitOfWork unitOfWork, DbConfig dbConfig)
  543. {
  544. var tableName = GetTableName<UserRoleEntity>();
  545. try
  546. {
  547. if (!IsSyncData(tableName, dbConfig))
  548. {
  549. return;
  550. }
  551. var rep = db.GetRepository<UserRoleEntity>();
  552. rep.UnitOfWork = unitOfWork;
  553. //数据列表
  554. var dataList = GetData<UserRoleEntity>(path: dbConfig.SyncDataPath);
  555. if (!(dataList?.Length > 0))
  556. {
  557. Console.WriteLine($"table: {tableName} import data []");
  558. return;
  559. }
  560. //查询
  561. var userRoleRecordList = dataList.Adapt<List<UserRoleRecord>>();
  562. var recordList = await rep.Where(a => rep.Orm.Select<UserRoleRecord>().WithMemory(userRoleRecordList).Where(b => b.UserId == a.UserId && b.RoleId == a.RoleId).Any()).ToListAsync();
  563. //新增
  564. var insertDataList = dataList.Where(a => !(recordList.Where(b => a.UserId == b.UserId && a.RoleId == b.RoleId).Any())).ToList();
  565. if (insertDataList.Any())
  566. {
  567. await rep.InsertAsync(insertDataList);
  568. }
  569. Console.WriteLine($"table: {tableName} sync data succeed");
  570. }
  571. catch (Exception ex)
  572. {
  573. var msg = $"table: {tableName} sync data failed.\n{ex.Message}";
  574. Console.WriteLine(msg);
  575. throw new Exception(msg);
  576. }
  577. }
  578. /// <summary>
  579. /// 用户部门记录
  580. /// </summary>
  581. /// <param name="UserId"></param>
  582. /// <param name="OrgId"></param>
  583. record UserOrgRecord(long UserId, long OrgId);
  584. /// <summary>
  585. /// 初始化用户部门
  586. /// </summary>
  587. /// <param name="db"></param>
  588. /// <param name="unitOfWork"></param>
  589. /// <param name="dbConfig"></param>
  590. /// <returns></returns>
  591. private async Task InitUserOrgAsync(IFreeSql db, IRepositoryUnitOfWork unitOfWork, DbConfig dbConfig)
  592. {
  593. var tableName = GetTableName<UserOrgEntity>();
  594. try
  595. {
  596. if (!IsSyncData(tableName, dbConfig))
  597. {
  598. return;
  599. }
  600. var rep = db.GetRepository<UserOrgEntity>();
  601. rep.UnitOfWork = unitOfWork;
  602. //数据列表
  603. var dataList = GetData<UserOrgEntity>(path: dbConfig.SyncDataPath);
  604. if (!(dataList?.Length > 0))
  605. {
  606. Console.WriteLine($"table: {tableName} import data []");
  607. return;
  608. }
  609. //查询
  610. var userOrgRecordList = dataList.Adapt<List<UserOrgRecord>>();
  611. var recordList = await rep.Where(a => rep.Orm.Select<UserOrgRecord>().WithMemory(userOrgRecordList).Where(b => b.UserId == a.UserId && b.OrgId == a.OrgId).Any()).ToListAsync();
  612. //新增
  613. var insertDataList = dataList.Where(a => !(recordList.Where(b => a.UserId == b.UserId && a.OrgId == b.OrgId).Any())).ToList();
  614. if (insertDataList.Any())
  615. {
  616. await rep.InsertAsync(insertDataList);
  617. }
  618. Console.WriteLine($"table: {tableName} sync data succeed");
  619. }
  620. catch (Exception ex)
  621. {
  622. var msg = $"table: {tableName} sync data failed.\n{ex.Message}";
  623. Console.WriteLine(msg);
  624. throw new Exception(msg);
  625. }
  626. }
  627. /// <summary>
  628. /// 角色权限记录
  629. /// </summary>
  630. /// <param name="RoleId"></param>
  631. /// <param name="PermissionId"></param>
  632. record RolePermissionRecord(long RoleId, long PermissionId);
  633. /// <summary>
  634. /// 初始化角色权限
  635. /// </summary>
  636. /// <param name="db"></param>
  637. /// <param name="unitOfWork"></param>
  638. /// <param name="dbConfig"></param>
  639. /// <returns></returns>
  640. private async Task InitRolePermissionAsync(IFreeSql db, IRepositoryUnitOfWork unitOfWork, DbConfig dbConfig)
  641. {
  642. var tableName = GetTableName<RolePermissionEntity>();
  643. try
  644. {
  645. if (!IsSyncData(tableName, dbConfig))
  646. {
  647. return;
  648. }
  649. var rep = db.GetRepository<RolePermissionEntity>();
  650. rep.UnitOfWork = unitOfWork;
  651. //数据列表
  652. var dataList = GetData<RolePermissionEntity>(path: dbConfig.SyncDataPath);
  653. if (!(dataList?.Length > 0))
  654. {
  655. Console.WriteLine($"table: {tableName} import data []");
  656. return;
  657. }
  658. //查询
  659. var rolePermissionRecordList = dataList.Adapt<List<RolePermissionRecord>>();
  660. var recordList = await rep.Where(a => rep.Orm.Select<RolePermissionRecord>().WithMemory(rolePermissionRecordList).Where(b => b.RoleId == a.RoleId && b.PermissionId == a.PermissionId).Any()).ToListAsync();
  661. //新增
  662. var insertDataList = dataList.Where(a => !(recordList.Where(b => a.RoleId == b.RoleId && a.PermissionId == b.PermissionId).Any())).ToList();
  663. if (insertDataList.Any())
  664. {
  665. await rep.InsertAsync(insertDataList);
  666. }
  667. Console.WriteLine($"table: {tableName} sync data succeed");
  668. }
  669. catch (Exception ex)
  670. {
  671. var msg = $"table: {tableName} sync data failed.\n{ex.Message}";
  672. Console.WriteLine(msg);
  673. throw new Exception(msg);
  674. }
  675. }
  676. /// <summary>
  677. /// 初始化租户
  678. /// </summary>
  679. /// <param name="db"></param>
  680. /// <param name="unitOfWork"></param>
  681. /// <param name="dbConfig"></param>
  682. /// <returns></returns>
  683. private async Task InitTenantAsync(IFreeSql db, IRepositoryUnitOfWork unitOfWork, DbConfig dbConfig)
  684. {
  685. var tableName = GetTableName<TenantEntity>();
  686. try
  687. {
  688. if (!IsSyncData(tableName, dbConfig))
  689. {
  690. return;
  691. }
  692. var rep = db.GetRepository<TenantEntity>();
  693. rep.UnitOfWork = unitOfWork;
  694. //数据列表
  695. var dataList = GetData<TenantEntity>(path: dbConfig.SyncDataPath);
  696. if (!(dataList?.Length > 0))
  697. {
  698. Console.WriteLine($"table: {tableName} import data []");
  699. return;
  700. }
  701. //查询
  702. var ids = dataList.Select(e => e.Id).ToList();
  703. var recordList = await rep.Where(a => ids.Contains(a.Id)).ToListAsync();
  704. //新增
  705. var recordIds = recordList.Select(a => a.Id).ToList();
  706. var insertDataList = dataList.Where(a => !recordIds.Contains(a.Id));
  707. if (insertDataList.Any())
  708. {
  709. await rep.InsertAsync(insertDataList);
  710. }
  711. //修改
  712. if (!dbConfig.SysUpdateData && recordList?.Count > 0)
  713. {
  714. var updateDataList = dataList.Where(a => recordIds.Contains(a.Id));
  715. await rep.UpdateAsync(updateDataList);
  716. }
  717. Console.WriteLine($"table: {tableName} sync data succeed");
  718. }
  719. catch (Exception ex)
  720. {
  721. var msg = $"table: {tableName} sync data failed.\n{ex.Message}";
  722. Console.WriteLine(msg);
  723. throw new Exception(msg);
  724. }
  725. }
  726. /// <summary>
  727. /// 租户权限记录
  728. /// </summary>
  729. /// <param name="TenantId"></param>
  730. /// <param name="PermissionId"></param>
  731. record TenantPermissionRecord(long TenantId, long PermissionId);
  732. /// <summary>
  733. /// 初始化租户权限
  734. /// </summary>
  735. /// <param name="db"></param>
  736. /// <param name="unitOfWork"></param>
  737. /// <param name="dbConfig"></param>
  738. /// <returns></returns>
  739. private async Task InitTenantPermissionAsync(IFreeSql db, IRepositoryUnitOfWork unitOfWork, DbConfig dbConfig)
  740. {
  741. var tableName = GetTableName<TenantPermissionEntity>();
  742. try
  743. {
  744. if (!IsSyncData(tableName, dbConfig))
  745. {
  746. return;
  747. }
  748. var rep = db.GetRepository<TenantPermissionEntity>();
  749. rep.UnitOfWork = unitOfWork;
  750. //数据列表
  751. var dataList = GetData<TenantPermissionEntity>(path: dbConfig.SyncDataPath);
  752. if (!(dataList?.Length > 0))
  753. {
  754. Console.WriteLine($"table: {tableName} import data []");
  755. return;
  756. }
  757. //查询
  758. var tenantPermissionRecordList = dataList.Adapt<List<TenantPermissionRecord>>();
  759. var recordList = await rep.Where(a => rep.Orm.Select<TenantPermissionRecord>().WithMemory(tenantPermissionRecordList).Where(b => b.TenantId == a.TenantId && b.PermissionId == a.PermissionId).Any()).ToListAsync();
  760. //新增
  761. var insertDataList = dataList.Where(a => !(recordList.Where(b => a.TenantId == b.TenantId && a.PermissionId == b.PermissionId).Any())).ToList();
  762. if (insertDataList.Any())
  763. {
  764. await rep.InsertAsync(insertDataList);
  765. }
  766. Console.WriteLine($"table: {tableName} sync data succeed");
  767. }
  768. catch (Exception ex)
  769. {
  770. var msg = $"table: {tableName} sync data failed.\n{ex.Message}";
  771. Console.WriteLine(msg);
  772. throw new Exception(msg);
  773. }
  774. }
  775. /// <summary>
  776. /// 同步数据
  777. /// </summary>
  778. /// <param name="db"></param>
  779. /// <param name="dbConfig"></param>
  780. /// <param name="appConfig"></param>
  781. /// <returns></returns>
  782. public virtual async Task SyncDataAsync(IFreeSql db, DbConfig dbConfig = null, AppConfig appConfig = null)
  783. {
  784. using var unitOfWork = db.CreateUnitOfWork();
  785. try
  786. {
  787. var isTenant = appConfig.Tenant;
  788. await InitDictTypeAsync(db, unitOfWork, dbConfig);
  789. await InitDictAsync(db, unitOfWork, dbConfig);
  790. await InitUserAsync(db, unitOfWork, dbConfig, isTenant);
  791. await InitUserStaffAsync(db, unitOfWork, dbConfig, isTenant);
  792. await InitOrgAsync(db, unitOfWork, dbConfig, isTenant);
  793. await InitRoleAsync(db, unitOfWork, dbConfig, isTenant);
  794. await InitApiAsync(db, unitOfWork, dbConfig);
  795. await InitViewAsync(db, unitOfWork, dbConfig);
  796. await InitPermissionAsync(db, unitOfWork, dbConfig);
  797. await InitPermissionApiAsync(db, unitOfWork, dbConfig);
  798. await InitUserRoleAsync(db, unitOfWork, dbConfig);
  799. await InitUserOrgAsync(db, unitOfWork, dbConfig);
  800. await InitRolePermissionAsync(db, unitOfWork, dbConfig);
  801. await InitTenantAsync(db, unitOfWork, dbConfig);
  802. await InitTenantPermissionAsync(db, unitOfWork, dbConfig);
  803. unitOfWork.Commit();
  804. }
  805. catch (Exception)
  806. {
  807. unitOfWork.Rollback();
  808. throw;
  809. }
  810. }
  811. }