From 4c2ee640fe7e43ca3f578eafec0b1fef8e6d63dd Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Lauri=20V=C3=B5sandi?= <lauri@k-space.ee>
Date: Mon, 7 Nov 2022 21:50:00 +0200
Subject: [PATCH] Add buffer size metrics

---
 log_shipper.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/log_shipper.py b/log_shipper.py
index 1280a3e..bc6e904 100755
--- a/log_shipper.py
+++ b/log_shipper.py
@@ -65,6 +65,9 @@ host_info = {
 
 log_files = dict()
 
+gauge_buffer_size = Gauge(
+    "logmower_buffer_size_bytes",
+    "Log files buffered in memory")
 gauge_backlog_size = Gauge(
     "logmower_backlog_size_bytes",
     "Content that is yet to be submitted")
@@ -498,6 +501,7 @@ async def watcher(loop, queue, coll):
 
 @app.route("/metrics")
 async def handler(request):
+    gauge_buffer_size.set(sum([len(j.buf) for j in log_files.values()]))
     gauge_backlog_size.set(sum([j.tail - j.head for j in log_files.values()]))
     c = collections.Counter([j.state for j in log_files.values()])
     for key in ("seeking", "replaying", "watching", "closing"):