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

// テスト対象のAPI URL（環境変数から取得、またはデフォルト値を使用）
const API_URL = __ENV.API_URL || 'https://xxxxxxxxx.execute-api.ap-northeast-1.amazonaws.com/v1';

/**
 * K6テストオプション設定
 * 
 * テスト対象: GET /users（ユーザー一覧取得API）
 * 
 * システム制約:
 * - Lambda予約済み同時実行数: 960
 * - 想定レスポンスタイム: 300ms
 * - 理論上の最大RPS: 960 ÷ 0.3秒 ≈ 3,200 RPS
 * 
 * テストの目的:
 * 1. 本番想定負荷（1,800 RPS）での安定性確認
 * 2. Lambda同時実行数制限でのスロットル挙動確認
 * 3. システムの限界性能測定
 */
export const options = {
  scenarios: {
    /**
     * シナリオ1: 同時接続数テスト
     * 
     * 目的: 一定数の同時接続を維持できるか確認
     * 実行時間: 0分〜2分
     * VU数: 500（仮想ユーザー数）
     * 期待される挙動: 安定したレスポンスタイムを維持
     */
    constant_vus: {
      executor: 'constant-vus',        // 一定のVU数を維持
      vus: 500,                        // 500同時接続
      duration: '2m',                  // 2分間実行
      startTime: '0s',                 // テスト開始直後から実行
    },

    /**
     * シナリオ2: 目標RPSテスト
     * 
     * 目的: 本番想定の1,800 RPSを維持できるか確認
     * 実行時間: 2分〜5分
     * RPS: 1,800リクエスト/秒
     * 期待される挙動: エラー率2%未満、P95 < 500ms
     */
    target_rps: {
      executor: 'constant-arrival-rate', // 一定のRPSを維持
      rate: 1800,                        // 1,800リクエスト/秒
      timeUnit: '1s',                    // 1秒あたり
      duration: '3m',                    // 3分間実行
      preAllocatedVUs: 100,              // 事前に割り当てるVU数
      maxVUs: 2000,                      // 最大VU数（負荷に応じて自動調整）
      startTime: '2m',                   // シナリオ1終了後に開始
    },

    /**
     * シナリオ3: 限界性能テスト（段階的負荷増加）
     * 
     * 目的: システムの限界RPSとボトルネックを特定
     * 実行時間: 5分〜12分
     * RPS: 100 → 3,500（段階的に増加）
     * 
     * 期待される挙動:
     * - 3,200 RPS付近でLambda同時実行数が960に到達
     * - 3,500 RPSでスロットル（429エラー）が発生
     */
    ramping_rps: {
      executor: 'ramping-arrival-rate',  // RPSを段階的に増加
      startRate: 100,                    // 開始時のRPS
      timeUnit: '1s',
      preAllocatedVUs: 100,
      maxVUs: 3000,                      // 最大3,000 VU（Lambda制約に合わせる）
      stages: [
        { duration: '1m', target: 500 },    // 1分で500 RPSまで増加
        { duration: '1m', target: 1000 },   // 1分で1,000 RPSまで増加
        { duration: '1m', target: 1800 },   // 1分で1,800 RPSまで増加（目標値）
        { duration: '1m', target: 2500 },   // 1分で2,500 RPSまで増加
        { duration: '1m', target: 3200 },   // 1分で3,200 RPSまで増加（Lambda理論上の限界）
        { duration: '1m', target: 3500 },   // 1分で3,500 RPSまで増加（スロットル発生を確認）
        { duration: '1m', target: 0 },      // 1分でクールダウン
      ],
      startTime: '5m',                   // シナリオ2終了後に開始
    },
  },

  /**
   * 閾値（Thresholds）設定
   * 
   * これらの条件を満たさない場合、テストは失敗とみなされます。
   * 
   * - http_req_duration: レスポンスタイム
   *   - P95（95パーセンタイル）: 500ms以下
   *   - P99（99パーセンタイル）: 1,000ms以下
   * 
   * - http_req_failed: HTTPリクエスト失敗率
   *   - 2%未満（Lambda スロットル時は増加する）
   */
  thresholds: {
    http_req_duration: ['p(95)<500', 'p(99)<1000'],
    http_req_failed: ['rate<0.02'],
  },
};

/**
 * メインテスト関数
 * 
 * 各仮想ユーザー（VU）が繰り返し実行する処理。
 * k6がシナリオ設定に従って、この関数を並列実行します。
 */
export default function () {
  // GET /users リクエストを送信（ユーザー一覧取得）
  const res = http.get(`${API_URL}/users`);

  /**
   * レスポンスの検証
   * 
   * check関数で複数の条件をチェックし、結果をk6のメトリクスに記録します。
   * 
   * チェック項目:
   * 1. ステータスコードが200（正常）
   * 2. ステータスコードが429でない（スロットルされていない）
   * 3. レスポンスタイムが500ms未満
   */
  check(res, {
    'status is 200': (r) => r.status === 200,
    'status is not 429 (throttled)': (r) => r.status !== 429,
    'response time < 500ms': (r) => r.timings.duration < 500,
  });

  /**
   * 次のリクエストまで0.1秒待機
   */
  sleep(0.1);
}