246 lines
8.7 KiB
C#
246 lines
8.7 KiB
C#
using CliWrap;
|
||
using CliWrap.Buffered;
|
||
using Microsoft.AspNetCore.Hosting;
|
||
using Microsoft.AspNetCore.Http;
|
||
using Microsoft.AspNetCore.Http.HttpResults;
|
||
using Microsoft.AspNetCore.Mvc;
|
||
using System.ComponentModel;
|
||
using System.Diagnostics;
|
||
using System.IO;
|
||
using System.Net.Http;
|
||
using System.Text;
|
||
using System.Text.Json.Serialization;
|
||
using static System.Net.Mime.MediaTypeNames;
|
||
using Newtonsoft.Json;
|
||
using BLV_API;
|
||
|
||
namespace FaceValidator.Controllers
|
||
{
|
||
[Route("api/[controller]/[action]")]
|
||
[ApiController]
|
||
public class FaceFeatureController : ControllerBase
|
||
{
|
||
private readonly IWebHostEnvironment _hostingEnvironment;
|
||
private readonly ILogger<FaceFeatureController> _logger;
|
||
|
||
public FaceFeatureController(
|
||
IWebHostEnvironment hostingEnvironment,
|
||
ILogger<FaceFeatureController> logger)
|
||
{
|
||
_hostingEnvironment = hostingEnvironment;
|
||
_logger = logger;
|
||
}
|
||
|
||
/// <summary>
|
||
/// 提交人脸Url/File
|
||
/// </summary>
|
||
/// <param name="data"></param>
|
||
/// <returns></returns>
|
||
[HttpPost]
|
||
public async Task<IActionResult> ValidateFaceVal([FromForm] UploadFileModel data)
|
||
{
|
||
try
|
||
{
|
||
var file = data.File;
|
||
var faceUrl = data.FaceUrl;
|
||
// 检查输入是否有效
|
||
if (file == null && string.IsNullOrWhiteSpace(faceUrl))
|
||
{
|
||
return BadRequest("必须提供图片文件或图片URL");
|
||
}
|
||
|
||
// 保存图片到本地临时文件
|
||
string tempImagePath;
|
||
if (file != null)
|
||
{
|
||
//
|
||
tempImagePath = await SaveUploadedFileAsync(file);
|
||
}
|
||
else
|
||
{
|
||
tempImagePath = await DownloadImageAsync(faceUrl);
|
||
}
|
||
|
||
// 调用验证方法
|
||
var validationResult = await FaceValidator(tempImagePath);
|
||
|
||
// 清理临时文件
|
||
//System.IO.File.Delete(tempImagePath);
|
||
|
||
return validationResult;
|
||
}
|
||
catch (Exception ex)
|
||
{
|
||
_logger.LogError(ex, "人脸验证过程中发生未处理异常");
|
||
return StatusCode(500, $"系统错误: {ex.Message}");
|
||
}
|
||
}
|
||
|
||
private async Task<string> SaveUploadedFileAsync(IFormFile file)
|
||
{
|
||
var tempFileName = $"{Guid.NewGuid()}{Path.GetExtension(file.FileName)}";
|
||
var tempFilePath = Path.Combine(_hostingEnvironment.ContentRootPath, "wwwroot/upload", tempFileName);
|
||
|
||
// 确保目录存在
|
||
Directory.CreateDirectory(Path.GetDirectoryName(tempFilePath));
|
||
|
||
using (var stream = new FileStream(tempFilePath, FileMode.Create))
|
||
{
|
||
await file.CopyToAsync(stream);
|
||
}
|
||
|
||
return tempFilePath;
|
||
}
|
||
|
||
private async Task<string> DownloadImageAsync(string faceUrl)
|
||
{
|
||
if (!Uri.TryCreate(faceUrl, UriKind.Absolute, out var uri))
|
||
{
|
||
throw new ArgumentException("提供的图片URL无效");
|
||
}
|
||
HttpClient httpClient = new();
|
||
var response = await httpClient.GetAsync(uri);
|
||
|
||
if (!response.IsSuccessStatusCode)
|
||
{
|
||
throw new ApplicationException($"无法下载图片: {response.StatusCode}");
|
||
}
|
||
|
||
var tempFileName = $"{Guid.NewGuid()}.jpg";
|
||
var tempFilePath = Path.Combine(_hostingEnvironment.ContentRootPath, "wwwroot/upload", tempFileName);
|
||
|
||
// 确保目录存在
|
||
Directory.CreateDirectory(Path.GetDirectoryName(tempFilePath));
|
||
|
||
await using (var fs = new FileStream(tempFilePath, FileMode.Create))
|
||
await using (var stream = await response.Content.ReadAsStreamAsync())
|
||
{
|
||
await stream.CopyToAsync(fs);
|
||
}
|
||
|
||
return tempFilePath;
|
||
}
|
||
|
||
[HttpPost]
|
||
public async Task<IActionResult> FaceValidator(string localImagePath)
|
||
{
|
||
// 1. 错误处理：确保参数不为空
|
||
if (string.IsNullOrWhiteSpace(localImagePath))
|
||
{
|
||
return BadRequest("脸部数据参数不能为空");
|
||
}
|
||
|
||
// 2. 确保图片文件存在
|
||
if (!System.IO.File.Exists(localImagePath))
|
||
{
|
||
return BadRequest($"本地图片文件不存在: {localImagePath}");
|
||
}
|
||
|
||
// 3. 使用绝对路径确保可靠性
|
||
var exePath = Path.Combine(_hostingEnvironment.ContentRootPath, "ValRel", "faceidtest.exe");
|
||
var exeDirectory = Path.GetDirectoryName(exePath);
|
||
|
||
// 4. 检查可执行文件是否存在
|
||
if (!System.IO.File.Exists(exePath))
|
||
{
|
||
_logger.LogError($"可执行文件不存在: {exePath}");
|
||
return StatusCode(500, $"人脸验证程序未找到");
|
||
}
|
||
|
||
try
|
||
{
|
||
var stdOutBuffer = new StringBuilder();
|
||
var stdErrBuffer = new StringBuilder();
|
||
|
||
// 传递文件路径作为参数
|
||
var arguments = $"{localImagePath} 80x v3";
|
||
|
||
_logger.LogInformation($"执行人脸验证: {exePath} {arguments}");
|
||
|
||
// 创建取消令牌源并设置超时（例如20秒）
|
||
using var cts = new CancellationTokenSource(TimeSpan.FromSeconds(20));
|
||
|
||
var result = await Cli.Wrap(exePath) // 指定可执行文件
|
||
.WithArguments(arguments) // 传递参数
|
||
.WithWorkingDirectory(exeDirectory) // 在exe所在目录执行
|
||
.WithValidation(CommandResultValidation.None) // 退出条件
|
||
.WithStandardOutputPipe(PipeTarget.ToStringBuilder(stdOutBuffer)) // 打印结果日志
|
||
.WithStandardErrorPipe(PipeTarget.ToStringBuilder(stdErrBuffer)) // 打印过程日志
|
||
.ExecuteAsync(cts.Token); // 设置取消令牌
|
||
|
||
var fullOutput = stdErrBuffer.ToString();
|
||
fullOutput += stdOutBuffer.ToString();
|
||
|
||
string[] lines = fullOutput.Split(["\r\n", "\r", "\n"], StringSplitOptions.None);
|
||
|
||
|
||
var validationResult = "";
|
||
_logger.LogInformation($"结果: {fullOutput}");
|
||
var name = "";
|
||
foreach (var item in lines)
|
||
{
|
||
if (item.IndexOf("quality:") > -1)
|
||
{
|
||
validationResult = item.Split(":")[2].Trim();
|
||
break;
|
||
}
|
||
|
||
}
|
||
var vurl = "http://blv-rd.tech:19099/upload/";
|
||
//var vurl = "http://piccheck.blv-oa.com/upload/";
|
||
//D:\\AFace\\wwwroot/upload\\d7d5f295-8c75-4308-b36a-367c650a8ac4.jpg
|
||
vurl += Path.GetFileName(localImagePath);
|
||
LogHelper.WriteLog("图片名称：" + Path.GetFileName(localImagePath) + $" 验证结果: {validationResult}");
|
||
if (double.TryParse(validationResult, out double validationResultNum))
|
||
{
|
||
return Ok(new
|
||
{
|
||
Success = true,
|
||
Result = result,
|
||
ExitCode = 200,
|
||
quality = validationResultNum,
|
||
Url = vurl,
|
||
});
|
||
}
|
||
else
|
||
{
|
||
return Ok(new
|
||
{
|
||
Success = false,
|
||
Result = result,
|
||
ExitCode = 200,
|
||
Msg = "验证错误",
|
||
Url = vurl,
|
||
});
|
||
}
|
||
|
||
|
||
}
|
||
catch (OperationCanceledException)
|
||
{
|
||
_logger.LogError("操作超时，人脸验证未完成");
|
||
return StatusCode(504, "验证超时");
|
||
}
|
||
catch (Exception ex)
|
||
{
|
||
_logger.LogError(ex, "人脸验证过程中发生未处理异常");
|
||
return StatusCode(500, $"系统错误: {ex.Message}");
|
||
}
|
||
}
|
||
}
|
||
public class UploadFileModel
|
||
{
|
||
public IFormFile? File { get; set; }
|
||
public string? FaceUrl { get; set; }
|
||
}
|
||
|
||
public class ResultClass
|
||
{
|
||
public bool Success { get; set; }
|
||
public string? Result { get; set; }
|
||
public int Code { get; set; }
|
||
public string? Msg { get; set; }
|
||
|
||
}
|
||
|
||
} |