From a88e857b5b4100bb62be8a73c1badd999561b328 Mon Sep 17 00:00:00 2001
From: Andrei Zeliankou <zelenkov@nginx.com>
Date: Wed, 8 Nov 2023 17:34:59 +0000
Subject: Var: $request_id variable.

This variable contains a string that is formed using random data and
can be used as a unique request identifier.

This closes #714 issue on GitHub.
---
 src/nxt_http_variables.c | 32 ++++++++++++++++++++++++++++++++
 1 file changed, 32 insertions(+)

(limited to 'src')

diff --git a/src/nxt_http_variables.c b/src/nxt_http_variables.c
index 46594a6b..ec744317 100644
--- a/src/nxt_http_variables.c
+++ b/src/nxt_http_variables.c
@@ -28,6 +28,8 @@ static u_char *nxt_http_log_date(u_char *buf, nxt_realtime_t *now,
     struct tm *tm, size_t size, const char *format);
 static nxt_int_t nxt_http_var_request_line(nxt_task_t *task, nxt_str_t *str,
     void *ctx, void *data);
+static nxt_int_t nxt_http_var_request_id(nxt_task_t *task, nxt_str_t *str,
+    void *ctx, void *data);
 static nxt_int_t nxt_http_var_status(nxt_task_t *task, nxt_str_t *str,
     void *ctx, void *data);
 static nxt_int_t nxt_http_var_body_bytes_sent(nxt_task_t *task, nxt_str_t *str,
@@ -89,6 +91,10 @@ static nxt_var_decl_t  nxt_http_vars[] = {
         .name = nxt_string("request_line"),
         .handler = nxt_http_var_request_line,
         .cacheable = 1,
+    }, {
+        .name = nxt_string("request_id"),
+        .handler = nxt_http_var_request_id,
+        .cacheable = 1,
     }, {
         .name = nxt_string("status"),
         .handler = nxt_http_var_status,
@@ -395,6 +401,32 @@ nxt_http_var_request_line(nxt_task_t *task, nxt_str_t *str, void *ctx,
 }
 
 
+static nxt_int_t
+nxt_http_var_request_id(nxt_task_t *task, nxt_str_t *str, void *ctx,
+    void *data)
+{
+    nxt_random_t        *rand;
+    nxt_http_request_t  *r;
+
+    r = ctx;
+
+    str->start = nxt_mp_nget(r->mem_pool, 32);
+    if (nxt_slow_path(str->start == NULL)) {
+        return NXT_ERROR;
+    }
+
+    str->length = 32;
+
+    rand = &task->thread->random;
+
+    (void) nxt_sprintf(str->start, str->start + 32, "%08xD%08xD%08xD%08xD",
+                       nxt_random(rand), nxt_random(rand),
+                       nxt_random(rand), nxt_random(rand));
+
+    return NXT_OK;
+}
+
+
 static nxt_int_t
 nxt_http_var_body_bytes_sent(nxt_task_t *task, nxt_str_t *str, void *ctx,
     void *data)
-- 
cgit