如何在locust中执行关联

from locust import HttpUser, task, between
import re
import json

class WebsiteUser(HttpUser):
    wait_time = between(1, 5)

    def on_start(self):
        self.token = None

    @task
    def get_token(self):
        response = self.client.get("/get-token")
        if response.status_code == 200:
            # 使用正则表达式从响应文本中提取token
            match = re.search(r'token=(\w+)', response.text)
            if match:
                self.token = match.group(1)
            else:
                raise Exception("Token not found in response")

    @task
    def use_token(self):
        if not self.token:
            raise Exception("Token is not set")
        
        headers = {"Authorization": f"Bearer {self.token}"}
        response = self.client.get("/use-token", headers=headers)
        if response.status_code != 200:
            raise Exception("Failed to use token")

# 运行Locust
# locust -f your_script.py --host=http://example.com