Form1.cs

using test.Models;
using test.Services;

namespace test
{
    public partial class Form1 : Form
    {
        private bool _isTesting = false;
        private TestStatistics _testStatistics = new TestStatistics();

        // 服务类
        private readonly UserAgentManager _userAgentManager;
        private readonly RandomDataGenerator _randomGenerator;
        private readonly VariableReplacer _variableReplacer;
        private readonly FormDataParser _formDataParser;
        private readonly HttpTestService _httpTestService;

        public Form1()
        {
            InitializeComponent();

            // 初始化服务
            _userAgentManager = new UserAgentManager();
            _randomGenerator = new RandomDataGenerator();
            _variableReplacer = new VariableReplacer(_randomGenerator);
            _formDataParser = new FormDataParser();
            _httpTestService = new HttpTestService(_variableReplacer, _formDataParser);

            InitializeUAComboBox();
            comboBoxMethod.SelectedIndexChanged += ComboBoxMethod_SelectedIndexChanged;
        }

        private void ComboBoxMethod_SelectedIndexChanged(object? sender, EventArgs e)
        {
            // 当选择POST时启用POST数据输入框
            textBoxPostData.Enabled = comboBoxMethod.SelectedItem?.ToString() == "POST";
        }

        private void InitializeUAComboBox()
        {
            comboBoxUA.Items.Add("自定义");
            foreach (var uaName in _userAgentManager.GetBuiltInUANames())
            {
                comboBoxUA.Items.Add(uaName);
            }
            comboBoxUA.SelectedIndex = 0;
            comboBoxUA.SelectedIndexChanged += ComboBoxUA_SelectedIndexChanged;
        }

        private void ComboBoxUA_SelectedIndexChanged(object? sender, EventArgs e)
        {
            if (comboBoxUA.SelectedIndex > 0)
            {
                string? selectedUA = comboBoxUA.SelectedItem?.ToString();
                if (!string.IsNullOrEmpty(selectedUA))
                {
                    string? ua = _userAgentManager.GetUserAgent(selectedUA);
                    if (ua != null)
                    {
                        textBoxCustomUA.Text = ua;
                    }
                }
            }
        }

        private async void button1_Click(object sender, EventArgs e)
        {
            if (_isTesting)
            {
                MessageBox.Show("测试正在进行中，请稍候...", "提示", MessageBoxButtons.OK, MessageBoxIcon.Information);
                return;
            }

            // 验证输入
            if (!ValidateInput())
            {
                return;
            }

            // 开始测试
            StartTest();

            // 获取测试参数
            string url = textBoxUrl.Text.Trim();
            int concurrentCount = (int)numericUpDownConcurrent.Value;
            int requestCount = (int)numericUpDownRequests.Value;
            string userAgent = textBoxCustomUA.Text.Trim();
            string httpMethod = comboBoxMethod.SelectedItem?.ToString() ?? "GET";
            string postData = textBoxPostData.Text.Trim();

            // 显示测试信息
            DisplayTestInfo(url, httpMethod, postData, userAgent, concurrentCount, requestCount);

            DateTime startTime = DateTime.Now;

            // 执行并发测试
            await ExecuteConcurrentTest(url, httpMethod, postData, userAgent, concurrentCount, requestCount);

            DateTime endTime = DateTime.Now;
            TimeSpan duration = endTime - startTime;

            // 显示测试结果
            DisplayTestResults(endTime, duration, requestCount);

            // 结束测试
            EndTest();
        }

        /// <summary>
        /// 验证输入参数
        /// </summary>
        private bool ValidateInput()
        {
            string url = textBoxUrl.Text.Trim();
            string httpMethod = comboBoxMethod.SelectedItem?.ToString() ?? "GET";
            string postData = textBoxPostData.Text.Trim();

            if (string.IsNullOrEmpty(url))
            {
                MessageBox.Show("请输入URL", "错误", MessageBoxButtons.OK, MessageBoxIcon.Error);
                return false;
            }

            if (httpMethod == "POST" && string.IsNullOrEmpty(postData))
            {
                MessageBox.Show("POST请求需要填写请求数据", "错误", MessageBoxButtons.OK, MessageBoxIcon.Error);
                return false;
            }

            return true;
        }

        /// <summary>
        /// 开始测试
        /// </summary>
        private void StartTest()
        {
            _testStatistics.Reset();
            _isTesting = true;
            button1.Enabled = false;
            button1.Text = "测试中...";
        }

