64 lines
1.8 KiB
JavaScript
64 lines
1.8 KiB
JavaScript
import http from 'k6/http';
|
|
import { check, sleep } from 'k6';
|
|
import encoding from 'k6/encoding';
|
|
|
|
/**
|
|
* ST-COUNT-01 — Lotes concurrentes hacia count-api vía gateway.
|
|
* En host compartido 8GiB: usar PROFILE=shared8g y liberar MLflow/Airflow antes.
|
|
*
|
|
* k6 run -e PROFILE=shared8g k6_count_batches.js
|
|
*/
|
|
const PROFILE = __ENV.PROFILE || 'shared8g';
|
|
const BASE_URL = __ENV.BASE_URL || 'http://127.0.0.1:8189';
|
|
|
|
const profiles = {
|
|
shared8g: {
|
|
rate: 1,
|
|
duration: '1m',
|
|
preAllocatedVUs: 2,
|
|
maxVUs: 5,
|
|
},
|
|
lab: {
|
|
rate: 5,
|
|
duration: '2m',
|
|
preAllocatedVUs: 10,
|
|
maxVUs: 20,
|
|
},
|
|
};
|
|
|
|
const cfg = profiles[PROFILE] || profiles.shared8g;
|
|
|
|
export const options = {
|
|
scenarios: {
|
|
batches: {
|
|
executor: 'constant-arrival-rate',
|
|
rate: cfg.rate,
|
|
timeUnit: '1s',
|
|
duration: cfg.duration,
|
|
preAllocatedVUs: cfg.preAllocatedVUs,
|
|
maxVUs: cfg.maxVUs,
|
|
},
|
|
},
|
|
thresholds: {
|
|
http_req_duration: ['p(95)<60000'],
|
|
checks: ['rate>0.85'],
|
|
},
|
|
tags: { profile: PROFILE },
|
|
};
|
|
|
|
const tinyPdf = encoding.b64decode(
|
|
'JVBERi0xLjAKMSAwIG9iago8PAovVHlwZSAvQ2F0YWxvZwovUGFnZXMgMiAwIFIKPj4KZW5kb2JqCjIgMCBvYmoKPDwKL1R5cGUgL1BhZ2VzCi9LaWRzIFszIDAgUl0KL0NvdW50IDEKPD4KZW5kb2JqCjMgMCBvYmoKPDwKL1R5cGUgL1BhZ2UKL1BhcmVudCAyIDAgUgovTWVkaWFCb3ggWzAgMCAzMDAgMTQ0XQo+PgplbmRvYmoKeHJlZgowIDQKMDAwMDAwMDAwMCA2NTUzNSBmIAowMDAwMDAwMDA5IDAwMDAwIG4gCjAwMDAwMDAwNjMgMDAwMDAgbiAKMDAwMDAwMDEyNCAwMDAwMCBuIAp0cmFpbGVyCjw8Ci9TaXplIDQKL1Jvb3QgMSAwIFIKPj4Kc3RhcnR4cmVmCjIwMQolJUVPRgo=',
|
|
);
|
|
|
|
export default function () {
|
|
const bin = __ENV.PDF_B64 ? encoding.b64decode(__ENV.PDF_B64) : tinyPdf;
|
|
const data = {
|
|
file: http.file(bin, 'sample.pdf', 'application/pdf'),
|
|
};
|
|
const res = http.post(`${BASE_URL}/count/v1/count`, data);
|
|
check(res, {
|
|
'not 5xx': (r) => r.status < 500,
|
|
});
|
|
sleep(0.1);
|
|
}
|