fix(log_client): harden background flush thread against exceptions

- Wrap _flush_loop main body in outer try/except to prevent any
  unexpected exception (network errors, serialization errors, etc.)
  from killing the background thread (which would cause unbounded
  queue growth and eventual OOM)
- Wrap _send() call in inner try/except as final safety net
- Wrap RpcClient() construction in try/except (connection failures
  during client creation were previously unhandled)
- Add _safe_json() helper: unserializable extra fields fall back to
  repr()/str() instead of raising TypeError that could kill thread
- Generate timestamp at push() (enqueue time) instead of _send() time,
  so timestamps reflect actual event time even when queue is backed up
- Replace unreliable time.sleep(0.6) in flush() with _flush_done Event
  for proper synchronization between caller and background thread
- Add _QUEUE_MAXSIZE=10000 with drop-oldest policy to prevent unbounded
  memory growth even in pathological cases
- Use put_nowait with Full handling to ensure push() never blocks
  business threads
- Sync fix to all 4 copies: confs/static/smtp/dnfs
This commit is contained in:
qing committed 2026-08-01 03:23:19 +08:00
1 parent e8fa26d79f
commit 1b9c727e80
4 files changed
+976 -692

No files matched your search

+244 -173
View File
@@ -1,173 +1,244 @@
"""统一日志客户端 — 各服务共用(dnfs / smtp / static 三份副本内容一致,改动请同步)。 """统一日志客户端 — 各服务共用(dnfs / smtp / static / confs 四份副本内容一致,改动请同步)。
设计要点: 设计要点:
- 通过 pymsg RPC 调用 dbv2 execute 路由,逐条 INSERT 到 Logs 表 - 通过 pymsg RPC 调用 dbv2 execute 路由,逐条 INSERT 到 Logs 表
- 异步非阻塞:内存队列 + 后台线程批量推送,不拖慢业务 - 异步非阻塞:内存队列 + 后台线程批量推送,不拖慢业务
- fail-safe:db 服务不可用时静默丢弃,绝不影响业务 - fail-safe:db 服务不可用时静默丢弃,绝不影响业务;后台线程任何异常都不会崩溃
- 缓冲批量:队列满 50 条或超时 2 秒时自动 flush - 缓冲批量:队列满 50 条或超时 2 秒时自动 flush
- 时间戳:日志入队时即记录 ts,反映事件真实发生时间
用法:
from log_client import logs 用法:
from log_client import logs
logs.info("dnfs", "节点新增成功 id=42")
logs.error("dnfs", "数据库连接失败", extra={"errno": 2002}) logs.info("dnfs", "节点新增成功 id=42")
logs.push("gateway", "WARNING", "登录失败次数超限", logs.error("dnfs", "数据库连接失败", extra={"errno": 2002})
trace_id="req-abc", extra={"ip": "1.2.3.4"}) logs.push("gateway", "WARNING", "登录失败次数超限",
""" trace_id="req-abc", extra={"ip": "1.2.3.4"})
"""
import json
import os as _os import json
import sys as _sys import os as _os
import threading import sys as _sys
import queue import threading
import time import queue
import atexit import time
from datetime import datetime, timezone import atexit
from datetime import datetime, timezone
_sys.path.insert(0, _os.path.join(_os.path.dirname(__file__), ".."))
from pymsg import RpcClient, RpcError # noqa: E402 _sys.path.insert(0, _os.path.join(_os.path.dirname(__file__), ".."))
from pymsg import RpcClient, RpcError # noqa: E402
# ── 配置 ──────────────────────────────────────────────────────────────
# ── 配置 ──────────────────────────────────────────────────────────────
_DBV2_ADDR = _os.environ.get('DB_HOST', '127.0.0.1')
_DBV2_PORT = int(_os.environ.get('DB_PORT', '8005')) _DBV2_ADDR = _os.environ.get('DB_HOST', '127.0.0.1')
_DBV2_PORT = int(_os.environ.get('DB_PORT', '8005'))
_BATCH_SIZE = 50 # 队列满 50 条时 flush
_FLUSH_INTERVAL = 2.0 # 超时 2 秒强制 flush _BATCH_SIZE = 50 # 队列满 50 条时 flush
_TIMEOUT = 5 # socket 超时(秒) _FLUSH_INTERVAL = 2.0 # 超时 2 秒强制 flush
_TIMEOUT = 5 # socket 超时(秒)
_QUEUE_MAXSIZE = 10000 # 队列上限,防止后台线程异常时内存无限增长;满则丢弃最旧条目
class LogClient:
"""异步批量日志推送客户端。"""
def _safe_json(obj):
def __init__(self, service_name="log", """安全序列化 extra 字段,不可序列化对象转为 repr 字符串,保证不抛异常。"""
batch_size=_BATCH_SIZE, try:
flush_interval=_FLUSH_INTERVAL): return json.dumps(obj, ensure_ascii=False, default=str)
self._service_name = service_name except Exception:
self._batch_size = batch_size return repr(obj)
self._flush_interval = flush_interval
self._queue = queue.Queue()
self._stop = threading.Event() def _now_iso():
self._flush_signal = threading.Event() """生成当前 UTC 时间 ISO 字符串(在入队时调用,保证时间戳准确)。"""
self._thread = threading.Thread(target=self._flush_loop, return datetime.now(timezone.utc).isoformat()
daemon=True, name='log-client')
self._thread.start()
atexit.register(self._shutdown) class LogClient:
"""异步批量日志推送客户端。"""
# ── 公开接口 ────────────────────────────────────────────────────────
def __init__(self, service_name="log",
def push(self, service, level, message, trace_id=None, extra=None): batch_size=_BATCH_SIZE,
"""推送一条日志(非阻塞,立即返回)。""" flush_interval=_FLUSH_INTERVAL):
self._queue.put({ self._service_name = service_name
'service': service, self._batch_size = batch_size
'level': level, self._flush_interval = flush_interval
'message': message, self._queue = queue.Queue(maxsize=_QUEUE_MAXSIZE)
'trace_id': trace_id, self._stop = threading.Event()
'extra': extra, self._flush_signal = threading.Event()
}) self._flush_done = threading.Event()
self._thread = threading.Thread(target=self._flush_loop,
def debug(self, service, message, **kw): daemon=True, name='log-client')
self.push(service, 'DEBUG', message, **kw) self._thread.start()
atexit.register(self._shutdown)
def info(self, service, message, **kw):
self.push(service, 'INFO', message, **kw) # ── 公开接口 ────────────────────────────────────────────────────────
def warning(self, service, message, **kw): def push(self, service, level, message, trace_id=None, extra=None):
self.push(service, 'WARNING', message, **kw) """推送一条日志(非阻塞,立即返回)。
def error(self, service, message, **kw): 队列满时静默丢弃最旧条目再放入新条目,保证不阻塞业务线程、不导致 OOM。
self.push(service, 'ERROR', message, **kw) """
entry = {
def critical(self, service, message, **kw): 'ts': _now_iso(), # 入队时即生成时间戳,反映真实事件时间
self.push(service, 'CRITICAL', message, **kw) 'service': service,
'level': level,
def flush(self): 'message': message,
"""手动 flush 队列中剩余的日志(含后台线程持有的局部 batch)。""" 'trace_id': trace_id,
self._flush_signal.set() # 通知后台线程立即 flush 其局部 batch 'extra': extra,
time.sleep(0.6) # 等待后台线程响应 }
batch = [] try:
while True: self._queue.put_nowait(entry)
try: except queue.Full:
batch.append(self._queue.get_nowait()) # 队列满:丢弃最旧的一条,再放入新条目(drop-oldest 策略)
if len(batch) >= self._batch_size: try:
self._send(batch) self._queue.get_nowait()
batch = [] except queue.Empty:
except queue.Empty: pass
break try:
if batch: self._queue.put_nowait(entry)
self._send(batch) except queue.Full:
self._flush_signal.clear() pass # 极端竞争下仍失败则静默丢弃
# ── 内部实现 ──────────────────────────────────────────────────────── def debug(self, service, message, **kw):
self.push(service, 'DEBUG', message, **kw)
def _flush_loop(self):
"""后台线程:积攒到 batch_size 或超时 flush_interval 时发送。""" def info(self, service, message, **kw):
batch = [] self.push(service, 'INFO', message, **kw)
last_flush = time.monotonic()
def warning(self, service, message, **kw):
while not self._stop.is_set(): self.push(service, 'WARNING', message, **kw)
timeout = self._flush_interval - (time.monotonic() - last_flush)
chk_timeout = max(0.1, min(timeout, 0.5)) def error(self, service, message, **kw):
try: self.push(service, 'ERROR', message, **kw)
item = self._queue.get(timeout=chk_timeout)
batch.append(item) def critical(self, service, message, **kw):
while len(batch) < self._batch_size: self.push(service, 'CRITICAL', message, **kw)
try:
batch.append(self._queue.get_nowait()) def flush(self):
except queue.Empty: """手动 flush 队列中剩余的日志(含后台线程持有的局部 batch)。"""
break self._flush_done.clear()
except queue.Empty: self._flush_signal.set() # 通知后台线程立即 flush 其局部 batch
pass # 等待后台线程完成 flush(最多等待 flush_interval + 2 秒余量)
self._flush_done.wait(timeout=self._flush_interval + 2.0)
should_flush = ( # 主线程排空队列中剩余条目
len(batch) >= self._batch_size or self._drain_queue_to_db()
self._flush_signal.is_set() or
(batch and time.monotonic() - last_flush >= self._flush_interval) def _drain_queue_to_db(self):
) """排空主线程可见的队列中剩余条目,直接发送到 db。"""
if should_flush and batch: batch = []
self._send(batch) while True:
batch = [] try:
last_flush = time.monotonic() batch.append(self._queue.get_nowait())
if len(batch) >= self._batch_size:
# 退出前 flush 剩余 self._send(batch)
if batch: batch = []
self._send(batch) except queue.Empty:
break
def _send(self, batch): if batch:
"""通过 pymsg RPC(execute 路由)逐条 INSERT 到 dbv2 Logs 表。""" self._send(batch)
now = datetime.now(timezone.utc).isoformat()
client = RpcClient(_DBV2_ADDR, _DBV2_PORT, self._service_name + "-log") # ── 内部实现 ────────────────────────────────────────────────────────
for entry in batch:
extra_val = entry.get("extra") def _flush_loop(self):
payload = { """后台线程:积攒到 batch_size 或超时 flush_interval 时发送。
"action": "insert",
"table": "Logs", 外层捕获所有异常,保证后台线程在任何情况下(网络故障、序列化错误等)
"values": { 都不会死亡,避免日志队列无限增长导致 OOM。
"ts": now, """
"service": entry.get("service"), batch = []
"level": entry.get("level"), last_flush = time.monotonic()
"message": entry.get("message"),
"trace_id": entry.get("trace_id"), while not self._stop.is_set():
"extra": json.dumps(extra_val, ensure_ascii=False) try:
if extra_val is not None else None, timeout = self._flush_interval - (time.monotonic() - last_flush)
}, chk_timeout = max(0.1, min(timeout, 0.5))
} try:
try: item = self._queue.get(timeout=chk_timeout)
client.call("execute", payload, timeout=_TIMEOUT) batch.append(item)
except RpcError as e: while len(batch) < self._batch_size:
# 业务错误(如表结构不符):打印后继续下一条 try:
print(f"[log_client] INSERT 失败: {e}", file=_sys.stderr) batch.append(self._queue.get_nowait())
except Exception as e: except queue.Empty:
# 网络错误:db 不可用,放弃本批剩余(fail-safe) break
print(f"[log_client] 发送失败: {e}", file=_sys.stderr) except queue.Empty:
break pass
def _shutdown(self): should_flush = (
"""进程退出时 flush 剩余日志。""" len(batch) >= self._batch_size or
self._stop.set() self._flush_signal.is_set() or
self.flush() (batch and time.monotonic() - last_flush >= self._flush_interval)
)
if should_flush and batch:
# ── 全局单例 ────────────────────────────────────────────────────────── try:
self._send(batch)
logs = LogClient() except Exception as e:
# _send 内部已有细粒度异常处理,此处是最终安全网
print(f"[log_client] _send 异常(已捕获,线程继续运行): {e}",
file=_sys.stderr)
batch = []
last_flush = time.monotonic()
if self._flush_signal.is_set():
self._flush_signal.clear()
self._flush_done.set()
except Exception as e:
# 最外层安全网:任何意外异常都不应杀死日志线程
print(f"[log_client] _flush_loop 异常(已捕获,线程继续运行): {e}",
file=_sys.stderr)
time.sleep(0.5) # 防止异常风暴导致 CPU 满载
# 退出前 flush 剩余
if batch:
try:
self._send(batch)
except Exception as e:
print(f"[log_client] 退出时 flush 异常: {e}", file=_sys.stderr)
self._flush_done.set()
def _send(self, batch):
"""通过 pymsg RPC(execute 路由)批量发送到 dbv2 Logs 表。
整个 batch 复用一个 TCP 连接;网络错误时放弃本批剩余条目(fail-safe),
不抛异常给调用方。
"""
try:
client = RpcClient(_DBV2_ADDR, _DBV2_PORT, self._service_name + "-log")
except Exception as e:
print(f"[log_client] 创建 RpcClient 失败: {e}", file=_sys.stderr)
return
for entry in batch:
extra_val = entry.get("extra")
try:
extra_json = _safe_json(extra_val) if extra_val is not None else None
except Exception:
extra_json = None
payload = {
"action": "insert",
"table": "Logs",
"values": {
"ts": entry.get("ts", _now_iso()),
"service": entry.get("service"),
"level": entry.get("level"),
"message": entry.get("message"),
"trace_id": entry.get("trace_id"),
"extra": extra_json,
},
}
try:
client.call("execute", payload, timeout=_TIMEOUT)
except RpcError as e:
# 业务错误(如表结构不符):打印后继续下一条
print(f"[log_client] INSERT 失败: {e}", file=_sys.stderr)
except Exception as e:
# 网络错误:db 不可用,放弃本批剩余(fail-safe)
print(f"[log_client] 发送失败: {e}", file=_sys.stderr)
break
def _shutdown(self):
"""进程退出时 flush 剩余日志。"""
self._stop.set()
self.flush()
# ── 全局单例 ──────────────────────────────────────────────────────────
logs = LogClient()
+244 -173
View File
@@ -1,173 +1,244 @@
"""统一日志客户端 — 各服务共用(dnfs / smtp / static 三份副本内容一致,改动请同步)。 """统一日志客户端 — 各服务共用(dnfs / smtp / static / confs 四份副本内容一致,改动请同步)。
设计要点: 设计要点:
- 通过 pymsg RPC 调用 dbv2 execute 路由,逐条 INSERT 到 Logs 表 - 通过 pymsg RPC 调用 dbv2 execute 路由,逐条 INSERT 到 Logs 表
- 异步非阻塞:内存队列 + 后台线程批量推送,不拖慢业务 - 异步非阻塞:内存队列 + 后台线程批量推送,不拖慢业务
- fail-safe:db 服务不可用时静默丢弃,绝不影响业务 - fail-safe:db 服务不可用时静默丢弃,绝不影响业务;后台线程任何异常都不会崩溃
- 缓冲批量:队列满 50 条或超时 2 秒时自动 flush - 缓冲批量:队列满 50 条或超时 2 秒时自动 flush
- 时间戳:日志入队时即记录 ts,反映事件真实发生时间
用法:
from log_client import logs 用法:
from log_client import logs
logs.info("dnfs", "节点新增成功 id=42")
logs.error("dnfs", "数据库连接失败", extra={"errno": 2002}) logs.info("dnfs", "节点新增成功 id=42")
logs.push("gateway", "WARNING", "登录失败次数超限", logs.error("dnfs", "数据库连接失败", extra={"errno": 2002})
trace_id="req-abc", extra={"ip": "1.2.3.4"}) logs.push("gateway", "WARNING", "登录失败次数超限",
""" trace_id="req-abc", extra={"ip": "1.2.3.4"})
"""
import json
import os as _os import json
import sys as _sys import os as _os
import threading import sys as _sys
import queue import threading
import time import queue
import atexit import time
from datetime import datetime, timezone import atexit
from datetime import datetime, timezone
_sys.path.insert(0, _os.path.join(_os.path.dirname(__file__), ".."))
from pymsg import RpcClient, RpcError # noqa: E402 _sys.path.insert(0, _os.path.join(_os.path.dirname(__file__), ".."))
from pymsg import RpcClient, RpcError # noqa: E402
# ── 配置 ──────────────────────────────────────────────────────────────
# ── 配置 ──────────────────────────────────────────────────────────────
_DBV2_ADDR = _os.environ.get('DB_HOST', '127.0.0.1')
_DBV2_PORT = int(_os.environ.get('DB_PORT', '8005')) _DBV2_ADDR = _os.environ.get('DB_HOST', '127.0.0.1')
_DBV2_PORT = int(_os.environ.get('DB_PORT', '8005'))
_BATCH_SIZE = 50 # 队列满 50 条时 flush
_FLUSH_INTERVAL = 2.0 # 超时 2 秒强制 flush _BATCH_SIZE = 50 # 队列满 50 条时 flush
_TIMEOUT = 5 # socket 超时(秒) _FLUSH_INTERVAL = 2.0 # 超时 2 秒强制 flush
_TIMEOUT = 5 # socket 超时(秒)
_QUEUE_MAXSIZE = 10000 # 队列上限,防止后台线程异常时内存无限增长;满则丢弃最旧条目
class LogClient:
"""异步批量日志推送客户端。"""
def _safe_json(obj):
def __init__(self, service_name="log", """安全序列化 extra 字段,不可序列化对象转为 repr 字符串,保证不抛异常。"""
batch_size=_BATCH_SIZE, try:
flush_interval=_FLUSH_INTERVAL): return json.dumps(obj, ensure_ascii=False, default=str)
self._service_name = service_name except Exception:
self._batch_size = batch_size return repr(obj)
self._flush_interval = flush_interval
self._queue = queue.Queue()
self._stop = threading.Event() def _now_iso():
self._flush_signal = threading.Event() """生成当前 UTC 时间 ISO 字符串(在入队时调用,保证时间戳准确)。"""
self._thread = threading.Thread(target=self._flush_loop, return datetime.now(timezone.utc).isoformat()
daemon=True, name='log-client')
self._thread.start()
atexit.register(self._shutdown) class LogClient:
"""异步批量日志推送客户端。"""
# ── 公开接口 ────────────────────────────────────────────────────────
def __init__(self, service_name="log",
def push(self, service, level, message, trace_id=None, extra=None): batch_size=_BATCH_SIZE,
"""推送一条日志(非阻塞,立即返回)。""" flush_interval=_FLUSH_INTERVAL):
self._queue.put({ self._service_name = service_name
'service': service, self._batch_size = batch_size
'level': level, self._flush_interval = flush_interval
'message': message, self._queue = queue.Queue(maxsize=_QUEUE_MAXSIZE)
'trace_id': trace_id, self._stop = threading.Event()
'extra': extra, self._flush_signal = threading.Event()
}) self._flush_done = threading.Event()
self._thread = threading.Thread(target=self._flush_loop,
def debug(self, service, message, **kw): daemon=True, name='log-client')
self.push(service, 'DEBUG', message, **kw) self._thread.start()
atexit.register(self._shutdown)
def info(self, service, message, **kw):
self.push(service, 'INFO', message, **kw) # ── 公开接口 ────────────────────────────────────────────────────────
def warning(self, service, message, **kw): def push(self, service, level, message, trace_id=None, extra=None):
self.push(service, 'WARNING', message, **kw) """推送一条日志(非阻塞,立即返回)。
def error(self, service, message, **kw): 队列满时静默丢弃最旧条目再放入新条目,保证不阻塞业务线程、不导致 OOM。
self.push(service, 'ERROR', message, **kw) """
entry = {
def critical(self, service, message, **kw): 'ts': _now_iso(), # 入队时即生成时间戳,反映真实事件时间
self.push(service, 'CRITICAL', message, **kw) 'service': service,
'level': level,
def flush(self): 'message': message,
"""手动 flush 队列中剩余的日志(含后台线程持有的局部 batch)。""" 'trace_id': trace_id,
self._flush_signal.set() # 通知后台线程立即 flush 其局部 batch 'extra': extra,
time.sleep(0.6) # 等待后台线程响应 }
batch = [] try:
while True: self._queue.put_nowait(entry)
try: except queue.Full:
batch.append(self._queue.get_nowait()) # 队列满:丢弃最旧的一条,再放入新条目(drop-oldest 策略)
if len(batch) >= self._batch_size: try:
self._send(batch) self._queue.get_nowait()
batch = [] except queue.Empty:
except queue.Empty: pass
break try:
if batch: self._queue.put_nowait(entry)
self._send(batch) except queue.Full:
self._flush_signal.clear() pass # 极端竞争下仍失败则静默丢弃
# ── 内部实现 ──────────────────────────────────────────────────────── def debug(self, service, message, **kw):
self.push(service, 'DEBUG', message, **kw)
def _flush_loop(self):
"""后台线程:积攒到 batch_size 或超时 flush_interval 时发送。""" def info(self, service, message, **kw):
batch = [] self.push(service, 'INFO', message, **kw)
last_flush = time.monotonic()
def warning(self, service, message, **kw):
while not self._stop.is_set(): self.push(service, 'WARNING', message, **kw)
timeout = self._flush_interval - (time.monotonic() - last_flush)
chk_timeout = max(0.1, min(timeout, 0.5)) def error(self, service, message, **kw):
try: self.push(service, 'ERROR', message, **kw)
item = self._queue.get(timeout=chk_timeout)
batch.append(item) def critical(self, service, message, **kw):
while len(batch) < self._batch_size: self.push(service, 'CRITICAL', message, **kw)
try:
batch.append(self._queue.get_nowait()) def flush(self):
except queue.Empty: """手动 flush 队列中剩余的日志(含后台线程持有的局部 batch)。"""
break self._flush_done.clear()
except queue.Empty: self._flush_signal.set() # 通知后台线程立即 flush 其局部 batch
pass # 等待后台线程完成 flush(最多等待 flush_interval + 2 秒余量)
self._flush_done.wait(timeout=self._flush_interval + 2.0)
should_flush = ( # 主线程排空队列中剩余条目
len(batch) >= self._batch_size or self._drain_queue_to_db()
self._flush_signal.is_set() or
(batch and time.monotonic() - last_flush >= self._flush_interval) def _drain_queue_to_db(self):
) """排空主线程可见的队列中剩余条目,直接发送到 db。"""
if should_flush and batch: batch = []
self._send(batch) while True:
batch = [] try:
last_flush = time.monotonic() batch.append(self._queue.get_nowait())
if len(batch) >= self._batch_size:
# 退出前 flush 剩余 self._send(batch)
if batch: batch = []
self._send(batch) except queue.Empty:
break
def _send(self, batch): if batch:
"""通过 pymsg RPC(execute 路由)逐条 INSERT 到 dbv2 Logs 表。""" self._send(batch)
now = datetime.now(timezone.utc).isoformat()
client = RpcClient(_DBV2_ADDR, _DBV2_PORT, self._service_name + "-log") # ── 内部实现 ────────────────────────────────────────────────────────
for entry in batch:
extra_val = entry.get("extra") def _flush_loop(self):
payload = { """后台线程:积攒到 batch_size 或超时 flush_interval 时发送。
"action": "insert",
"table": "Logs", 外层捕获所有异常,保证后台线程在任何情况下(网络故障、序列化错误等)
"values": { 都不会死亡,避免日志队列无限增长导致 OOM。
"ts": now, """
"service": entry.get("service"), batch = []
"level": entry.get("level"), last_flush = time.monotonic()
"message": entry.get("message"),
"trace_id": entry.get("trace_id"), while not self._stop.is_set():
"extra": json.dumps(extra_val, ensure_ascii=False) try:
if extra_val is not None else None, timeout = self._flush_interval - (time.monotonic() - last_flush)
}, chk_timeout = max(0.1, min(timeout, 0.5))
} try:
try: item = self._queue.get(timeout=chk_timeout)
client.call("execute", payload, timeout=_TIMEOUT) batch.append(item)
except RpcError as e: while len(batch) < self._batch_size:
# 业务错误(如表结构不符):打印后继续下一条 try:
print(f"[log_client] INSERT 失败: {e}", file=_sys.stderr) batch.append(self._queue.get_nowait())
except Exception as e: except queue.Empty:
# 网络错误:db 不可用,放弃本批剩余(fail-safe) break
print(f"[log_client] 发送失败: {e}", file=_sys.stderr) except queue.Empty:
break pass
def _shutdown(self): should_flush = (
"""进程退出时 flush 剩余日志。""" len(batch) >= self._batch_size or
self._stop.set() self._flush_signal.is_set() or
self.flush() (batch and time.monotonic() - last_flush >= self._flush_interval)
)
if should_flush and batch:
# ── 全局单例 ────────────────────────────────────────────────────────── try:
self._send(batch)
logs = LogClient() except Exception as e:
# _send 内部已有细粒度异常处理,此处是最终安全网
print(f"[log_client] _send 异常(已捕获,线程继续运行): {e}",
file=_sys.stderr)
batch = []
last_flush = time.monotonic()
if self._flush_signal.is_set():
self._flush_signal.clear()
self._flush_done.set()
except Exception as e:
# 最外层安全网:任何意外异常都不应杀死日志线程
print(f"[log_client] _flush_loop 异常(已捕获,线程继续运行): {e}",
file=_sys.stderr)
time.sleep(0.5) # 防止异常风暴导致 CPU 满载
# 退出前 flush 剩余
if batch:
try:
self._send(batch)
except Exception as e:
print(f"[log_client] 退出时 flush 异常: {e}", file=_sys.stderr)
self._flush_done.set()
def _send(self, batch):
"""通过 pymsg RPC(execute 路由)批量发送到 dbv2 Logs 表。
整个 batch 复用一个 TCP 连接;网络错误时放弃本批剩余条目(fail-safe),
不抛异常给调用方。
"""
try:
client = RpcClient(_DBV2_ADDR, _DBV2_PORT, self._service_name + "-log")
except Exception as e:
print(f"[log_client] 创建 RpcClient 失败: {e}", file=_sys.stderr)
return
for entry in batch:
extra_val = entry.get("extra")
try:
extra_json = _safe_json(extra_val) if extra_val is not None else None
except Exception:
extra_json = None
payload = {
"action": "insert",
"table": "Logs",
"values": {
"ts": entry.get("ts", _now_iso()),
"service": entry.get("service"),
"level": entry.get("level"),
"message": entry.get("message"),
"trace_id": entry.get("trace_id"),
"extra": extra_json,
},
}
try:
client.call("execute", payload, timeout=_TIMEOUT)
except RpcError as e:
# 业务错误(如表结构不符):打印后继续下一条
print(f"[log_client] INSERT 失败: {e}", file=_sys.stderr)
except Exception as e:
# 网络错误:db 不可用,放弃本批剩余(fail-safe)
print(f"[log_client] 发送失败: {e}", file=_sys.stderr)
break
def _shutdown(self):
"""进程退出时 flush 剩余日志。"""
self._stop.set()
self.flush()
# ── 全局单例 ──────────────────────────────────────────────────────────
logs = LogClient()
+244 -173
View File
@@ -1,173 +1,244 @@
"""统一日志客户端 — 各服务共用(dnfs / smtp / static 三份副本内容一致,改动请同步)。 """统一日志客户端 — 各服务共用(dnfs / smtp / static / confs 四份副本内容一致,改动请同步)。
设计要点: 设计要点:
- 通过 pymsg RPC 调用 dbv2 execute 路由,逐条 INSERT 到 Logs 表 - 通过 pymsg RPC 调用 dbv2 execute 路由,逐条 INSERT 到 Logs 表
- 异步非阻塞:内存队列 + 后台线程批量推送,不拖慢业务 - 异步非阻塞:内存队列 + 后台线程批量推送,不拖慢业务
- fail-safe:db 服务不可用时静默丢弃,绝不影响业务 - fail-safe:db 服务不可用时静默丢弃,绝不影响业务;后台线程任何异常都不会崩溃
- 缓冲批量:队列满 50 条或超时 2 秒时自动 flush - 缓冲批量:队列满 50 条或超时 2 秒时自动 flush
- 时间戳:日志入队时即记录 ts,反映事件真实发生时间
用法:
from log_client import logs 用法:
from log_client import logs
logs.info("dnfs", "节点新增成功 id=42")
logs.error("dnfs", "数据库连接失败", extra={"errno": 2002}) logs.info("dnfs", "节点新增成功 id=42")
logs.push("gateway", "WARNING", "登录失败次数超限", logs.error("dnfs", "数据库连接失败", extra={"errno": 2002})
trace_id="req-abc", extra={"ip": "1.2.3.4"}) logs.push("gateway", "WARNING", "登录失败次数超限",
""" trace_id="req-abc", extra={"ip": "1.2.3.4"})
"""
import json
import os as _os import json
import sys as _sys import os as _os
import threading import sys as _sys
import queue import threading
import time import queue
import atexit import time
from datetime import datetime, timezone import atexit
from datetime import datetime, timezone
_sys.path.insert(0, _os.path.join(_os.path.dirname(__file__), ".."))
from pymsg import RpcClient, RpcError # noqa: E402 _sys.path.insert(0, _os.path.join(_os.path.dirname(__file__), ".."))
from pymsg import RpcClient, RpcError # noqa: E402
# ── 配置 ──────────────────────────────────────────────────────────────
# ── 配置 ──────────────────────────────────────────────────────────────
_DBV2_ADDR = _os.environ.get('DB_HOST', '127.0.0.1')
_DBV2_PORT = int(_os.environ.get('DB_PORT', '8005')) _DBV2_ADDR = _os.environ.get('DB_HOST', '127.0.0.1')
_DBV2_PORT = int(_os.environ.get('DB_PORT', '8005'))
_BATCH_SIZE = 50 # 队列满 50 条时 flush
_FLUSH_INTERVAL = 2.0 # 超时 2 秒强制 flush _BATCH_SIZE = 50 # 队列满 50 条时 flush
_TIMEOUT = 5 # socket 超时(秒) _FLUSH_INTERVAL = 2.0 # 超时 2 秒强制 flush
_TIMEOUT = 5 # socket 超时(秒)
_QUEUE_MAXSIZE = 10000 # 队列上限,防止后台线程异常时内存无限增长;满则丢弃最旧条目
class LogClient:
"""异步批量日志推送客户端。"""
def _safe_json(obj):
def __init__(self, service_name="log", """安全序列化 extra 字段,不可序列化对象转为 repr 字符串,保证不抛异常。"""
batch_size=_BATCH_SIZE, try:
flush_interval=_FLUSH_INTERVAL): return json.dumps(obj, ensure_ascii=False, default=str)
self._service_name = service_name except Exception:
self._batch_size = batch_size return repr(obj)
self._flush_interval = flush_interval
self._queue = queue.Queue()
self._stop = threading.Event() def _now_iso():
self._flush_signal = threading.Event() """生成当前 UTC 时间 ISO 字符串(在入队时调用,保证时间戳准确)。"""
self._thread = threading.Thread(target=self._flush_loop, return datetime.now(timezone.utc).isoformat()
daemon=True, name='log-client')
self._thread.start()
atexit.register(self._shutdown) class LogClient:
"""异步批量日志推送客户端。"""
# ── 公开接口 ────────────────────────────────────────────────────────
def __init__(self, service_name="log",
def push(self, service, level, message, trace_id=None, extra=None): batch_size=_BATCH_SIZE,
"""推送一条日志(非阻塞,立即返回)。""" flush_interval=_FLUSH_INTERVAL):
self._queue.put({ self._service_name = service_name
'service': service, self._batch_size = batch_size
'level': level, self._flush_interval = flush_interval
'message': message, self._queue = queue.Queue(maxsize=_QUEUE_MAXSIZE)
'trace_id': trace_id, self._stop = threading.Event()
'extra': extra, self._flush_signal = threading.Event()
}) self._flush_done = threading.Event()
self._thread = threading.Thread(target=self._flush_loop,
def debug(self, service, message, **kw): daemon=True, name='log-client')
self.push(service, 'DEBUG', message, **kw) self._thread.start()
atexit.register(self._shutdown)
def info(self, service, message, **kw):
self.push(service, 'INFO', message, **kw) # ── 公开接口 ────────────────────────────────────────────────────────
def warning(self, service, message, **kw): def push(self, service, level, message, trace_id=None, extra=None):
self.push(service, 'WARNING', message, **kw) """推送一条日志(非阻塞,立即返回)。
def error(self, service, message, **kw): 队列满时静默丢弃最旧条目再放入新条目,保证不阻塞业务线程、不导致 OOM。
self.push(service, 'ERROR', message, **kw) """
entry = {
def critical(self, service, message, **kw): 'ts': _now_iso(), # 入队时即生成时间戳,反映真实事件时间
self.push(service, 'CRITICAL', message, **kw) 'service': service,
'level': level,
def flush(self): 'message': message,
"""手动 flush 队列中剩余的日志(含后台线程持有的局部 batch)。""" 'trace_id': trace_id,
self._flush_signal.set() # 通知后台线程立即 flush 其局部 batch 'extra': extra,
time.sleep(0.6) # 等待后台线程响应 }
batch = [] try:
while True: self._queue.put_nowait(entry)
try: except queue.Full:
batch.append(self._queue.get_nowait()) # 队列满:丢弃最旧的一条,再放入新条目(drop-oldest 策略)
if len(batch) >= self._batch_size: try:
self._send(batch) self._queue.get_nowait()
batch = [] except queue.Empty:
except queue.Empty: pass
break try:
if batch: self._queue.put_nowait(entry)
self._send(batch) except queue.Full:
self._flush_signal.clear() pass # 极端竞争下仍失败则静默丢弃
# ── 内部实现 ──────────────────────────────────────────────────────── def debug(self, service, message, **kw):
self.push(service, 'DEBUG', message, **kw)
def _flush_loop(self):
"""后台线程:积攒到 batch_size 或超时 flush_interval 时发送。""" def info(self, service, message, **kw):
batch = [] self.push(service, 'INFO', message, **kw)
last_flush = time.monotonic()
def warning(self, service, message, **kw):
while not self._stop.is_set(): self.push(service, 'WARNING', message, **kw)
timeout = self._flush_interval - (time.monotonic() - last_flush)
chk_timeout = max(0.1, min(timeout, 0.5)) def error(self, service, message, **kw):
try: self.push(service, 'ERROR', message, **kw)
item = self._queue.get(timeout=chk_timeout)
batch.append(item) def critical(self, service, message, **kw):
while len(batch) < self._batch_size: self.push(service, 'CRITICAL', message, **kw)
try:
batch.append(self._queue.get_nowait()) def flush(self):
except queue.Empty: """手动 flush 队列中剩余的日志(含后台线程持有的局部 batch)。"""
break self._flush_done.clear()
except queue.Empty: self._flush_signal.set() # 通知后台线程立即 flush 其局部 batch
pass # 等待后台线程完成 flush(最多等待 flush_interval + 2 秒余量)
self._flush_done.wait(timeout=self._flush_interval + 2.0)
should_flush = ( # 主线程排空队列中剩余条目
len(batch) >= self._batch_size or self._drain_queue_to_db()
self._flush_signal.is_set() or
(batch and time.monotonic() - last_flush >= self._flush_interval) def _drain_queue_to_db(self):
) """排空主线程可见的队列中剩余条目,直接发送到 db。"""
if should_flush and batch: batch = []
self._send(batch) while True:
batch = [] try:
last_flush = time.monotonic() batch.append(self._queue.get_nowait())
if len(batch) >= self._batch_size:
# 退出前 flush 剩余 self._send(batch)
if batch: batch = []
self._send(batch) except queue.Empty:
break
def _send(self, batch): if batch:
"""通过 pymsg RPC(execute 路由)逐条 INSERT 到 dbv2 Logs 表。""" self._send(batch)
now = datetime.now(timezone.utc).isoformat()
client = RpcClient(_DBV2_ADDR, _DBV2_PORT, self._service_name + "-log") # ── 内部实现 ────────────────────────────────────────────────────────
for entry in batch:
extra_val = entry.get("extra") def _flush_loop(self):
payload = { """后台线程:积攒到 batch_size 或超时 flush_interval 时发送。
"action": "insert",
"table": "Logs", 外层捕获所有异常,保证后台线程在任何情况下(网络故障、序列化错误等)
"values": { 都不会死亡,避免日志队列无限增长导致 OOM。
"ts": now, """
"service": entry.get("service"), batch = []
"level": entry.get("level"), last_flush = time.monotonic()
"message": entry.get("message"),
"trace_id": entry.get("trace_id"), while not self._stop.is_set():
"extra": json.dumps(extra_val, ensure_ascii=False) try:
if extra_val is not None else None, timeout = self._flush_interval - (time.monotonic() - last_flush)
}, chk_timeout = max(0.1, min(timeout, 0.5))
} try:
try: item = self._queue.get(timeout=chk_timeout)
client.call("execute", payload, timeout=_TIMEOUT) batch.append(item)
except RpcError as e: while len(batch) < self._batch_size:
# 业务错误(如表结构不符):打印后继续下一条 try:
print(f"[log_client] INSERT 失败: {e}", file=_sys.stderr) batch.append(self._queue.get_nowait())
except Exception as e: except queue.Empty:
# 网络错误:db 不可用,放弃本批剩余(fail-safe) break
print(f"[log_client] 发送失败: {e}", file=_sys.stderr) except queue.Empty:
break pass
def _shutdown(self): should_flush = (
"""进程退出时 flush 剩余日志。""" len(batch) >= self._batch_size or
self._stop.set() self._flush_signal.is_set() or
self.flush() (batch and time.monotonic() - last_flush >= self._flush_interval)
)
if should_flush and batch:
# ── 全局单例 ────────────────────────────────────────────────────────── try:
self._send(batch)
logs = LogClient() except Exception as e:
# _send 内部已有细粒度异常处理,此处是最终安全网
print(f"[log_client] _send 异常(已捕获,线程继续运行): {e}",
file=_sys.stderr)
batch = []
last_flush = time.monotonic()
if self._flush_signal.is_set():
self._flush_signal.clear()
self._flush_done.set()
except Exception as e:
# 最外层安全网:任何意外异常都不应杀死日志线程
print(f"[log_client] _flush_loop 异常(已捕获,线程继续运行): {e}",
file=_sys.stderr)
time.sleep(0.5) # 防止异常风暴导致 CPU 满载
# 退出前 flush 剩余
if batch:
try:
self._send(batch)
except Exception as e:
print(f"[log_client] 退出时 flush 异常: {e}", file=_sys.stderr)
self._flush_done.set()
def _send(self, batch):
"""通过 pymsg RPC(execute 路由)批量发送到 dbv2 Logs 表。
整个 batch 复用一个 TCP 连接;网络错误时放弃本批剩余条目(fail-safe),
不抛异常给调用方。
"""
try:
client = RpcClient(_DBV2_ADDR, _DBV2_PORT, self._service_name + "-log")
except Exception as e:
print(f"[log_client] 创建 RpcClient 失败: {e}", file=_sys.stderr)
return
for entry in batch:
extra_val = entry.get("extra")
try:
extra_json = _safe_json(extra_val) if extra_val is not None else None
except Exception:
extra_json = None
payload = {
"action": "insert",
"table": "Logs",
"values": {
"ts": entry.get("ts", _now_iso()),
"service": entry.get("service"),
"level": entry.get("level"),
"message": entry.get("message"),
"trace_id": entry.get("trace_id"),
"extra": extra_json,
},
}
try:
client.call("execute", payload, timeout=_TIMEOUT)
except RpcError as e:
# 业务错误(如表结构不符):打印后继续下一条
print(f"[log_client] INSERT 失败: {e}", file=_sys.stderr)
except Exception as e:
# 网络错误:db 不可用,放弃本批剩余(fail-safe)
print(f"[log_client] 发送失败: {e}", file=_sys.stderr)
break
def _shutdown(self):
"""进程退出时 flush 剩余日志。"""
self._stop.set()
self.flush()
# ── 全局单例 ──────────────────────────────────────────────────────────
logs = LogClient()
+244 -173
View File
@@ -1,173 +1,244 @@
"""统一日志客户端 — 各服务共用(dnfs / smtp / static 三份副本内容一致,改动请同步)。 """统一日志客户端 — 各服务共用(dnfs / smtp / static / confs 四份副本内容一致,改动请同步)。
设计要点: 设计要点:
- 通过 pymsg RPC 调用 dbv2 execute 路由,逐条 INSERT 到 Logs 表 - 通过 pymsg RPC 调用 dbv2 execute 路由,逐条 INSERT 到 Logs 表
- 异步非阻塞:内存队列 + 后台线程批量推送,不拖慢业务 - 异步非阻塞:内存队列 + 后台线程批量推送,不拖慢业务
- fail-safe:db 服务不可用时静默丢弃,绝不影响业务 - fail-safe:db 服务不可用时静默丢弃,绝不影响业务;后台线程任何异常都不会崩溃
- 缓冲批量:队列满 50 条或超时 2 秒时自动 flush - 缓冲批量:队列满 50 条或超时 2 秒时自动 flush
- 时间戳:日志入队时即记录 ts,反映事件真实发生时间
用法:
from log_client import logs 用法:
from log_client import logs
logs.info("dnfs", "节点新增成功 id=42")
logs.error("dnfs", "数据库连接失败", extra={"errno": 2002}) logs.info("dnfs", "节点新增成功 id=42")
logs.push("gateway", "WARNING", "登录失败次数超限", logs.error("dnfs", "数据库连接失败", extra={"errno": 2002})
trace_id="req-abc", extra={"ip": "1.2.3.4"}) logs.push("gateway", "WARNING", "登录失败次数超限",
""" trace_id="req-abc", extra={"ip": "1.2.3.4"})
"""
import json
import os as _os import json
import sys as _sys import os as _os
import threading import sys as _sys
import queue import threading
import time import queue
import atexit import time
from datetime import datetime, timezone import atexit
from datetime import datetime, timezone
_sys.path.insert(0, _os.path.join(_os.path.dirname(__file__), ".."))
from pymsg import RpcClient, RpcError # noqa: E402 _sys.path.insert(0, _os.path.join(_os.path.dirname(__file__), ".."))
from pymsg import RpcClient, RpcError # noqa: E402
# ── 配置 ──────────────────────────────────────────────────────────────
# ── 配置 ──────────────────────────────────────────────────────────────
_DBV2_ADDR = _os.environ.get('DB_HOST', '127.0.0.1')
_DBV2_PORT = int(_os.environ.get('DB_PORT', '8005')) _DBV2_ADDR = _os.environ.get('DB_HOST', '127.0.0.1')
_DBV2_PORT = int(_os.environ.get('DB_PORT', '8005'))
_BATCH_SIZE = 50 # 队列满 50 条时 flush
_FLUSH_INTERVAL = 2.0 # 超时 2 秒强制 flush _BATCH_SIZE = 50 # 队列满 50 条时 flush
_TIMEOUT = 5 # socket 超时(秒) _FLUSH_INTERVAL = 2.0 # 超时 2 秒强制 flush
_TIMEOUT = 5 # socket 超时(秒)
_QUEUE_MAXSIZE = 10000 # 队列上限,防止后台线程异常时内存无限增长;满则丢弃最旧条目
class LogClient:
"""异步批量日志推送客户端。"""
def _safe_json(obj):
def __init__(self, service_name="log", """安全序列化 extra 字段,不可序列化对象转为 repr 字符串,保证不抛异常。"""
batch_size=_BATCH_SIZE, try:
flush_interval=_FLUSH_INTERVAL): return json.dumps(obj, ensure_ascii=False, default=str)
self._service_name = service_name except Exception:
self._batch_size = batch_size return repr(obj)
self._flush_interval = flush_interval
self._queue = queue.Queue()
self._stop = threading.Event() def _now_iso():
self._flush_signal = threading.Event() """生成当前 UTC 时间 ISO 字符串(在入队时调用,保证时间戳准确)。"""
self._thread = threading.Thread(target=self._flush_loop, return datetime.now(timezone.utc).isoformat()
daemon=True, name='log-client')
self._thread.start()
atexit.register(self._shutdown) class LogClient:
"""异步批量日志推送客户端。"""
# ── 公开接口 ────────────────────────────────────────────────────────
def __init__(self, service_name="log",
def push(self, service, level, message, trace_id=None, extra=None): batch_size=_BATCH_SIZE,
"""推送一条日志(非阻塞,立即返回)。""" flush_interval=_FLUSH_INTERVAL):
self._queue.put({ self._service_name = service_name
'service': service, self._batch_size = batch_size
'level': level, self._flush_interval = flush_interval
'message': message, self._queue = queue.Queue(maxsize=_QUEUE_MAXSIZE)
'trace_id': trace_id, self._stop = threading.Event()
'extra': extra, self._flush_signal = threading.Event()
}) self._flush_done = threading.Event()
self._thread = threading.Thread(target=self._flush_loop,
def debug(self, service, message, **kw): daemon=True, name='log-client')
self.push(service, 'DEBUG', message, **kw) self._thread.start()
atexit.register(self._shutdown)
def info(self, service, message, **kw):
self.push(service, 'INFO', message, **kw) # ── 公开接口 ────────────────────────────────────────────────────────
def warning(self, service, message, **kw): def push(self, service, level, message, trace_id=None, extra=None):
self.push(service, 'WARNING', message, **kw) """推送一条日志(非阻塞,立即返回)。
def error(self, service, message, **kw): 队列满时静默丢弃最旧条目再放入新条目,保证不阻塞业务线程、不导致 OOM。
self.push(service, 'ERROR', message, **kw) """
entry = {
def critical(self, service, message, **kw): 'ts': _now_iso(), # 入队时即生成时间戳,反映真实事件时间
self.push(service, 'CRITICAL', message, **kw) 'service': service,
'level': level,
def flush(self): 'message': message,
"""手动 flush 队列中剩余的日志(含后台线程持有的局部 batch)。""" 'trace_id': trace_id,
self._flush_signal.set() # 通知后台线程立即 flush 其局部 batch 'extra': extra,
time.sleep(0.6) # 等待后台线程响应 }
batch = [] try:
while True: self._queue.put_nowait(entry)
try: except queue.Full:
batch.append(self._queue.get_nowait()) # 队列满:丢弃最旧的一条,再放入新条目(drop-oldest 策略)
if len(batch) >= self._batch_size: try:
self._send(batch) self._queue.get_nowait()
batch = [] except queue.Empty:
except queue.Empty: pass
break try:
if batch: self._queue.put_nowait(entry)
self._send(batch) except queue.Full:
self._flush_signal.clear() pass # 极端竞争下仍失败则静默丢弃
# ── 内部实现 ──────────────────────────────────────────────────────── def debug(self, service, message, **kw):
self.push(service, 'DEBUG', message, **kw)
def _flush_loop(self):
"""后台线程:积攒到 batch_size 或超时 flush_interval 时发送。""" def info(self, service, message, **kw):
batch = [] self.push(service, 'INFO', message, **kw)
last_flush = time.monotonic()
def warning(self, service, message, **kw):
while not self._stop.is_set(): self.push(service, 'WARNING', message, **kw)
timeout = self._flush_interval - (time.monotonic() - last_flush)
chk_timeout = max(0.1, min(timeout, 0.5)) def error(self, service, message, **kw):
try: self.push(service, 'ERROR', message, **kw)
item = self._queue.get(timeout=chk_timeout)
batch.append(item) def critical(self, service, message, **kw):
while len(batch) < self._batch_size: self.push(service, 'CRITICAL', message, **kw)
try:
batch.append(self._queue.get_nowait()) def flush(self):
except queue.Empty: """手动 flush 队列中剩余的日志(含后台线程持有的局部 batch)。"""
break self._flush_done.clear()
except queue.Empty: self._flush_signal.set() # 通知后台线程立即 flush 其局部 batch
pass # 等待后台线程完成 flush(最多等待 flush_interval + 2 秒余量)
self._flush_done.wait(timeout=self._flush_interval + 2.0)
should_flush = ( # 主线程排空队列中剩余条目
len(batch) >= self._batch_size or self._drain_queue_to_db()
self._flush_signal.is_set() or
(batch and time.monotonic() - last_flush >= self._flush_interval) def _drain_queue_to_db(self):
) """排空主线程可见的队列中剩余条目,直接发送到 db。"""
if should_flush and batch: batch = []
self._send(batch) while True:
batch = [] try:
last_flush = time.monotonic() batch.append(self._queue.get_nowait())
if len(batch) >= self._batch_size:
# 退出前 flush 剩余 self._send(batch)
if batch: batch = []
self._send(batch) except queue.Empty:
break
def _send(self, batch): if batch:
"""通过 pymsg RPC(execute 路由)逐条 INSERT 到 dbv2 Logs 表。""" self._send(batch)
now = datetime.now(timezone.utc).isoformat()
client = RpcClient(_DBV2_ADDR, _DBV2_PORT, self._service_name + "-log") # ── 内部实现 ────────────────────────────────────────────────────────
for entry in batch:
extra_val = entry.get("extra") def _flush_loop(self):
payload = { """后台线程:积攒到 batch_size 或超时 flush_interval 时发送。
"action": "insert",
"table": "Logs", 外层捕获所有异常,保证后台线程在任何情况下(网络故障、序列化错误等)
"values": { 都不会死亡,避免日志队列无限增长导致 OOM。
"ts": now, """
"service": entry.get("service"), batch = []
"level": entry.get("level"), last_flush = time.monotonic()
"message": entry.get("message"),
"trace_id": entry.get("trace_id"), while not self._stop.is_set():
"extra": json.dumps(extra_val, ensure_ascii=False) try:
if extra_val is not None else None, timeout = self._flush_interval - (time.monotonic() - last_flush)
}, chk_timeout = max(0.1, min(timeout, 0.5))
} try:
try: item = self._queue.get(timeout=chk_timeout)
client.call("execute", payload, timeout=_TIMEOUT) batch.append(item)
except RpcError as e: while len(batch) < self._batch_size:
# 业务错误(如表结构不符):打印后继续下一条 try:
print(f"[log_client] INSERT 失败: {e}", file=_sys.stderr) batch.append(self._queue.get_nowait())
except Exception as e: except queue.Empty:
# 网络错误:db 不可用,放弃本批剩余(fail-safe) break
print(f"[log_client] 发送失败: {e}", file=_sys.stderr) except queue.Empty:
break pass
def _shutdown(self): should_flush = (
"""进程退出时 flush 剩余日志。""" len(batch) >= self._batch_size or
self._stop.set() self._flush_signal.is_set() or
self.flush() (batch and time.monotonic() - last_flush >= self._flush_interval)
)
if should_flush and batch:
# ── 全局单例 ────────────────────────────────────────────────────────── try:
self._send(batch)
logs = LogClient() except Exception as e:
# _send 内部已有细粒度异常处理,此处是最终安全网
print(f"[log_client] _send 异常(已捕获,线程继续运行): {e}",
file=_sys.stderr)
batch = []
last_flush = time.monotonic()
if self._flush_signal.is_set():
self._flush_signal.clear()
self._flush_done.set()
except Exception as e:
# 最外层安全网:任何意外异常都不应杀死日志线程
print(f"[log_client] _flush_loop 异常(已捕获,线程继续运行): {e}",
file=_sys.stderr)
time.sleep(0.5) # 防止异常风暴导致 CPU 满载
# 退出前 flush 剩余
if batch:
try:
self._send(batch)
except Exception as e:
print(f"[log_client] 退出时 flush 异常: {e}", file=_sys.stderr)
self._flush_done.set()
def _send(self, batch):
"""通过 pymsg RPC(execute 路由)批量发送到 dbv2 Logs 表。
整个 batch 复用一个 TCP 连接;网络错误时放弃本批剩余条目(fail-safe),
不抛异常给调用方。
"""
try:
client = RpcClient(_DBV2_ADDR, _DBV2_PORT, self._service_name + "-log")
except Exception as e:
print(f"[log_client] 创建 RpcClient 失败: {e}", file=_sys.stderr)
return
for entry in batch:
extra_val = entry.get("extra")
try:
extra_json = _safe_json(extra_val) if extra_val is not None else None
except Exception:
extra_json = None
payload = {
"action": "insert",
"table": "Logs",
"values": {
"ts": entry.get("ts", _now_iso()),
"service": entry.get("service"),
"level": entry.get("level"),
"message": entry.get("message"),
"trace_id": entry.get("trace_id"),
"extra": extra_json,
},
}
try:
client.call("execute", payload, timeout=_TIMEOUT)
except RpcError as e:
# 业务错误(如表结构不符):打印后继续下一条
print(f"[log_client] INSERT 失败: {e}", file=_sys.stderr)
except Exception as e:
# 网络错误:db 不可用,放弃本批剩余(fail-safe)
print(f"[log_client] 发送失败: {e}", file=_sys.stderr)
break
def _shutdown(self):
"""进程退出时 flush 剩余日志。"""
self._stop.set()
self.flush()
# ── 全局单例 ──────────────────────────────────────────────────────────
logs = LogClient()