Display_D_HP/TeamAAS-VM/Services/MesService.cs

using Prism.Events;
using Prism.Ioc;
using System;
using System.Collections.Generic;
using System.Net.Http;
using System.Text;
using System.Threading;
using System.Threading.Tasks;
using TeamAAS_VP.Enums;
using TeamAAS_VP.Events;
using TeamAAS_VP.Interfaces;
using TeamAAS_VP.Models;

namespace TeamAAS_VP.Services
{
    /// <summary>
    /// 基于 <see cref="HttpClient"/> 的 MES 通信服务实现。
    /// <para>
    /// 本服务通过注入的 <see cref="IConfigService"/> 获取设备配置（包含 MES 接口地址），
    /// 提供对站点查询（StationGetAsync）和提交（SubmitGetAsync）的 GET 请求封装并带有重试逻辑。
    /// </para>
    /// <para>
    /// 注意：
    /// - <see cref="_httpClient"/> 为可复用的单一实例，建议在服务生命周期内重用以避免端口耗尽。
    /// - 本类实现了显式的 <see cref="Dispose"/> 方法以释放内部 <see cref="HttpClient"/> 资源，调用后实例不可再用。
    /// </para>
    /// </summary>
    public class MesService : IMesService
    {
        IEventAggregator _eventAggregator;
        /// <summary>
        /// 用于执行 HTTP 请求的客户端实例。建议在服务生命周期内重用，避免频繁创建导致套接字/端口耗尽。
        /// </summary>
        private readonly HttpClient _httpClient;

        /// <summary>
        /// 配置服务，用于获取设备信息（包含 MES URL、是否启用 MES 等）。
        /// </summary>
        private readonly IConfigService _configService;

        /// <summary>
        /// 标记实例是否已释放，防止重复释放和在释放后继续使用导致未定义行为。
        /// </summary>
        private bool _disposed;

        /// <summary>
        /// 使用指定的配置服务创建 <see cref="MesService"/> 实例。
        /// </summary>
        /// <param name="configService">用于获取设备配置信息的实现，不能为空。</param>
        /// <exception cref="ArgumentNullException">当 <paramref name="configService"/> 为 null 时抛出。</exception>
        public MesService(IConfigService configService, IContainerProvider container, IEventAggregator ea)
        {
            _configService = configService ?? throw new ArgumentNullException(nameof(configService));
            _httpClient = new HttpClient();
            _disposed = false;
            _eventAggregator = ea;
        }


        /// <summary>
        /// 向配置中指定的 MES 站点 URL 发起 GET 请求（用于过站查询）。
        /// </summary>
        /// <param name="sn">要查询的序列号（SN）。</param>
        /// <param name="cancellationToken">可选的取消令牌，用于请求超时或取消。</param>
        /// <returns>
        /// 返回元组 (IsSuccess, Response)：
        /// - IsSuccess: 请求是否视为成功（即 MES 返回了预期的 "SFC_OK" 且包含 "unit_process_check=OK"）。
        /// - Response: 原始响应文本或错误信息说明（如 "MES功能未启用!"、"Missing MES station URL"、"Canceled" 等）。
        /// </returns>
        /// <remarks>
        /// 行为说明：
        /// - 从配置获取设备信息，若 MES 未启用则直接返回 IsSuccess=true 且 Response 为提示文本（不视为错误）。
        /// - 若未配置 MES URL 则返回 IsSuccess=false 与错误文本。
        /// - 使用配置的 URL（通过 <see cref="string.Format"/> 填充 sn）发起 GET 请求，最多重试 10 次。
        /// - 只有当响应包含 "SFC_OK" 且包含 "unit_process_check=OK" 时视为成功并立即返回。
        /// - 捕获 <see cref="OperationCanceledException"/> 返回 "Canceled"，捕获其它异常则返回异常消息。
        /// - 若实例已被释放则抛出 <see cref="ObjectDisposedException"/>。
        /// </remarks>
        public async Task<(bool IsSuccess, string Response)> StationGetAsync(string sn, string comp, CancellationToken cancellationToken = default)
        {
            // 如果已经释放，则抛出异常，防止在已释放资源上继续操作。
            if (_disposed) throw new ObjectDisposedException(nameof(MesService));

            var device = _configService.GetDeviceInfo();
            if (device == null || !device.EnableMES)
            {
                return (true, "MES功能未启用!");
            }
            if (string.IsNullOrWhiteSpace(device.MesUrl))
                return (false, "Missing MES station URL");
            if (device.F == "PTL")
            {
                if (string.IsNullOrEmpty(device.comp))
                {
                    return (false, "PTL模式下comp不能为空");
                }
            }

            string url = "";
            if (device.F == "PTL")
            {
                url = $"{device.MesUrl}?c=QUERY_RECORD&line={device.Line}&station={device.Station}&fixtureid={device.FixtureId}&sn={sn}&comp={device.comp}:{comp}&p=unit_process_check,message,model";
            }
            else
            {
                url = $"{device.MesUrl}?c=QUERY_RECORD&line={device.Line}&station={device.Station}&fixtureid={device.FixtureId}&sn={sn}&p=unit_process_check,message,model";
            }

            LogHelper.WriteLogMes($"【询问URL】{url}");
            SendTaskMessage($"【询问URL】{url}", MessageLevel.Info);
            (bool IsSuccess, string Response) result = (false, string.Empty);

            for (int i = 0; i < 3; i++)
            {
                try
                {
                    using (var rsp = await _httpClient.GetAsync(url))
                    {
                        var text = await rsp.Content.ReadAsStringAsync();
                        LogHelper.WriteLogMes($"【HTTP接收】{text}");
                        SendTaskMessage($"【HTTP接收】{text}", MessageLevel.Info);

                        // "SFC_OK" 代表和 MES 连接成功
                        if (text.Contains("SFC_OK"))
                        {
                            if (text.Contains("unit_process_check=OK"))
                            {
                                return (true, text);
                            }
                        }
                        result = (false, text);
                    }
                }
                catch (OperationCanceledException)
                {
                    result = (false, "Canceled");
                }
                catch (Exception ex)
                {
                    result = (false, ex.Message);
                }
            }
            return result;
        }

