压力测试代码python
时间: 2023-10-20 07:34:59 浏览: 101
当进行压力测试时,你可以使用Python的一些库和技术来编写代码。以下是一个简单的例子,使用`requests`库发送并发请求来模拟压力:
```python
import requests
from concurrent.futures import ThreadPoolExecutor
def make_request(url):
try:
response = requests.get(url)
print(f"Response from {url}: {response.status_code}")
except Exception as e:
print(f"Error: {e}")
urls = ["http://example.com", "http://google.com", "http://github.com"]
num_requests = 10
with ThreadPoolExecutor(max_workers=5) as executor:
for _ in range(num_requests):
executor.map(make_request, urls)
```
在这个例子中,我们定义了一个`make_request`函数来发送一个GET请求,并打印响应状态码。我们创建了一个包含要测试的URL列表的`urls`变量。然后,我们使用`ThreadPoolExecutor`来创建一个线程池,并指定最大工作线程数为5。接下来,我们循环执行`make_request`函数,通过`executor.map`方法将URL列表和函数传递给线程池,以并发地发送请求。
请注意,这只是一个简单的示例,你可能需要根据你的具体需求来调整代码。另外,还有其他一些库和工具可以帮助你进行更复杂的压力测试,如`locust`和`pytest-benchmark`。
阅读全文