using New_College.Common.Helper;
using New_College.IRepository.UnitOfWork;
using New_College.IServices;
using New_College.Model;
using New_College.Model.Models;
using Microsoft.AspNetCore.Hosting;
using Microsoft.AspNetCore.Mvc;
using Microsoft.Extensions.Hosting;
using Newtonsoft.Json;
using System;
using System.IO;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
namespace New_College.Controllers
{
[Route("api/[controller]/[action]")]
[ApiController]
//[Authorize(Permissions.Name)]
public class MigrateController : ControllerBase
{
private readonly IUnitOfWork _unitOfWork;
private readonly IRoleModulePermissionServices _roleModulePermissionServices;
private readonly IUserRoleServices _userRoleServices;
private readonly IRoleServices _roleServices;
private readonly IPermissionServices _permissionServices;
private readonly IModuleServices _moduleServices;
private readonly IWebHostEnvironment _env;
public MigrateController(IUnitOfWork unitOfWork,
IRoleModulePermissionServices roleModulePermissionServices,
IUserRoleServices userRoleServices,
IRoleServices roleServices,
IPermissionServices permissionServices,
IModuleServices moduleServices,
IWebHostEnvironment env)
{
_unitOfWork = unitOfWork;
_roleModulePermissionServices = roleModulePermissionServices;
_userRoleServices = userRoleServices;
_roleServices = roleServices;
_permissionServices = permissionServices;
_moduleServices = moduleServices;
_env = env;
}
///
/// 获取权限部分Map数据(从库)
/// 迁移到新库(主库)
///
///
[HttpGet]
public async Task> DataMigrateFromOld2New()
{
var data = new MessageModel() { success = true, msg = "" };
if (_env.IsDevelopment())
{
try
{
// 获取权限集合数据
var rmps = await _roleModulePermissionServices.GetRMPMaps();
// 当然,你可以做个where查询
//rmps = rmps.Where(d => d.ModuleId > 88).ToList();
// 开启事务,保证数据一致性
_unitOfWork.BeginTran();
var rid = 0;
var pid = 0;
var mid = 0;
var rpmid = 0;
// 注意信息的完整性,不要重复添加,确保主库没有要添加的数据
foreach (var item in rmps)
{
// 角色信息,防止重复添加,做了判断
if (item.Role != null)
{
var isExit = (await _roleServices.Query(d => d.Name == item.Role.Name && d.IsDeleted == false)).FirstOrDefault();
if (isExit == null)
{
rid = await _roleServices.Add(item.Role);
Console.WriteLine($"Role Added:{item.Role.Name}");
}
else
{
rid = isExit.Id;
}
}
// 菜单
if (item.Permission != null)
{
pid = await _permissionServices.Add(item.Permission);
Console.WriteLine($"Permission Added:{item.Permission.Name}");
}
// 接口
if (item.Module != null)
{
mid = await _moduleServices.Add(item.Module);
Console.WriteLine($"Module Added:{item.Module.LinkUrl}");
}
// 关系
if (rid > 0 && pid > 0 && mid > 0)
{
rpmid = await _roleModulePermissionServices.Add(new RoleModulePermission()
{
IsDeleted = false,
CreateTime = DateTime.Now,
ModifyTime = DateTime.Now,
ModuleId = mid,
PermissionId = pid,
RoleId = rid,
});
Console.WriteLine($"RMP Added:{rpmid}");
}
}
_unitOfWork.CommitTran();
data.success = true;
data.msg = "导入成功!";
}
catch (Exception)
{
_unitOfWork.RollbackTran();
}
}
else
{
data.success = false;
data.msg = "当前不处于开发模式,代码生成不可用!";
}
return data;
}
///
/// 权限数据库导出tsv
///
///
[HttpGet]
public async Task> SaveData2TsvAsync()
{
var data = new MessageModel() { success = true, msg = "" };
if (_env.IsDevelopment())
{
try
{
// 取出数据,序列化,自己可以处理判空
var rolesJson = JsonConvert.SerializeObject(await _roleServices.Query(d => d.IsDeleted == false));
FileHelper.WriteFile(Path.Combine(_env.WebRootPath, "BlogCore.Data.json", "Role_New.tsv"), rolesJson, Encoding.UTF8);
var permissionsJson = JsonConvert.SerializeObject(await _permissionServices.Query(d => d.IsDeleted == false));
FileHelper.WriteFile(Path.Combine(_env.WebRootPath, "BlogCore.Data.json", "Permission_New.tsv"), permissionsJson, Encoding.UTF8);
var modulesJson = JsonConvert.SerializeObject(await _moduleServices.Query(d => d.IsDeleted == false));
FileHelper.WriteFile(Path.Combine(_env.WebRootPath, "BlogCore.Data.json", "Modules_New.tsv"), modulesJson, Encoding.UTF8);
var rmpsJson = JsonConvert.SerializeObject(await _roleModulePermissionServices.Query(d => d.IsDeleted == false));
FileHelper.WriteFile(Path.Combine(_env.WebRootPath, "BlogCore.Data.json", "RoleModulePermission_New.tsv"), rmpsJson, Encoding.UTF8);
}
catch (Exception)
{
}
data.success = true;
data.msg = "生成成功!";
}
else
{
data.success = false;
data.msg = "当前不处于开发模式,代码生成不可用!";
}
return data;
}
}
}