        /// <summary>
        /// 向 MES 发起站点查询请求，使用超时（秒）作为便捷重载。
        /// </summary>
        /// <param name="sn">序列号（SN）。</param>
        /// <param name="timeoutInSeconds">请求超时，单位为秒。</param>
        /// <returns>与 <see cref="StationGetAsync(string, CancellationToken)"/> 相同的返回语义。</returns>
        public async Task<(bool IsSuccess, string Response)> StationGetAsync(string sn, string comp, double timeoutInSeconds)
        {
            using (var cts = new CancellationTokenSource(TimeSpan.FromSeconds(timeoutInSeconds)))
            {
                return await StationGetAsync(sn, comp, cts.Token);
            }
        }

        /// <summary>
        /// 向配置中指定的 MES 提交 URL 发起 GET 请求（用于提交测试结果/记录）。
        /// </summary>
        /// <param name="sn">要提交的序列号（SN）。</param>
        /// <param name="isSuccess">表示此次记录是否通过（true => PASS，false => FAIL）。</param>
        /// <param name="start_time">开始时间，提交时使用 "yyyy-MM-dd HH:mm:ss" 格式。</param>
        /// <param name="stop_time">结束时间，提交时使用 "yyyy-MM-dd HH:mm:ss" 格式。</param>
        /// <param name="cancellationToken">可选的取消令牌，用于请求超时或取消。</param>
        /// <returns>
        /// 返回元组 (IsSuccess, Response)：
        /// - IsSuccess: 提交是否被 MES 视为成功（收到 "SFC_OK" 且响应中无额外多行信息）。
        /// - Response: MES 返回文本或错误说明；当 MES 返回 "SFC_OK" 但包含额外多行信息（代表 SN 数据问题）时，IsSuccess 为 false 且 Response 为该文本。
        /// </returns>
        /// <remarks>
        /// 行为说明：
        /// - 从配置获取设备信息，若 MES 未启用则直接返回 IsSuccess=true 且 Response 为提示文本（不视为错误）。
        /// - 若未配置 MES 提交 URL 则返回 IsSuccess=false 与错误文本。
        /// - 使用配置的 URL（通过 <see cref="string.Format"/> 按顺序填充 sn、PASS/FAIL、start_time、stop_time）发起 GET 请求，最多重试 10 次。
        /// - 在收到包含 "SFC_OK" 的响应时，如果返回文本包含多行（有附加信息）则视为提交失败并返回该文本，否则视为提交成功。
        /// - 捕获 <see cref="OperationCanceledException"/> 返回 "Canceled"，捕获其它异常则返回异常消息。
        /// - 若实例已被释放则抛出 <see cref="ObjectDisposedException"/>。
        /// </remarks>
        public async Task<(bool IsSuccess, string Response)> SubmitGetAsync(string sn, string comp, bool isSuccess, DateTime start_time, DateTime stop_time, CancellationToken cancellationToken = default)
        {
            // 如果已经释放，则抛出异常，防止在已释放资源上继续操作。
            if (_disposed) throw new ObjectDisposedException(nameof(MesService));

            var device = _configService.GetDeviceInfo();
            if (device == null || !device.EnableMES)
            {
                return (true, "MES功能未启用!");
            }
            if (string.IsNullOrWhiteSpace(device.MesUrl))
                return (false, "Missing MES submit URL");
            if (device.F == "PTL")
            {
                if (string.IsNullOrEmpty(device.comp))
                {
                    return (false, "PTL模式下comp不能为空");
                }
            }

            string res = isSuccess ? "PASS" : "FAIL";
            string url = "";
            if (device.F == "PTL")
            {
                url = $"{device.MesUrl}?c=ADD_RECORD&line={device.Line}&station={device.Station}&fixtureid={device.FixtureId}&sn={sn}&f=PTL&comp={device.comp}:{comp}" +
                    $"&result={res}&start_time={start_time.ToString("yyyy-MM-dd HH:mm:ss")}&stop_time={stop_time.ToString("yyyy-MM-dd HH:mm:ss")}";
            }
            else
            {
                url = $"{device.MesUrl}?c=ADD_RECORD&line={device.Line}&station={device.Station}&fixtureid={device.FixtureId}&sn={sn}&f=PQC" +
                    $"&result={res}&start_time={start_time.ToString("yyyy-MM-dd HH:mm:ss")}&stop_time={stop_time.ToString("yyyy-MM-dd HH:mm:ss")}";
            }
            LogHelper.WriteLogMes($"【上传URL】{url}");
            SendTaskMessage($"【上传URL】{url}", MessageLevel.Info);
            (bool IsSuccess, string Response) result = (false, string.Empty);

            for (int i = 0; i < 3; i++)
            {
                try
                {
                    using (var rsp = await _httpClient.GetAsync(url, cancellationToken))
                    {
                        var text = await rsp.Content.ReadAsStringAsync();
                        LogHelper.WriteLogMes($"【HTTP接收】{text}");
                        SendTaskMessage($"【HTTP接收】{text}", MessageLevel.Info);
                        // "SFC_OK" 代表和 MES 连接成功
                        if (text.Contains("SFC_OK"))
                        {
                            // 如果有附加信息（多行），代表此 SN 数据有问题，视为失败并返回该文本
                            if (text.Split(new[] { '\n' }, StringSplitOptions.RemoveEmptyEntries).Length > 1)
                            {
                                return (false, text);
                            }
                            return (true, text);
                        }
                        result = (false, text);
                    }
                }
                catch (OperationCanceledException)
                {
                    result = (false, "Canceled");
                }
                catch (Exception ex)
                {
                    result = (false, ex.Message);
                }
            }
            return result;
        }

