using Aop.Api.Domain;
|
using EFCore.BulkExtensions;
|
using Furion.DatabaseAccessor;
|
using Furion.DatabaseAccessor.Extensions;
|
using MediatR;
|
using Microsoft.EntityFrameworkCore;
|
using System;
|
using System.Collections.Generic;
|
using System.Linq;
|
using System.Text;
|
using System.Threading.Tasks;
|
|
namespace FlexJobApi.Core
|
{
|
/// <summary>
|
/// 同步数据库
|
/// </summary>
|
public class SyncDatabaseCommandHandler() :
|
IRequestHandler<SyncDatabaseCommand, int>
|
{
|
|
/// <summary>
|
/// 同步数据库
|
/// </summary>
|
/// <param name="request"></param>
|
/// <param name="cancellationToken"></param>
|
/// <returns></returns>
|
public async Task<int> Handle(SyncDatabaseCommand request, CancellationToken cancellationToken)
|
{
|
var count = 0;
|
count += await SyncData<DictionaryCategory>();
|
count += await SyncTreeData<DictionaryData>();
|
//count += await SyncData<ElectronSignSetting>();
|
//count += await SyncData<Enterprise>();
|
//count += await SyncData<EnterpriseAuth>();
|
//count += await SyncData<EnterpriseCost>();
|
//count += await SyncData<EnterpriseElectronSignSetting>();
|
//count += await SyncData<EnterpriseWallet>();
|
//count += await SyncData<EnterpriseWalletExpandindirectOrder>();
|
//count += await SyncData<EnterpriseWalletExpandindirectOrderFile>();
|
//count += await SyncData<User>();
|
//count += await SyncData<UserAuth>();
|
count += await SyncTreeData<Menu>();
|
count += await SyncData<Resource>();
|
count += await SyncData<Role>();
|
count += await SyncData<RoleMenu>();
|
count += await SyncData<RoleResource>();
|
//count += await SyncData<UserRole>();
|
//count += await SyncData<ContractTemplate>();
|
//count += await SyncData<ContractTemplateValue>();
|
return count;
|
}
|
|
private async Task<int> SyncData<TEntity>()
|
where TEntity : CommonEntity, new()
|
{
|
var count = 0;
|
var repTarget = Db.GetRepository<TEntity, SyncTargetDbContextLocator>();
|
var repSource = Db.GetRepository<TEntity, MasterDbContextLocator>();
|
|
var sources = await repSource.AsQueryable().AsNoTracking()
|
.OrderBy(it => it.Sort).ThenBy(it => it.CreatedTime)
|
.ToListAsync();
|
var targets = await repTarget.AsQueryable().AsNoTracking()
|
.OrderBy(it => it.Sort).ThenBy(it => it.CreatedTime)
|
.ToListAsync();
|
|
var adds = sources.Where(s => !targets.Any(t => s.Id == t.Id)).ToList();
|
if (adds.IsNotNull())
|
{
|
await repTarget.Context.BulkInsertAsync(adds);
|
count += adds.Count;
|
}
|
|
var updates = sources.Where(s => targets.Any(t => s.Id == t.Id)).ToList();
|
if (updates.IsNotNull())
|
{
|
await repTarget.Context.BulkUpdateAsync(updates);
|
count += updates.Count;
|
}
|
|
var deletes = targets.Where(s => sources.Any(t => s.Id == t.Id)).ToList();
|
if (deletes.IsNotNull())
|
{
|
await repTarget.Context.BulkUpdateAsync(deletes);
|
count += deletes.Count;
|
}
|
|
return count;
|
}
|
|
private async Task<int> SyncTreeData<TEntity>()
|
where TEntity : CommonEntity, ITreeData<TEntity>, new()
|
{
|
var count = 0;
|
var repTarget = Db.GetRepository<TEntity, SyncTargetDbContextLocator>();
|
var repSource = Db.GetRepository<TEntity, MasterDbContextLocator>();
|
|
var sourceData = await repSource.AsQueryable().AsNoTracking()
|
.OrderBy(it => it.Sort).ThenBy(it => it.CreatedTime)
|
.ToListAsync();
|
var targetData = await repTarget.AsQueryable().AsNoTracking()
|
.OrderBy(it => it.Sort).ThenBy(it => it.CreatedTime)
|
.ToListAsync();
|
|
var sources = new List<TEntity>();
|
var sourceParents = sourceData.Where(it => it.ParentId == null).ToList();
|
LoopTreeData(sources, sourceData, sourceParents);
|
|
var targets = new List<TEntity>();
|
var targetParents = targetData.Where(it => it.ParentId == null).ToList();
|
LoopTreeData(targets, targetData, targetParents);
|
|
var adds = sources.Where(s => !targets.Any(t => s.Id == t.Id)).ToList();
|
if (adds.IsNotNull())
|
{
|
await repTarget.Context.BulkInsertAsync(adds);
|
count += adds.Count;
|
}
|
|
var updates = sources.Where(s => targets.Any(t => s.Id == t.Id)).ToList();
|
if (updates.IsNotNull())
|
{
|
await repTarget.Context.BulkUpdateAsync(updates);
|
count += updates.Count;
|
}
|
|
var deletes = targets.Where(s => sources.Any(t => s.Id == t.Id)).ToList();
|
if (deletes.IsNotNull())
|
{
|
await repTarget.Context.BulkUpdateAsync(deletes);
|
count += deletes.Count;
|
}
|
|
return count;
|
}
|
|
private void LoopTreeData<TEntity>(List<TEntity> entities, List<TEntity> all, List<TEntity> parents)
|
where TEntity : CommonEntity, ITreeData<TEntity>, new()
|
{
|
if (parents.IsNotNull())
|
{
|
foreach (var parent in parents)
|
{
|
entities.Add(parent);
|
var children = all.Where(it => it.ParentId == parent.Id).ToList();
|
LoopTreeData(entities, all, children);
|
}
|
}
|
}
|
}
|
}
|
