Skip to main content Link Search Menu Expand Document (external link)
Backend-KR

Table of contents

  1. Locust Install
  2. Run Locust
    1. 옵션 설명
    2. 옵션 적용 후 초기 실행 화면
  3. 스크립트 작성
    1. 코드 설명

Locust Install

pip install locust

Run Locust

8089 port로 기본적으로 열립니다.

locust -f <경로>/<파일 이름>.py

현재 위치에 locustfile.py 라는 파일이 존재하면

locust

–host 등 커맨드 옵션을 알고 싶다면

locust --help

–host 등 커맨드 옵션 적용해보기

locust -f sample.py --host http://172.16.0.253:8081 --users 4 --spawn-rate 4

옵션 설명

[host]

실행할 때 detault로 들어갈 주소를 설정할 수 있습니다.

[users]

실행할 때 detault로 트래픽을 만들어내는 유저 수를 설정할 수 있습니다.

[spawn-rate]

실행할 때 detault로 몇초에 한번씩 유저가 추가되는지 설정할 수 있습니다.

옵션 적용 후 초기 실행 화면

  • 상위 host, user, spawn-rate를 사용하지 않을 시 실행할때마다 입력을 해야하는 번거로움이 있을 수 있다.

스크립트 작성

class AccountBehavior(TaskSet):
    @task
    def get_account_number_detail(self):
        account_number = 123445152
        self.client.get(f'/asset/{account_number}')

    @task(weight=2)
    def home(self):
        self.client.get('/')

    def on_start(self):
        print("START LOCUST")

    def on_stop(self):
        print("STOP LOCUST")


class LocustUser(HttpUser):
    host = "https://locust.load-test.com"
    tasks = [AccountBehavior]
    wait_time = between(1, 4)

코드 설명

[HttpUser]

서버에 부하를 가할 주체를 나타낸다. tasks 애트리뷰트에 선언된 작업 또는 @task 데코레이터가 붙여진 작업를 수행한다.

[TaskSet]

유저가 수행할 작업들을 하나의 클래스로 만들어, tasks 애트리뷰트에 선언된 작업들 또는 @task 데코레이터가 붙여진 작업들 중 랜덤으로 수행한다.

[task]

HttpUser는 @task 데코레이터가 붙은 메서드를 찾아 수행하게 된다. weight 파라미터를 넣어주면 각 테스크마다 가중치를 부여할 수 있다. home이 get_account_number_detail보다 실행될 확률이 2배나 된다.

[between]

HttpUser나 TaskSet의 wait_time 애트리 뷰트를 사용할 때 해당 함수를 사용할 수 있다. wait_time = between(1, 4) 선언해주면 1초 ~ 4초 사이 간격으로 랜덤하게 작업이 수행된다는 뜻이다.

[on_start]

Locust가 실행 후 처리할 로직을 처리합니다.

[on_stop]

Locust가 종료 후 처리할 로직을 처리합니다.

[host]

상위에서 Locust를 실행할때 옵션인 –host 옵션을 코드상에서 주입하게 해준다.