TIL | [부하테스트] Artillery 로 Redis 테스트하기

Artillery는 간단하고 유연한 부하 테스팅 도구로, HTTP, WebSocket, Socket.io 등 다양한 프로토콜을 지원합니다. Redis와 같은 데이터베이스에 대한 부하 테스트를 수행하려면, Artillery가 직접 지원하지 않는 프로토콜이라도 커스텀 스크립트를 작성하여 테스트를 진행할 수 있습니다. 다음은 Redis 부하 테스트를 위해 Artillery를 사용하는 방법에 대한 개요입니다.

1단계: Artillery 설치

먼저, Artillery를 글로벌로 설치해야 합니다. 터미널에서 다음 명령어를 실행합니다:

npm install -g artillery

2단계: Artillery 플러그인 또는 커스텀 스크립트 준비

Redis 테스트를 위해서는 직접적인 지원이 없기 때문에, Artillery의 커스텀 JavaScript 기능을 사용하거나, Redis를 위한 플러그인(있을 경우)을 사용해야 합니다. 커스텀 JavaScript 테스트 스크립트를 작성할 수 있으며, 이 스크립트 내에서 Redis 클라이언트 라이브러리를 사용하여 Redis 명령을 실행할 수 있습니다.

3단계: 커스텀 스크립트 작성

Artillery는 beforeRequest 및 afterResponse와 같은 훅을 제공하여 커스텀 로직을 실행할 수 있습니다. 그러나 Redis 부하 테스트를 위해서는 더 많은 커스텀 작업이 필요할 수 있습니다. 예를 들어, 다음은 Node.js에서 Redis 클라이언트를 사용하는 간단한 예시입니다:

const redis = require('redis');
const client = redis.createClient({ /* Redis 설정 */ });

client.on('error', (err) => console.log('Redis Client Error', err));

async function testRedis() {
  await client.connect();

  // Redis 명령 실행 예시
  await client.set('key', 'value');
  const value = await client.get('key');

  console.log(value);

  await client.quit();
}

testRedis();

이 예시처럼, Artillery의 beforeScenario, afterScenario, beforeRequest, afterRequest 훅을 사용하여 Redis 연결을 관리하고, 명령을 실행하는 커스텀 로직을 작성할 수 있습니다. 그러나 Artillery의 기본적인 HTTP 요청 흐름 외에 이런 종류의 테스트를 수행하기 위해서는 Artillery 스크립트 구성보다는 순수 Node.js 스크립트가 더 적합할 수 있습니다.

4단계: 테스트 실행

커스텀 스크립트를 준비한 후, Node.js를 통해 스크립트를 실행하여 Redis에 대한 부하 테스트를 수행할 수 있습니다. Artillery의 --overrides 옵션을 사용하여 실행 시간에 스크립트 변수를 조정할 수도 있습니다.