kv.js

import { sleep } from "k6";
import { openKv } from "k6/x/kv";

export const options = {
    scenarios: {
        poll: {
            exec: "poll",
            executor: "constant-vus",
            vus: 1,
            duration: "20s",
        },
        write: {
            exec: "write",
            executor: "ramping-vus",
            stages: [
                { duration: '5s', target: 5 },
                { duration: '10s', target: 5 },
                { duration: '5s', target: 0 },
            ],
        },
    },
};

const polling_kv = openKv();

export async function setup() {
    await polling_kv.clear();
}

export async function poll() {
    const entries = await polling_kv.list({ prefix: "request_" });
    for (const entry of entries) {
        console.log(`poll request (iteration_${__ITER}): ${entry.key}`);
        polling_kv.delete(entry.key);
    }
}

export async function write() {
    // If request takes more than X seconds -> Send it to the polling queue
    await polling_kv.set(`request_${__VU}_${__ITER}`, "value");
    sleep(1);
}