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

/**
 * 기본 설정
 *  - BASE_URL       : 요청을 보낼 대상 서버 (환경 변수로 덮어쓰기 가능)
 *  - RATE           : 초당 요청 수(TPS). 미설정 시 100 TPS
 *  - TEST_DURATION  : 테스트 수행 시간. 미설정 시 24h
 */
//const BASE_URL = __ENV.BASE_URL || 'http://10.0.1.8:24002';
const BASE_URL = __ENV.BASE_URL || 'http://127.0.0.1:24002';  // ← 기본값
const RATE = __ENV.RATE ? Number(__ENV.RATE) : 30;
const TEST_DURATION = __ENV.TEST_DURATION || '10m';
//const TEST_DURATION = __ENV.TEST_DURATION || '24h';

// 불규칙 트래픽을 만들기 위한 API 경로와 가중치
// 예: orders 호출이 가장 빈번, shipments는 가장 드묾
const API_WEIGHTS = [
  { path: '/api/spans/orders',   weight: 4 },
  { path: '/api/spans/payments', weight: 4 },
  { path: '/api/spans/shipments', weight: 2 },
];

/**
 * 가중치 기반 랜덤 API 선택
 *  - weight 합을 기준으로 0 ~ totalWeight 사이 난수를 뽑아서
 *    해당 구간에 걸리는 path를 선택
 */
function pickRandomApi() {
  const totalWeight = API_WEIGHTS.reduce((sum, item) => sum + item.weight, 0);
  let rand = Math.random() * totalWeight;

  for (const item of API_WEIGHTS) {
    if (rand < item.weight) {
      return item.path;
    }
    rand -= item.weight;
  }

  // 방어적 코드: 논리상 이 지점은 오지 않아야 하지만,
  // 예외 상황 대비하여 첫 번째 항목을 반환
  return API_WEIGHTS[0].path;
}

/**
 * VU별로 유지되는 traceId
 *  - 각 VU마다 별도 값
 *  - default() 안에서 확률적으로 갱신
 */
let currentTraceId = `${__VU}-${Date.now()}`;

/**
 * k6 옵션
 *  - executor: constant-arrival-rate
 *    → 시나리오 전체 기준으로 RATE 값만큼 초당 iteration 실행 시도
 *  - rate     : 목표 TPS (초당 요청 수)
 *  - timeUnit : rate 기준 단위 (1초)
 *  - duration : 전체 테스트 시간
 *
 *  preAllocatedVUs / maxVUs
 *   - 응답 시간에 따라 필요한 동시 VU 수가 달라질 수 있으므로
 *     여유 있게 설정
 */
export const options = {
  scenarios: {
    span_scenario: {
      executor: 'constant-arrival-rate',
      rate: RATE,               // 목표 TPS (기본 100)
      timeUnit: '1s',
      duration: TEST_DURATION,  // 기본 24h, 필요 시 환경 변수로 조정

      // 초기에 확보할 VU 수
      preAllocatedVUs: 50,
      // 최대 허용 VU 수 (TPS가 안 나오면 이 값을 더 늘려볼 수 있음)
      maxVUs: 200,
    },
  },
  thresholds: {
    // 95퍼센타일 응답 시간 1.5초 미만
    http_req_duration: ['p(95)<1500'],
    // 실패율 1% 미만
    http_req_failed: ['rate<0.01'],
  },
};

export default function () {
  // 30% 확률로 traceId 변경 → 일정 기간 동안 동일 traceId로 묶인 요청 유지
  if (Math.random() < 0.3) {
    currentTraceId = `${__VU}-${Date.now()}`;
  }

  const path = pickRandomApi();

  const res = http.get(`${BASE_URL}${path}`, {
    headers: {
      // 백엔드/APM에서 추적용으로 사용할 커스텀 헤더
      'x-trace-id': currentTraceId,
    },
    tags: {
      // k6 메트릭(예: http_reqs)에 endpoint 태그가 붙도록 설정
      endpoint: path,
    },
  });

  // 엔드포인트별 상태 코드 검증
  check(res, {
    [`${path} status is 200`]: (r) => r.status === 200,
  });

  /**
   * iteration 사이에 약간의 지터 추가 (0 ~ 300ms)
   *
   * executor가 constant-arrival-rate 이기 때문에
   * 전체 TPS는 rate에 의해 제어되고,
   * 이 sleep은 개별 VU들의 타이밍을 살짝 어긋나게 만들어
   * 완전히 규칙적인 패턴을 피하는 정도의 역할만 수행한다.
   */
  sleep(Math.random() * 0.3);
}