Question

locust --no-web --client = 1 --hatch-rate = 1 --num-request = 2 --host = http://localhost

我想阅读HTTPLocust类中cmd行中提供的--host值。我知道我可以使用host属性进行直接赋值，但我不想要它。我想在HTTPLocust类中读取cmd行的值。我正在构建自定义日志，并希望将该值传递给日志。我尝试了 HTTPLocust.host ，但没有返回。

另外我想从python代码中读取--web-port。

Answer 1

这是我的代码：

import time,csv,argparse

class MySQLLocust(Locust):
    parser = argparse.ArgumentParser()
    parser.add_argument('--host', '--host')
    args, unknown = parser.parse_known_args()
    print("Host = " + args.host)

现在我给：

locust -f mysql_locust.py --host=myhost-vm-101 --no-web --clients=2 --hatch-rate=10 --run-time=5m

我得到预期的打印报表结果：

Host = myhost-vm-101

Answer 2

您可以通过sys.argv

访问它

或通过argparse

解析选项

import argparse

parser = argparse.ArgumentParser()
parser.add_argument('-H', '--host')
args, unknown = parser.parse_known_args()

print str(args.host)

Answer 3

您可以通过host类的实例访问User()变量。请参阅以下示例：

from locust import HttpLocust, TaskSet, task
import random, requests, time, os, inspect, json, sys


class UserBehaviour(TaskSet):
    @task(1)
    def test1(self):
        user = User()  
        print(user.host)
        self.client.get("/v3/User/ListOfCoupon/")

class User(HttpLocust):
    task_set = UserBehaviour
    min_wait = 1000
    max_wait = 3000

见日志：

~/P/m/p/general (master ⚡↑) locust -f app_couponlist.py --host=http://www.google.com
[2017-09-19 14:33:13,020] Mesuts-MacBook.local/INFO/locust.main: Starting web monitor at *:8089
[2017-09-19 14:33:13,021] Mesuts-MacBook.local/INFO/locust.main: Starting Locust 0.8a3
[2017-09-19 14:33:22,281] Mesuts-MacBook.local/INFO/locust.runners: Hatching and swarming 5 clients at the rate 1 clients/s...
[2017-09-19 14:33:22,282] Mesuts-MacBook.local/INFO/stdout: http://www.google.com
[2017-09-19 14:33:22,282] Mesuts-MacBook.local/INFO/stdout:
[2017-09-19 14:33:23,285] Mesuts-MacBook.local/INFO/stdout: http://www.google.com
[2017-09-19 14:33:23,285] Mesuts-MacBook.local/INFO/stdout:
[2017-09-19 14:33:24,226] Mesuts-MacBook.local/INFO/stdout: http://www.google.com

Answer 4

新答案

有一个比我下面的第一个更直接的解决方案。每个TaskSet都有一个locust属性，该属性链接回其父级Locust [locust-instance]，因此类似这样的操作将完全满足您的需求：

class UserBehaviour(TaskSet):
    def __init__(self, parent):
        super().__init__(parent)
        print(self.locust.host)

旧答案

看着code for HttpSession，看来base_url是您想要的。

因此，这样的操作应该可以为您提供当前的host（默认设置或在命令行上指定）：

class HostGetter(locust.TaskSet):
    @locust.Task()
    def get_host(self):
        print(self.client.base_url)

刺槐。从python代码

4 个答案:

新答案

旧答案