        /// <summary>
        /// 结束测试
        /// </summary>
        private void EndTest()
        {
            button1.Enabled = true;
            button1.Text = "开始测试";
            _isTesting = false;
        }

        /// <summary>
        /// 显示测试信息
        /// </summary>
        private void DisplayTestInfo(string url, string httpMethod, string postData, string userAgent, int concurrentCount, int requestCount)
        {
            textBoxResult.Clear();
            textBoxResult.AppendText($"开始HTTP并发测试...\r\n");
            textBoxResult.AppendText($"请求方法: {httpMethod}\r\n");
            textBoxResult.AppendText($"URL: {url}\r\n");
            if (httpMethod == "POST")
            {
                textBoxResult.AppendText($"POST数据: {postData}\r\n");
            }
            textBoxResult.AppendText($"User-Agent: {userAgent}\r\n");
            textBoxResult.AppendText($"并发数: {concurrentCount}\r\n");
            textBoxResult.AppendText($"总请求数: {requestCount}\r\n");
            textBoxResult.AppendText($"开始时间: {DateTime.Now:yyyy-MM-dd HH:mm:ss}\r\n\r\n");
        }

        /// <summary>
        /// 执行并发测试
        /// </summary>
        private async Task ExecuteConcurrentTest(string url, string httpMethod, string postData, string userAgent, int concurrentCount, int requestCount)
        {
            // 使用SemaphoreSlim控制并发数
            SemaphoreSlim semaphore = new SemaphoreSlim(concurrentCount, concurrentCount);
            List<Task> tasks = new List<Task>();

            for (int i = 0; i < requestCount; i++)
            {
                int requestIndex = i + 1;
                tasks.Add(Task.Run(async () =>
                {
                    await semaphore.WaitAsync();
                    try
                    {
                        await ProcessHttpRequest(url, httpMethod, postData, userAgent, requestIndex);
                    }
                    finally
                    {
                        semaphore.Release();
                    }
                }));
            }

            // 等待所有任务完成
            await Task.WhenAll(tasks);
        }

        /// <summary>
        /// 处理 HTTP 请求
        /// </summary>
        private async Task ProcessHttpRequest(string url, string httpMethod, string postData, string userAgent, int requestIndex)
        {
            var result = await _httpTestService.SendHttpRequestAsync(url, httpMethod, postData, userAgent, requestIndex);

            // 更新统计
            if (result.IsSuccess)
            {
                _testStatistics.IncrementSuccess();
            }
            else
            {
                _testStatistics.IncrementFail();
            }

            // 在UI线程更新结果
            if (InvokeRequired)
            {
                Invoke(new Action(() => AppendResult(result)));
            }
            else
            {
                AppendResult(result);
            }
        }

        /// <summary>
        /// 追加结果到文本框
        /// </summary>
        private void AppendResult(HttpRequestResult result)
        {
            if (result.IsSuccess)
            {
                textBoxResult.AppendText($"[{result.RequestIndex}] 成功 - 状态码: {result.StatusCode}, 响应时间: {result.ResponseTimeMs:F2}ms\r\n");
            }
            else
            {
                if (!string.IsNullOrEmpty(result.ErrorMessage))
                {
                    textBoxResult.AppendText($"[{result.RequestIndex}] 失败 - 错误: {result.ErrorMessage}\r\n");
                }
                else
                {
                    textBoxResult.AppendText($"[{result.RequestIndex}] 失败 - 状态码: {result.StatusCode}, 响应时间: {result.ResponseTimeMs:F2}ms\r\n");
                }
            }
        }

        /// <summary>
        /// 显示测试结果
        /// </summary>
        private void DisplayTestResults(DateTime endTime, TimeSpan duration, int requestCount)
        {
            textBoxResult.AppendText($"\r\n========== 测试完成 ==========\r\n");
            textBoxResult.AppendText($"结束时间: {endTime:yyyy-MM-dd HH:mm:ss}\r\n");
            textBoxResult.AppendText($"总耗时: {duration.TotalSeconds:F2} 秒\r\n");
            textBoxResult.AppendText($"成功: {_testStatistics.SuccessCount}\r\n");
            textBoxResult.AppendText($"失败: {_testStatistics.FailCount}\r\n");
            textBoxResult.AppendText($"成功率: {_testStatistics.SuccessRate:F2}%\r\n");
            textBoxResult.AppendText($"平均响应时间: {(duration.TotalMilliseconds / requestCount):F2} ms\r\n");
        }

    }
}