Propagate Request Latency

src/script.c

@@ -164,7 +164,7 @@ void script_request(lua_State *L, char **buf, size_t *len) {
     lua_pop(L, pop);
 }
 
-void script_response(lua_State *L, int status, buffer *headers, buffer *body) {
+void script_response(lua_State *L, int status, buffer *headers, buffer *body, uint64_t req_latency) {
     lua_getglobal(L, "response");
     lua_pushinteger(L, status);
     lua_newtable(L);
@@ -176,7 +176,8 @@ void script_response(lua_State *L, int status, buffer *headers, buffer *body) {
     }
 
     lua_pushlstring(L, body->buffer, body->cursor - body->buffer);
-    lua_call(L, 3, 0);
+    lua_pushnumber(L, req_latency);
+    lua_call(L, 4, 0);
 
     buffer_reset(headers);
     buffer_reset(body);

src/script.h

@@ -18,7 +18,7 @@ void script_done(lua_State *, stats *, stats *);
 void script_init(lua_State *, thread *, int, char **);
 uint64_t script_delay(lua_State *);
 void script_request(lua_State *, char **, size_t *);
-void script_response(lua_State *, int, buffer *, buffer *);
+void script_response(lua_State *, int, buffer *, buffer *, uint64_t);
 size_t script_verify_request(lua_State *L);
 
 bool script_is_static(lua_State *);

src/wrk.c

@@ -336,7 +336,7 @@ static int response_complete(http_parser *parser) {
 
     if (c->headers.buffer) {
         *c->headers.cursor++ = '\0';
-        script_response(thread->L, status, &c->headers, &c->body);
+        script_response(thread->L, status, &c->headers, &c->body, now - c->start);
         c->state = FIELD;
     }