85 lines
2.7 KiB
C#
85 lines
2.7 KiB
C#
namespace WebApi.Controllers;
|
||
|
||
/// <summary>
|
||
/// 账单导入控制器
|
||
/// </summary>
|
||
[ApiController]
|
||
[Route("api/[controller]/[action]")]
|
||
public class BillImportController(
|
||
ILogger<BillImportController> logger,
|
||
IImportService importService
|
||
) : ControllerBase
|
||
{
|
||
/// <summary>
|
||
/// 上传账单文件
|
||
/// </summary>
|
||
/// <param name="file">账单文件</param>
|
||
/// <param name="type">账单类型（Alipay | WeChat）</param>
|
||
/// <returns></returns>
|
||
[HttpPost]
|
||
public async Task<BaseResponse<object>> UploadFile(
|
||
[FromForm] IFormFile file,
|
||
[FromForm] string type
|
||
)
|
||
{
|
||
try
|
||
{
|
||
// 验证参数
|
||
if (file.Length == 0)
|
||
{
|
||
return BaseResponse<object>.Fail("请选择要上传的文件");
|
||
}
|
||
|
||
if (string.IsNullOrWhiteSpace(type) || (type != "Alipay" && type != "WeChat"))
|
||
{
|
||
return BaseResponse<object>.Fail("账单类型参数错误，必须是 Alipay 或 WeChat");
|
||
}
|
||
|
||
// 验证文件类型
|
||
var allowedExtensions = new[] { ".csv", ".xlsx", ".xls" };
|
||
var fileExtension = Path.GetExtension(file.FileName).ToLowerInvariant();
|
||
if (!allowedExtensions.Contains(fileExtension))
|
||
{
|
||
return BaseResponse<object>.Fail("只支持 CSV 或 Excel 文件格式");
|
||
}
|
||
|
||
// 验证文件大小（10MB限制）
|
||
const long maxFileSize = 10 * 1024 * 1024;
|
||
if (file.Length > maxFileSize)
|
||
{
|
||
return BaseResponse<object>.Fail("文件大小不能超过 10MB");
|
||
}
|
||
|
||
// 生成唯一文件名
|
||
var fileName = $"{type}_{DateTime.Now:yyyyMMddHHmmss}_{Guid.NewGuid():N}{fileExtension}";
|
||
|
||
// 保存文件
|
||
var ok = false;
|
||
var message = string.Empty;
|
||
await using (var stream = new MemoryStream())
|
||
{
|
||
await file.CopyToAsync(stream);
|
||
if (type == "Alipay")
|
||
{
|
||
(ok, message) = await importService.ImportAlipayAsync(stream, fileExtension);
|
||
}
|
||
else if (type == "WeChat")
|
||
{
|
||
(ok, message) = await importService.ImportWeChatAsync(stream, fileExtension);
|
||
}
|
||
}
|
||
|
||
return new BaseResponse<object>
|
||
{
|
||
Success = ok,
|
||
Message = message
|
||
};
|
||
}
|
||
catch (Exception ex)
|
||
{
|
||
logger.LogError(ex, "文件上传失败，类型: {Type}", type);
|
||
return BaseResponse<object>.Fail($"文件上传失败: {ex.Message}");
|
||
}
|
||
}
|
||
}
|