AWS CDK で Graviton EC2 に k6 負荷テスト環境を構築する

k6-test/
├── bin/
│   └── k6-test.ts
├── lib/
│   ├── k6-test-stack.ts  # CDKスタック定義
│   └── init.sh           # UserDataスクリプト
└── package.json

// init.shを読み込み
const userDataScript = fs.readFileSync(path.join(__dirname, "init.sh"), "utf8");

const userData = ec2.UserData.forLinux();
userData.addCommands(userDataScript);

// EC2インスタンスの作成
new ec2.Instance(this, "K6Instance", {
  vpc,
  vpcSubnets: {
    subnetType: ec2.SubnetType.PRIVATE_WITH_EGRESS,
  },
  instanceType: ec2.InstanceType.of(
    ec2.InstanceClass.T4G,
    ec2.InstanceSize.MEDIUM
  ),
  machineImage: ec2.MachineImage.latestAmazonLinux2023({
    cpuType: ec2.AmazonLinuxCpuType.ARM_64,
  }),
  securityGroup: instanceSecurityGroup,
  userData,
});

#!/bin/bash
set -e

# ============================================
# k6のインストール (ARM64対応)
# ============================================
# k6の公式 RPM は x86_64 のみに対応しているため、ARM64 ではバイナリを直接ダウンロードする
K6_VERSION=$(curl -s https://api.github.com/repos/grafana/k6/releases/latest | grep -o '"tag_name": "[^"]*' | cut -d'"' -f4)
curl -L "https://github.com/grafana/k6/releases/download/${K6_VERSION}/k6-${K6_VERSION}-linux-arm64.tar.gz" -o /tmp/k6.tar.gz
tar -xzf /tmp/k6.tar.gz -C /tmp
mv /tmp/k6-${K6_VERSION}-linux-arm64/k6 /usr/local/bin/k6
chmod +x /usr/local/bin/k6
rm -rf /tmp/k6.tar.gz /tmp/k6-${K6_VERSION}-linux-arm64

# ============================================
# Node.jsのインストール
# ============================================
# TypeScript 環境を構築するために必要な Node.js と npm をインストール
dnf install -y nodejs npm

# ============================================
# プロジェクトのセットアップ
# ============================================
PROJECT_DIR="/home/ec2-user/k6-project"
mkdir -p "$PROJECT_DIR"
cd "$PROJECT_DIR"

# package.json のセットアップ
npm init -y

# TypeScript 開発に必要なライブラリをインストール
npm install --save-dev typescript @types/k6 esbuild

# ビルドスクリプトの設定
# esbuild を使って TypeScript を k6 で動く JavaScript (ES2015) に変換
npm pkg set scripts.build="esbuild *.ts --bundle --target=es2015 --platform=node --outdir=dist --external:k6 --external:k6/*"

# ============================================
# 権限修正
# ============================================
# UserDataはroot権限で実行されるため、ec2-userがアクセスできるように所有者を変更
chown -R ec2-user:ec2-user "$PROJECT_DIR"

dnf install -y https://dl.k6.io/rpm/repo.rpm
dnf install -y k6

pnpx cdk deploy

$ k6 --version
k6 v1.5.0 (commit/7961cefa12, go1.25.5, linux/arm64)

$ cd k6-project

$ vi test.ts

import http from "k6/http";
import { check, sleep } from "k6";

export const options = {
  vus: 10,
  duration: "30s",
};

export default function () {
  const res = http.get("https://test.k6.io");
  check(res, {
    "status is 200": (r) => r.status === 200,
  });
  sleep(1);
}

$ npm run build
> k6-project@1.0.0 build
> esbuild *.ts --bundle --target=es2015 --platform=node --outdir=dist --external:k6 --external:k6/*

  dist/test.js  1.9kb

⚡ Done in 3ms

k6 run dist/test.js