!2 log to loki

scriptBase/base.py

@@ -1,6 +1,8 @@
 import builtins
 import datetime
-import scriptBase.loki_log
+import socket
+import threading
+from scriptBase.loki_log import *
 
 # 保存原始的 print 函数
 original_print = print
@@ -16,6 +18,16 @@ def custom_print(*args, **kwargs):
     # 调用原始的 print 函数打印内容
     original_print(*modified_args, **kwargs)
 
+    dstStr = ' '.join(str(arg) for arg in args)
+
+    log_thread = threading.Thread(
+        target=log_to_loki,
+        args=(socket.gethostname(), dstStr),
+        daemon=True  # 设为守护线程（主线程退出时自动结束）
+    )
+    log_thread.start()  # 启动线程（不阻塞主线程）
+
+
 # 替换内置的 print 函数为自定义的输出函数
 builtins.print = custom_print
 

scriptBase/loki_log.py

@@ -4,7 +4,7 @@ import time
 def log_to_loki(label : str, message : str):
     log_data = {
     "streams": [{
-        "stream": {"app": "loki", "env": label},
+        "stream": {"app": label, "env": label},
         "values": [[str(int(time.time()*1e9)), message]]
         }]
     }