You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
388 lines
13 KiB
388 lines
13 KiB
using System.Data;
|
|
using System.Drawing.Printing;
|
|
using System.Linq.Expressions;
|
|
using System.Text;
|
|
using System.Text.Json;
|
|
using System.Text.Json.Serialization;
|
|
using Azure.Core;
|
|
using Dapper;
|
|
using Magicodes.ExporterAndImporter.Core.Extension;
|
|
using Magicodes.ExporterAndImporter.Excel;
|
|
using Microsoft.AspNetCore.Http;
|
|
using Microsoft.AspNetCore.Mvc;
|
|
using Microsoft.EntityFrameworkCore;
|
|
using Microsoft.Extensions.DependencyInjection;
|
|
using TaskManager.Contracts.Dtos;
|
|
using TaskManager.Entity;
|
|
using TaskManager.EntityFramework;
|
|
using TaskManager.EntityFramework.Repository;
|
|
|
|
namespace TaskManager.Controllers
|
|
{
|
|
public class CheryRecurringJobInputPageController<T,OUTPUT> : RecurringJobBaseController where T : BaseEntity ,new()
|
|
{
|
|
protected readonly IRepository<T> _repository;
|
|
public CheryRecurringJobInputPageController(HttpClient httpClient, JobDbContext jobDbContext, LogController log, IRepository<T> repository) : base(httpClient, jobDbContext, log)
|
|
{
|
|
_repository = repository;
|
|
}
|
|
|
|
protected override async Task DoExecutingAsync(string url, string path, string takName, string client)
|
|
{
|
|
Url = url;
|
|
Path = path;
|
|
TaskName = takName;
|
|
Client = client;
|
|
await SyncTaskSubTable(TaskName,Client);
|
|
|
|
}
|
|
private async Task<QRReturnInfo> PostPageAsync(PagedRequest<T> t)
|
|
{
|
|
try
|
|
{
|
|
var inputjson = JsonSerializer.Serialize(t,
|
|
new JsonSerializerOptions
|
|
{
|
|
PropertyNamingPolicy = JsonNamingPolicy.CamelCase,
|
|
WriteIndented = true // 可选,用于格式化输出
|
|
}
|
|
);
|
|
inputjson = RemoveWhitespace(inputjson);
|
|
var content = await Post(Url, Path, inputjson);
|
|
if (!string.IsNullOrEmpty(content))
|
|
{
|
|
|
|
var options = new JsonSerializerOptions
|
|
{
|
|
PropertyNamingPolicy = JsonNamingPolicy.CamelCase,
|
|
Converters = {
|
|
new JsonStringEnumConverter(), // 枚举转换
|
|
new CustomDateTimeConverter("yyyy-MM-dd HH:mm:ss") // 日期转换
|
|
}
|
|
};
|
|
return JsonSerializer.Deserialize<QRReturnInfo>(content, options);
|
|
}
|
|
else
|
|
{
|
|
await _logger.AddError($"调用接口无返回值{Url}", TaskName);
|
|
return null;
|
|
}
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
await _logger.AddError($"调用接口无返回值错误{ex.Message}", TaskName);
|
|
return null;
|
|
}
|
|
|
|
}
|
|
|
|
|
|
public static string GenerateRandomStringWith8EG()
|
|
{
|
|
const string chars = "ABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789";
|
|
var random = new Random();
|
|
var sb = new StringBuilder(32);
|
|
|
|
// 随机生成 "8EG" 的插入位置 (0-29)
|
|
int position = random.Next(0, 30);
|
|
|
|
// 生成前半部分随机字符串
|
|
for (int i = 0; i < position; i++)
|
|
{
|
|
sb.Append(chars[random.Next(chars.Length)]);
|
|
}
|
|
|
|
// 插入 "8EG"
|
|
sb.Append("8EG");
|
|
|
|
// 生成后半部分随机字符串,补足32位
|
|
int remainingLength = 32 - sb.Length;
|
|
for (int i = 0; i < remainingLength; i++)
|
|
{
|
|
sb.Append(chars[random.Next(chars.Length)]);
|
|
}
|
|
|
|
return sb.ToString();
|
|
}
|
|
|
|
|
|
private async Task SyncTaskSubTable(string taskName, string client)
|
|
{
|
|
|
|
if (string.IsNullOrEmpty(taskName) || string.IsNullOrEmpty(client))
|
|
{
|
|
await _logger.AddError("任务名称或客户端不能为空",taskName);
|
|
return;
|
|
}
|
|
|
|
var sublist = _jobDbContext.TaskSub.Where(p => p.TaskName == taskName && p.WriteState == false && p.Subscriber == client).ToList();
|
|
int pageSize = 1000;
|
|
if (!sublist.Any())
|
|
{
|
|
foreach (var sub in sublist)
|
|
{
|
|
string querystr = $"select * from {sub.TableName} where TaskId='{sub.TaskId}' and WriteState=0 order by uid";//任务表
|
|
|
|
var entites = _jobDbContext.Database.GetDbConnection().Query<T>(querystr);//明细表
|
|
if (entites.Any())
|
|
{
|
|
var total = entites.Count();
|
|
int totalPages = (int)Math.Ceiling((double)total / pageSize);
|
|
|
|
for (int i = 1; i <= totalPages; i++)
|
|
{
|
|
|
|
var records = entites.Skip((i - 1) * pageSize)
|
|
.Take(pageSize).ToList();
|
|
|
|
PagedRequest<T> pagedRequest = new PagedRequest<T>()
|
|
{
|
|
batchNo = GenerateRandomStringWith8EG(),
|
|
total = entites.Count(),
|
|
pageSize = pageSize,
|
|
list = records,
|
|
pageNum = i
|
|
};
|
|
foreach (var itm in records)
|
|
{
|
|
itm.WriteState = true;
|
|
}
|
|
var result = await PostPageAsync(pagedRequest);
|
|
if (result.code == "200")
|
|
{
|
|
_jobDbContext.BulkUpdate(records);
|
|
sub.SyncedPageCount = i;
|
|
|
|
if (result.data.totalGet == total)
|
|
{
|
|
sub.WriteState = true;
|
|
_jobDbContext.Set<TaskSub>().Update(sub);
|
|
_jobDbContext.SaveChanges();
|
|
}
|
|
}
|
|
else
|
|
{
|
|
await _logger.AddError($"第 {i} 页数据保存失败,请检查数据。", TaskName);
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
|
|
[HttpGet]
|
|
public async Task<ActionResult<IEnumerable<T>>> GetAll()
|
|
{
|
|
return await _repository.GetAllAsync() as List<T>;
|
|
}
|
|
|
|
[HttpGet("{id}")]
|
|
public async Task<ActionResult<T>> GetById(int id)
|
|
{
|
|
var entity = await _repository.GetByIdAsync(id);
|
|
if (entity == null) return NotFound();
|
|
return entity;
|
|
}
|
|
|
|
[HttpPost]
|
|
public async Task<ActionResult<T>> Create(T entity)
|
|
{
|
|
entity.CreationTime = DateTime.Now;
|
|
var createdEntity = await _repository.AddAsync(entity);
|
|
return new JsonResult(new { Code = 200, Message = "创建成功!" }); ;
|
|
}
|
|
|
|
[HttpPut("{id}")]
|
|
public async Task<IActionResult> Update(T entity)
|
|
{
|
|
var _first = await _repository.GetByIdAsync(entity.UId);
|
|
if (_first == null)
|
|
{
|
|
return new JsonResult(new { Code = 400, Message = "修改失败!" });
|
|
}
|
|
|
|
|
|
await _repository.UpdateAsync(entity);
|
|
return new JsonResult(new { Code = 200, Message = "修改成功!" });
|
|
}
|
|
|
|
[HttpDelete("{id}")]
|
|
public async Task<IActionResult> Delete(int id)
|
|
{
|
|
await _repository.DeleteAsync(id);
|
|
return new JsonResult(new { Code = 200, Message = "删除成功!" }); ;
|
|
}
|
|
[HttpGet]
|
|
public async Task<ActionResult> GetPaged(
|
|
[FromQuery] int pageNumber = 1,
|
|
[FromQuery] int pageSize = 10,
|
|
[FromQuery] string sortBy = "",
|
|
[FromQuery] bool isAscending = true,
|
|
[FromQuery] Dictionary<string, string> filters = null)
|
|
{
|
|
var pagingParams = new PagingParams
|
|
{
|
|
PageNumber = pageNumber,
|
|
PageSize = pageSize,
|
|
SortBy = sortBy,
|
|
IsAscending = isAscending,
|
|
Filters = filters
|
|
};
|
|
|
|
// 可以在这里构建表达式树过滤条件
|
|
Expression<Func<T, bool>> filter = null;
|
|
|
|
var pagedResult = await _repository.GetPagedAsync(filter, pagingParams);
|
|
return new JsonResult(pagedResult);
|
|
}
|
|
|
|
[HttpGet]
|
|
public async Task<FileStreamResult> Export([FromQuery] int pageNumber = 1,
|
|
[FromQuery] int pageSize = 10,
|
|
[FromQuery] string sortBy = "",
|
|
[FromQuery] bool isAscending = true,
|
|
[FromQuery] Dictionary<string, string> filters = null)
|
|
{
|
|
var pagingParams = new PagingParams
|
|
{
|
|
PageNumber = pageNumber,
|
|
PageSize = pageSize,
|
|
SortBy = sortBy,
|
|
IsAscending = isAscending,
|
|
Filters = filters
|
|
};
|
|
|
|
// 可以在这里构建表达式树过滤条件
|
|
//Expression<Func<T, bool>> filter = null;
|
|
|
|
var pagedResult = await _repository.GetPagedAsync(null, pagingParams);
|
|
return await ExportFile(pagedResult.Data, Guid.NewGuid().ToString() + ".xlsx");
|
|
|
|
|
|
}
|
|
protected async Task<FileStreamResult> ExportFile(ICollection<T> dtos, string fileName)
|
|
{
|
|
var excelExporter = HttpContext.RequestServices.GetRequiredService<IExcelExporter>();
|
|
var res = await excelExporter.ExportAsByteArray(dtos);
|
|
return new FileStreamResult(new MemoryStream(res), "application/octet-stream") { FileDownloadName = DateTime.Now.ToString("yyyyMMddHHmm") + "_" + fileName };
|
|
}
|
|
|
|
|
|
|
|
|
|
[HttpGet]
|
|
|
|
public async Task<IActionResult> GetImportTemplate()
|
|
{
|
|
try
|
|
{
|
|
// 创建导入模板生成器
|
|
var importer = new ExcelImporter();
|
|
|
|
// 生成导入模板流(这里假设使用 YourModel 作为导入模型)
|
|
var bytes = await importer.GenerateTemplateBytes<T>();
|
|
|
|
|
|
//using var stream = new MemoryStream(bytes);
|
|
//stream.Seek(0, SeekOrigin.Begin);
|
|
|
|
// 设置友好的文件名,例如:"导入模板_20250530.xlsx"
|
|
var fileName = $"导入模板_{DateTime.Now:yyyyMMdd}.xlsx";
|
|
|
|
// 返回文件流结果
|
|
|
|
return new FileStreamResult(new MemoryStream(bytes), "application/octet-stream") { FileDownloadName = DateTime.Now.ToString("yyyyMMddHHmm") + "_" + fileName };
|
|
|
|
|
|
|
|
//return File(
|
|
// fileStream: stream,
|
|
// contentType: "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet",
|
|
// fileDownloadName: fileName);
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
// 记录异常日志
|
|
Console.WriteLine($"生成导入模板时出错: {ex.Message}");
|
|
|
|
// 返回错误响应
|
|
return StatusCode(500, "生成导入模板时发生错误");
|
|
}
|
|
}
|
|
[HttpPost("Import")]
|
|
public async virtual Task<IActionResult> Import(IFormFile file)
|
|
{
|
|
if (file == null || file.Length <= 0)
|
|
{
|
|
return BadRequest("No file uploaded.");
|
|
}
|
|
|
|
try
|
|
{
|
|
var excelImporter = HttpContext.RequestServices.GetRequiredService<IExcelImporter>();
|
|
var importResult = await excelImporter.Import<T>(file.OpenReadStream());
|
|
//if (importResult.HasError)
|
|
//{
|
|
// return BadRequest(importResult.ErrorMessage);
|
|
//}
|
|
|
|
// 处理导入的数据
|
|
List<T> list = new List<T>();
|
|
foreach (var item in importResult.Data)
|
|
{
|
|
list.Add(item);
|
|
|
|
}
|
|
await ImportBefore(list);
|
|
|
|
await _jobDbContext.BulkInsertAsync(list);
|
|
|
|
await ImportAfter(list);
|
|
|
|
|
|
return new JsonResult(new { Code = 200, Message = "修改成功!" });
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
await _logger.AddError(ex.Message, TaskName);
|
|
return new JsonResult(new { Code = 400, Message = "导入失败!" });
|
|
}
|
|
}
|
|
|
|
/// <summary>
|
|
/// 重写插入前进行操作,如校验等
|
|
/// </summary>
|
|
/// <param name="p_list"></param>
|
|
/// <returns></returns>
|
|
protected virtual async Task ImportBefore(List<T> p_list)
|
|
{
|
|
|
|
}
|
|
|
|
/// <summary>
|
|
/// 重写插入后进行操作如
|
|
/// </summary>
|
|
/// <param name="p_list"></param>
|
|
/// <returns></returns>
|
|
protected virtual async Task ImportAfter(List<T> p_list)
|
|
{
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
}
|
|
|