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.

389 lines
13 KiB

2 weeks ago
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;
2 weeks ago
using Dapper;
2 weeks ago
using Magicodes.ExporterAndImporter.Core.Extension;
using Magicodes.ExporterAndImporter.Excel;
2 weeks ago
using Microsoft.AspNetCore.Http;
2 weeks ago
using Microsoft.AspNetCore.Mvc;
2 weeks ago
using Microsoft.EntityFrameworkCore;
2 weeks ago
using Microsoft.Extensions.DependencyInjection;
2 weeks ago
using TaskManager.Contracts.Dtos;
using TaskManager.Entity;
using TaskManager.EntityFramework;
2 weeks ago
using TaskManager.EntityFramework.Repository;
2 weeks ago
namespace TaskManager.Controllers
3 weeks ago
{
2 weeks ago
public class CheryRecurringJobInputPageController<T,OUTPUT> : RecurringJobBaseController where T : BaseEntity ,new()
2 weeks ago
{
2 weeks ago
protected readonly IRepository<T> _repository;
2 weeks ago
public CheryRecurringJobInputPageController(HttpClient httpClient, JobDbContext jobDbContext, LogController log, IRepository<T> repository) : base(httpClient, jobDbContext, log)
2 weeks ago
{
2 weeks ago
_repository = repository;
2 weeks ago
}
2 weeks ago
protected override async Task DoExecutingAsync(string url, string path, string takName, string client)
2 weeks ago
{
2 weeks ago
Url = url;
Path = path;
TaskName = takName;
Client = client;
await SyncTaskSubTable(TaskName,Client);
2 weeks ago
}
2 weeks ago
private async Task<QRReturnInfo> PostPageAsync(PagedRequest<T> t)
2 weeks ago
{
2 weeks ago
try
2 weeks ago
{
2 weeks ago
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))
2 weeks ago
{
2 weeks ago
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;
2 weeks ago
}
}
2 weeks ago
catch (Exception ex)
{
await _logger.AddError($"调用接口无返回值错误{ex.Message}", TaskName);
return null;
}
2 weeks ago
2 weeks ago
}
2 weeks ago
2 weeks ago
public static string GenerateRandomStringWith8EG()
{
const string chars = "ABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789";
var random = new Random();
var sb = new StringBuilder(32);
2 weeks ago
2 weeks ago
// 随机生成 "8EG" 的插入位置 (0-29)
int position = random.Next(0, 30);
2 weeks ago
2 weeks ago
// 生成前半部分随机字符串
for (int i = 0; i < position; i++)
2 weeks ago
{
2 weeks ago
sb.Append(chars[random.Next(chars.Length)]);
2 weeks ago
}
2 weeks ago
// 插入 "8EG"
sb.Append("8EG");
// 生成后半部分随机字符串,补足32位
int remainingLength = 32 - sb.Length;
for (int i = 0; i < remainingLength; i++)
2 weeks ago
{
2 weeks ago
sb.Append(chars[random.Next(chars.Length)]);
2 weeks ago
}
2 weeks ago
return sb.ToString();
}
2 weeks ago
2 weeks ago
private async Task SyncTaskSubTable(string taskName, string client)
2 weeks ago
{
2 weeks ago
if (string.IsNullOrEmpty(taskName) || string.IsNullOrEmpty(client))
{
await _logger.AddError("任务名称或客户端不能为空",taskName);
return;
}
2 weeks ago
var sublist = _jobDbContext.TaskSub.Where(p => p.TaskName == taskName && p.WriteState == false && p.Subscriber == client).ToList();
2 weeks ago
int pageSize = 1000;
if (!sublist.Any())
{
foreach (var sub in sublist)
{
2 weeks ago
string querystr = $"select * from {sub.TableName} where TaskId='{sub.TaskId}' and WriteState=0 order by uid";//任务表
2 weeks ago
2 weeks ago
var entites = _jobDbContext.Database.GetDbConnection().Query<T>(querystr);//明细表
if (entites.Any())
2 weeks ago
{
2 weeks ago
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);
}
}
2 weeks ago
2 weeks ago
}
2 weeks ago
2 weeks ago
}
}
2 weeks ago
}
2 weeks ago
2 weeks ago
[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]
2 weeks ago
public async Task<ActionResult> GetPaged(
2 weeks ago
[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);
2 weeks ago
return new JsonResult(pagedResult);
2 weeks ago
}
[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
};
// 可以在这里构建表达式树过滤条件
2 weeks ago
//Expression<Func<T, bool>> filter = null;
2 weeks ago
var pagedResult = await _repository.GetPagedAsync(null, pagingParams);
2 weeks ago
return await ExportFile(pagedResult.Data, Guid.NewGuid().ToString() + ".xlsx");
2 weeks ago
}
2 weeks ago
protected async Task<FileStreamResult> ExportFile(ICollection<T> dtos, string fileName)
2 weeks ago
{
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 };
}
2 weeks ago
[HttpGet]
public async Task<IActionResult> GetImportTemplate()
{
try
{
// 创建导入模板生成器
var importer = new ExcelImporter();
// 生成导入模板流(这里假设使用 YourModel 作为导入模型)
var bytes = await importer.GenerateTemplateBytes<T>();
2 weeks ago
//using var stream = new MemoryStream(bytes);
//stream.Seek(0, SeekOrigin.Begin);
2 weeks ago
// 设置友好的文件名,例如:"导入模板_20250530.xlsx"
var fileName = $"导入模板_{DateTime.Now:yyyyMMdd}.xlsx";
// 返回文件流结果
2 weeks ago
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);
2 weeks ago
}
catch (Exception ex)
{
// 记录异常日志
Console.WriteLine($"生成导入模板时出错: {ex.Message}");
// 返回错误响应
return StatusCode(500, "生成导入模板时发生错误");
}
}
2 weeks ago
[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)
{
}
2 weeks ago
2 weeks ago
2 weeks ago
}
3 weeks ago
}