From f76d634ac693291583b6d8b4dc5d1f871e6e874b Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?G=C3=A1bor=20Farkas?= <gabor.farkas@gmail.com>
Date: Fri, 10 Mar 2023 16:09:59 +0100
Subject: [PATCH] devenv: loki: generate timestamps with various precision
 (#64509)

---
 devenv/docker/blocks/loki/data/data.js | 28 ++++++++++++++++++++------
 1 file changed, 22 insertions(+), 6 deletions(-)

diff --git a/devenv/docker/blocks/loki/data/data.js b/devenv/docker/blocks/loki/data/data.js
index 7ab907c0797..8462deed770 100644
--- a/devenv/docker/blocks/loki/data/data.js
+++ b/devenv/docker/blocks/loki/data/data.js
@@ -47,10 +47,7 @@ async function sleep(duration) {
   });
 }
 
-async function lokiSendLogLine(timestampMs, line, tags) {
-  // we keep nanosecond-timestamp in a string because
-  // as a number it would be too large
-  const timestampNs = `${timestampMs}000000`;
+async function lokiSendLogLine(timestampNs, line, tags) {
   const data = {
     streams: [
       {
@@ -136,6 +133,24 @@ function calculateDelays(pointsCount) {
   return delays;
 }
 
+function getRandomNanosecPart() {
+  // we want to have cases with milliseconds-only, with microsec and nanosec.
+  const mode = Math.random();
+
+  if (mode < 0.333) {
+    // only milisec precision
+    return '000000';
+  }
+
+  if (mode < 0.666) {
+    // microsec precision
+    return Math.trunc(Math.random()*1000).toString().padStart(3, '0') + '000'
+  }
+
+  // nanosec precision
+  return Math.trunc(Math.random()*1000000).toString().padStart(6, '0')
+}
+
 
 async function main() {
   const delays = calculateDelays(DAYS * POINTS_PER_DAY);
@@ -144,9 +159,10 @@ async function main() {
   for(let i =0; i < delays.length; i++ ) { // i cannot do a forEach because of the `await` inside
     const delay = delays[i];
     timestampMs += Math.trunc(delay * timeRange);
+    const timestampNs = `${timestampMs}${getRandomNanosecPart()}`;
     const item = getRandomLogItem(i + 1)
-    await lokiSendLogLine(timestampMs, JSON.stringify(item), {place:'moon', source: 'data', instance: 'server\\1', job: '"grafana/data"'});
-    await lokiSendLogLine(timestampMs, logFmtLine(item), {place:'luna', source: 'data', instance: 'server\\2', job: '"grafana/data"'});
+    await lokiSendLogLine(timestampNs, JSON.stringify(item), {place:'moon', source: 'data', instance: 'server\\1', job: '"grafana/data"'});
+    await lokiSendLogLine(timestampNs, logFmtLine(item), {place:'luna', source: 'data', instance: 'server\\2', job: '"grafana/data"'});
   };
 }