From fede106118889a4715a2572fd572f9481419e680 Mon Sep 17 00:00:00 2001
From: Ben Bierens <39762930+benbierens@users.noreply.github.com>
Date: Tue, 3 Oct 2023 11:53:58 +0200
Subject: [PATCH] Numbered log entries (#559)

* add numbered log entries (hack)

* Changed from CLI argument to compiler define

* Use uint64 for counting

---------

Co-authored-by: gmega <giuliano.mega@gmail.com>
---
 .github/workflows/docker-dist-tests.yml |  2 +-
 codex/conf.nim                          | 13 ++++++++++++-
 2 files changed, 13 insertions(+), 2 deletions(-)

diff --git a/.github/workflows/docker-dist-tests.yml b/.github/workflows/docker-dist-tests.yml
index ee25cd10..40dbacfd 100644
--- a/.github/workflows/docker-dist-tests.yml
+++ b/.github/workflows/docker-dist-tests.yml
@@ -24,7 +24,7 @@ jobs:
     name: Build and Push
     uses: ./.github/workflows/docker-reusable.yml
     with:
-      nimflags: '-d:disableMarchNative -d:codex_enable_api_debug_peers=true -d:codex_enable_simulated_proof_failures'
+      nimflags: '-d:disableMarchNative -d:codex_enable_api_debug_peers=true -d:codex_enable_simulated_proof_failures -d:codex_enable_log_counter=true'
       nat_ip_auto: true
       tag_latest: ${{ github.ref_name == github.event.repository.default_branch || startsWith(github.ref, 'refs/tags/') }}
       tag_suffix: dist-tests
diff --git a/codex/conf.nim b/codex/conf.nim
index d9488076..bf7b9ec2 100644
--- a/codex/conf.nim
+++ b/codex/conf.nim
@@ -42,6 +42,7 @@ export net, DefaultQuotaBytes, DefaultBlockTtl, DefaultBlockMaintenanceInterval,
 const
   codex_enable_api_debug_peers* {.booldefine.} = false
   codex_enable_proof_failures* {.booldefine.} = false
+  codex_enable_log_counter* {.booldefine.} = false
 
 type
   StartUpCommand* {.pure.} = enum
@@ -463,7 +464,7 @@ proc setupLogging*(conf: CodexConf) =
 
     defaultChroniclesStream.outputs[1].writer = noOutput
 
-    defaultChroniclesStream.outputs[0].writer =
+    let writer =
       case conf.logFormat:
       of LogKind.Auto:
         if isatty(stdout):
@@ -478,6 +479,16 @@ proc setupLogging*(conf: CodexConf) =
       of LogKind.None:
         noOutput
 
+    when codex_enable_log_counter:
+      var counter = 0.uint64
+      proc numberedWriter(logLevel: LogLevel, msg: LogOutputStr) =
+        inc(counter)
+        let withoutNewLine = msg[0..^2]
+        writer(logLevel, withoutNewLine & " count=" & $counter & "\n")
+      defaultChroniclesStream.outputs[0].writer = numberedWriter
+    else:
+      defaultChroniclesStream.outputs[0].writer = writer
+
   try:
     updateLogLevel(conf.logLevel)
   except ValueError as err: