import random
from locust import HttpUser, task, between

class APIUser(HttpUser):
    # 设置模拟用户之间的等待时间（例如：每个用户之间随机等待 1-3 秒）
    wait_time = between(1, 3)

    # 定义一个包含多个章节 ID 的列表
    chapter_ids = ["1", "3", "6", "8", "15"]

    @task
    def get_chapter_relations(self):
        # 从章节 ID 列表中随机选择一个章节 ID
        chapter_id = random.choice(self.chapter_ids)
        self.client.get(f"/chapters/{chapter_id}/relations")

    @task
    def get_all_relations(self):
        self.client.get("/chapters/relations/")

    @task
    def get_relations_by_level(self):
        level = random.randint(1, 5)  # 随机选择一个层级，范围为 1 到 5
        self.client.get(f"/chapters/relations/level/{level}")