在 k6 中重试 http 请求

Question

我有一套基于 python requests 的 API 测试套件，可以自动重试每个请求并返回 408 或 5xx 响应。我正在考虑在 k6 中重新实现其中一些以进行负载测试。 k6 是否支持重试 http 请求？

Answer 1

k6 中没有这样的功能，但是您可以通过包装 k6/http 函数来相当简单地添加它，例如：

function httpGet(url, params) {
    var res; 
    for (var retries = 3; retries > 0; retries--) {
        res = http.get(url, params)
        if (res.status != 408 && res.status < 500) {
            return res;
        }
    }
    return res;

}

然后只需使用 httpGet 而不是 http.get ;)

Answer 2

您可以创建一个可重用的重试函数并将其放入一个模块中，以供您的测试脚本导入。

函数可以通用：

期待函数重试
期望一个谓词来区分成功和失败的调用
重试上限

function retry(limit, fn, pred) {
  while (limit--) {
    let result = fn();
    if (pred(result)) return result;
  }
  return undefined;
}

然后在调用时提供正确的参数：

retry(
  3,
  () => http.get('http://example.com'),
  r => !(r.status == 408 || r.status >= 500));

当然，可以随意将其包装在一个或多个更具体的函数中：

function get3(url) {
  return request3(() => http.get(url));
}

function request3(req) {
  return retry(
    3,
    req,
    r => !(r.status == 408 || r.status >= 500));
}

let getResponse = get3('http://example.com');
let postResponse = request3(() => http.post(
  'https://httpbin.org/post',
  'body',
  { headers: { 'content-type': 'text/plain' } });

好处：您可以通过实现一个巧妙命名的函数来使调用代码更具表现力，该函数会反转其结果，而不是使用否定运算符：

function when(pred) {
  return x => !pred(x);
}

然后

retry(
  3,
  () => http.get('http://example.com'),
  when(r => r.status == 408 || r.status >= 500));

或者完全改变谓词的行为并测试失败的请求而不是成功的请求：

function retry(fn, pred, limit) {
  while (limit--) {
    let result = fn();
    if (!pred(result)) return result;
  }
  return undefined;
}

function unless(pred) {
  return x => !pred(x);
}

retry(
  3,
  () => http.get('http://example.com'),
  r => r.status == 408 || r.status >= 500);
retry(
  3,
  () => http.get('http://example.com'),
  unless(r => r.status != 408 && r.status < 500));

在 k6 中重试 http 请求

Retry http requests in k6

request

k6