From 502c9a750ccaa4022a4339e7c9f4202e617ac6e7 Mon Sep 17 00:00:00 2001
From: Ethan Shold <sholdee@gmail.com>
Date: Fri, 21 Jun 2024 23:53:53 -0500
Subject: [PATCH] stdout flush

---
 adguard_exporter.py | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/adguard_exporter.py b/adguard_exporter.py
index 4e5265b..6695a6f 100755
--- a/adguard_exporter.py
+++ b/adguard_exporter.py
@@ -1,6 +1,7 @@
 import time
 import json
 import os
+import sys
 from prometheus_client import start_http_server, Counter
 
 # Define a single Prometheus metric
@@ -15,27 +16,32 @@ def get_last_position():
             pos = int(f.read().strip())
             inode = os.stat(log_file_path).st_ino
             print(f"Read last position: {pos}, inode: {inode}")
+            sys.stdout.flush()
             return pos, inode
     except (FileNotFoundError, ValueError) as e:
         print(f"Error reading last position: {e}")
+        sys.stdout.flush()
         return 0, None
 
 def save_last_position(pos, inode):
     with open(position_file_path, 'w') as f:
         f.write(f"{pos}\n{inode}")
     print(f"Saved position: {pos}, inode: {inode}")
+    sys.stdout.flush()
 
 def read_new_lines(file, start_pos):
     file.seek(start_pos)
     lines = file.readlines()
     new_pos = file.tell()
     print(f"Read {len(lines)} new lines, new position: {new_pos}")
+    sys.stdout.flush()
     return lines, new_pos
 
 def reset_metrics():
     # Reset the Counter metrics
     dns_queries._metrics.clear()
     print("Metrics reset")
+    sys.stdout.flush()
 
 def parse_and_export(lines):
     for line in lines:
@@ -52,20 +58,25 @@ def parse_and_export(lines):
                     upstream=data.get('Upstream', 'unknown')
                 ).inc()
                 print(f"Exported metric for query: {data.get('QH', 'unknown')}")
+                sys.stdout.flush()
             except json.JSONDecodeError as e:
                 print(f"Error decoding JSON: {e}, line: {line}")
+                sys.stdout.flush()
 
 if __name__ == '__main__':
     # Start the Prometheus metrics server
     start_http_server(8000)
     print("Prometheus metrics server started on port 8000")
+    sys.stdout.flush()
 
     # Wait for the log file to exist
     while not os.path.exists(log_file_path):
         print(f"Waiting for {log_file_path} to be created...")
+        sys.stdout.flush()
         time.sleep(10)
 
     print(f"Log file {log_file_path} found")
+    sys.stdout.flush()
 
     # Get the last read position and inode
     last_position, last_inode = get_last_position()
@@ -79,6 +90,7 @@ def parse_and_export(lines):
                 last_position = 0
                 reset_metrics()
                 print(f"Log file rotated, resetting position to {last_position}")
+                sys.stdout.flush()
 
             with open(log_file_path, 'r') as log_file:
                 new_lines, new_position = read_new_lines(log_file, last_position)
@@ -92,6 +104,7 @@ def parse_and_export(lines):
 
         except Exception as e:
             print(f"Error during processing: {e}")
+            sys.stdout.flush()
 
         # Sleep for a while before reading the log again
         time.sleep(10)