It’s possible to use Locust as a library, instead of running Locust using the locust command.
將 Locust 作為 Python 的一個(gè)三方庫归斤，通過 Python 代碼運(yùn)行爽彤，代替命令行的方式。
To run Locust as a library you need to create an Environment instance:
通過 Python 代碼來運(yùn)行，你需要?jiǎng)?chuàng)建 Environment 的實(shí)例：
from locust.env import Environment
env = Environment(user_classes=[MyTestUser])
The Environment instance’s create_local_runner, create_master_runner or create_worker_runner can then be used to start a Runner instance, which can be used to start a load test:
Environment 實(shí)例中的 Runner 用來使用不同的方式啟動(dòng)負(fù)載測試坎吻，分別為 create_local_runner, create_master_runner 或 create_worker_runner距贷。
env.create_local_runner()
env.runner.start(5000, hatch_rate=20)
env.runner.greenlet.join()
We could also use the Environment instance’s create_web_ui method to start a Web UI that can be used to view the stats, and to control the runner (e.g. start and stop load tests):
也可以使用 Environment 實(shí)例中的 create_web_ui 方法來啟動(dòng) web UI 以便控制運(yùn)行（啟動(dòng)或停止負(fù)載測試）和查看統(tǒng)計(jì)信息：
env.create_local_runner()
env.create_web_ui()
env.web_ui.greenlet.join()
3# Full example 完整示例
import gevent
from locust import HttpUser, task, between
from locust.env import Environment
from locust.stats import stats_printer
from locust.log import setup_logging
setup_logging("INFO", None)
class User(HttpUser):
wait_time = between(1, 3)
host = "https://docs.locust.io"
@task
def my_task(self):
self.client.get("/")
@task
def task_404(self):
self.client.get("/non-existing-path")
# setup Environment and Runner
# 設(shè)置 Environment 和 Runner
env = Environment(user_classes=[User])
env.create_local_runner()
# start a WebUI instance
# 啟動(dòng) WebUI 實(shí)例
env.create_web_ui("127.0.0.1", 8089)
# start a greenlet that periodically outputs the current stats
# 啟動(dòng) greenlet 定期輸出當(dāng)前的統(tǒng)計(jì)信息
gevent.spawn(stats_printer(env.stats))
# start the test
# 啟動(dòng)負(fù)載測試
env.runner.start(1, hatch_rate=10)
# in 60 seconds stop the runner
# 運(yùn)行 60s 后停止運(yùn)行
gevent.spawn_later(60, lambda: env.runner.quit())
# wait for the greenlets
# 等待 greenlets
env.runner.greenlet.join()
# stop the web server for good measures
# 采取良好的措施停止 web server
env.web_ui.stop()