        /// <summary>
        /// 向 MES 提交请求的超时重载，使用超时（秒）作为便捷参数。
        /// </summary>
        /// <param name="sn">序列号（SN）。</param>
        /// <param name="isSuccess">是否通过（PASS/FAIL）。</param>
        /// <param name="start_time">开始时间。</param>
        /// <param name="stop_time">结束时间。</param>
        /// <param name="timeoutInSeconds">请求超时，单位为秒。</param>
        /// <returns>与 <see cref="SubmitGetAsync(string, bool, DateTime, DateTime, CancellationToken)"/> 相同的返回语义。</returns>
        public async Task<(bool IsSuccess, string Response)> SubmitGetAsync(string sn, string comp, bool isSuccess, DateTime start_time, DateTime stop_time, double timeoutInSeconds)
        {
            using (var cts = new CancellationTokenSource(TimeSpan.FromSeconds(timeoutInSeconds)))
            {
                return await SubmitGetAsync(sn, comp, isSuccess, start_time, stop_time, cts.Token);
            }
        }

        /// <summary>
        /// 释放服务占用的托管资源（当前仅 <see cref="_httpClient"/>）。
        /// <para>本方法为幂等操作，重复调用不会产生异常。调用后实例不可再用于发起请求，尝试使用将抛出 <see cref="ObjectDisposedException"/>。</para>
        /// </summary>
        public void Dispose()
        {
            if (!_disposed)
            {
                _httpClient.Dispose();
                _disposed = true;
            }
        }
        public void SendTaskMessage(string msg, MessageLevel level)
        {
            App.Current.Dispatcher.Invoke(() =>
            {
                _eventAggregator.GetEvent<TaskMessageNotification>().Publish(new Models.MessageStruct() { Message = msg, level = level });
            });
        }
    }
}