优化种子数据

This commit is contained in:
不做码农
2023-06-11 13:16:22 +08:00
parent df99da04fb
commit 9be0036b4e
2 changed files with 84 additions and 66 deletions

View File

@@ -5,7 +5,6 @@ using Infrastructure.Extensions;
using Infrastructure.Model;
using Microsoft.AspNetCore.Mvc;
using Microsoft.Extensions.Options;
using MiniExcelLibs;
using Newtonsoft.Json;
using ZR.Admin.WebApi.Extensions;
using ZR.Admin.WebApi.Filters;
@@ -179,85 +178,30 @@ namespace ZR.Admin.WebApi.Controllers
/// <summary>
/// 初始化种子数据
/// </summary>
/// <param name="clean">是否清空数据</param>
/// <returns></returns>
[HttpGet]
[ApiExplorerSettings(IgnoreApi = true)]
[ActionPermissionFilter(Permission = "common")]
[Log(BusinessType = BusinessType.INSERT, Title = "初始化数据")]
public IActionResult InitSeedData()
public IActionResult InitSeedData(bool clean = false)
{
if (!WebHostEnvironment.IsDevelopment())
{
return ToResponse(ResultCode.FAIL, "导入数据失败");
return ToResponse(ResultCode.CUSTOM_ERROR, "导入数据失败");
}
var path = Path.Combine(WebHostEnvironment.WebRootPath, "data.xlsx");
//var sheetNames = MiniExcel.GetSheetNames(path);
SeedDataService seedDataService = new();
var sysUser = MiniExcel.Query<SysUser>(path, sheetName: "user").ToList();
var result1 = seedDataService.InitUserData(sysUser);
var sysPost = MiniExcel.Query<SysPost>(path, sheetName: "post").ToList();
var result2 = seedDataService.InitPostData(sysPost);
var sysRole = MiniExcel.Query<SysRole>(path, sheetName: "role").ToList();
var result3 = seedDataService.InitRoleData(sysRole);
var sysUserRole = MiniExcel.Query<SysUserRole>(path, sheetName: "user_role").ToList();
var result4 = seedDataService.InitUserRoleData(sysUserRole);
var sysMenu = MiniExcel.Query<SysMenu>(path, sheetName: "menu").ToList();
var result5 = seedDataService.InitMenuData(sysMenu);
var sysConfig = MiniExcel.Query<SysConfig>(path, sheetName: "config").ToList();
var result6 = seedDataService.InitConfigData(sysConfig);
var sysRoleMenu = MiniExcel.Query<SysRoleMenu>(path, sheetName: "role_menu").ToList();
var result7 = seedDataService.InitRoleMenuData(sysRoleMenu);
var sysDict = MiniExcel.Query<SysDictType>(path, sheetName: "dict_type").ToList();
var result8 = seedDataService.InitDictType(sysDict);
var sysDictData = MiniExcel.Query<SysDictData>(path, sheetName: "dict_data").ToList();
var result9 = seedDataService.InitDictData(sysDictData);
var sysDept = MiniExcel.Query<SysDept>(path, sheetName: "dept").ToList();
var result10 = seedDataService.InitDeptData(sysDept);
var sysArticleCategory = MiniExcel.Query<ArticleCategory>(path, sheetName: "article_category").ToList();
var result11 = seedDataService.InitArticleCategoryData(sysArticleCategory);
var sysTask = MiniExcel.Query<SysTasks>(path, sheetName: "task").ToList();
var result12 = seedDataService.InitTaskData(sysTask);
var result = seedDataService.InitSeedData(path, clean);
Console.ForegroundColor = ConsoleColor.Red;
Console.WriteLine(result1.Item1);
Console.WriteLine(result2.Item1);
Console.WriteLine(result3.Item1);
Console.WriteLine(result4.Item1);
Console.WriteLine(result5.Item1);
Console.WriteLine(result6.Item1);
Console.WriteLine(result7.Item1);
Console.WriteLine(result8.Item1);
Console.WriteLine(result9.Item1);
Console.WriteLine(result10.Item1);
Console.WriteLine(result11.Item1);
Console.WriteLine(result12.Item1);
foreach (var item in result)
{
Console.WriteLine(item);
}
Console.ForegroundColor = ConsoleColor.White;
return SUCCESS(new
{
result1 = result1.Item1,
result2 = result2.Item1,
result3 = result3.Item1,
result4 = result4.Item1,
result5 = result5.Item1,
result6 = result6.Item1,
result7 = result7.Item1,
result8 = result8.Item1,
result9 = result9.Item1,
result10 = result10.Item1,
result11 = result11.Item1,
result12 = result12.Item1
result
});
}
}