<think>嗯，用户想了解使用Python编写性能测试脚本的方法。根据提供的引用内容，主要涉及requests库的基础测试、多线程并发测试以及Locust工具的使用。 用户可能是个测试工程师或开发人员，正在寻找高效的性能测试方案。从引用[2]可以看出用户已经意识到Python多线程的局限性（GIL导致单核瓶颈），引用[3][4]则显示用户对Locust这类专业工具有兴趣。需要重点突出不同方案的适用场景。 先梳理核心方法： 基础单请求测试用requests+unittest足够简单，适合功能验证；多线程适用于低并发场景但受GIL限制；Locust的协程机制能实现更高并发，且支持分布式和Web监控界面，是专业级选择。特别要注意引用[3]提到的性能对比——Locust分布式性能比JMeter低1/3，这点需要客观说明。 用户可能还关心脚本编写复杂度。从引用[4]的示例看，Locust脚本结构清晰：定义TaskSet类包含@task装饰的请求方法，设置HttpLocust类配置主机和等待时间。比直接用threading更抽象但更易维护。 最后考虑给出选择建议：小规模测试用多线程快速验证，大规模压测首选Locust分布式。补充性能优化方向（如替换geventhttpclient）和资源链接，满足用户延伸学习需求。 </think> 基于引用内容，以下是使用 Python 编写性能测试脚本的 **三种核心方法** 及适用场景： --- ### 一、基础单请求测试（`requests` + `unittest`） **适用场景**：接口功能验证、基础性能采样 **特点**：简单易用，但无原生并发支持 ```python import requests import unittest import time class APITest(unittest.TestCase): def test_single_request(self): start = time.time() response = requests.get("https://api.example.com/data") latency = (time.time() - start) * 1000 # 毫秒 self.assertEqual(response.status_code, 200) print(f"响应时间: {latency:.2f}ms") if __name__ == "__main__": unittest.main() ``` --- ### 二、多线程并发测试（`threading`） **适用场景**：模拟中低并发场景（受 Python GIL 限制） **特点**：开发简单，但单机性能瓶颈明显[^2] ```python import threading import requests def send_request(url): response = requests.get(url) print(f"状态码: {response.status_code}, 线程ID: {threading.get_ident()}") # 并发配置 threads = [] url = "https://api.example.com" concurrent_users = 50 # 并发数 for _ in range(concurrent_users): t = threading.Thread(target=send_request, args=(url,)) threads.append(t) t.start() for t in threads: t.join() ``` --- ### 三、专业压测工具 Locust（推荐方案） **适用场景**：高并发分布式压测、实时监控 **特点**： 1. 基于协程，单机支持数千并发 2. 分布式扩展能力 3. Web 实时监控界面[^4] **示例脚本** (`locustfile.py`): ```python from locust import HttpUser, task, between class ApiUser(HttpUser): wait_time = between(0.5, 2) # 请求间隔(秒) @task(weight=3) # 权重占比 def get_data(self): self.client.get("/api/data") # 自动统计响应时间 @task(weight=1) def create_item(self): self.client.post("/api/items", json={"name": "test"}) ``` **启动命令**： ```bash locust -f locustfile.py --host=https://api.example.com ``` 访问 `http://localhost:8089` 配置并发参数并查看实时报表。 --- ### 关键对比总结 | 方法 | 并发能力 | 学习曲线 | 分布式支持 | 监控界面 | |---------------|---------------|----------|------------|----------| | 单请求 | 单线程 | ★☆☆☆☆ | ❌ | ❌ | | 多线程 | 数百并发 | ★★☆☆☆ | ❌ | ❌ | | **Locust** | **数万并发** | ★★★☆☆ | ✅ | ✅ | > 💡 **建议选择**： > - 快速验证接口 → 基础 `requests` > - 中低并发测试 → 多线程方案 > - **生产级压测 → Locust（分布式部署可突破单机性能瓶颈）[^3]** ---