113 lines
2.7 KiB
Python
113 lines
2.7 KiB
Python
|
"""Prometheus 指标定义：发送计数/耗时/失败/图像匹配/DB校验/队列/运行态/熔断/缓存/自适应等待。
|
|||
|
|
|
|||
|
|
所有指标用 prometheus_client 定义，通过 /metrics 端点暴露（app.py 挂载）。
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
from __future__ import annotations
|
|||
|
|
|
|||
|
|
import logging
|
|||
|
|
import time
|
|||
|
|
|
|||
|
|
from prometheus_client import (
|
|||
|
|
Counter,
|
|||
|
|
Gauge,
|
|||
|
|
Histogram,
|
|||
|
|
Info,
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
logger = logging.getLogger("woc-bridge")
|
|||
|
|
|
|||
|
|
# 发送总数（按 flow_name / result 标签）
|
|||
|
|
send_total = Counter(
|
|||
|
|
"woc_send_total",
|
|||
|
|
"Total send operations",
|
|||
|
|
["flow_name", "result"], # result: success / failed / skipped
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
# 发送耗时分布
|
|||
|
|
send_duration = Histogram(
|
|||
|
|
"woc_send_duration_seconds",
|
|||
|
|
"Send operation duration in seconds",
|
|||
|
|
["flow_name"],
|
|||
|
|
buckets=(0.5, 1.0, 2.0, 5.0, 10.0, 30.0, 60.0),
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
# 按失败状态分类的失败计数
|
|||
|
|
send_failed_by_state = Counter(
|
|||
|
|
"woc_send_failed_by_state_total",
|
|||
|
|
"Send failures by flow state",
|
|||
|
|
["flow_name", "state"],
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
# 图像匹配置信度分布
|
|||
|
|
image_match_confidence = Histogram(
|
|||
|
|
"woc_image_match_confidence",
|
|||
|
|
"OpenCV template match confidence",
|
|||
|
|
["kind"],
|
|||
|
|
buckets=(0.5, 0.7, 0.8, 0.85, 0.9, 0.95, 1.0),
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
# DB 校验耗时
|
|||
|
|
db_verify_duration = Histogram(
|
|||
|
|
"woc_db_verify_duration_seconds",
|
|||
|
|
"DB verify duration in seconds",
|
|||
|
|
buckets=(0.1, 0.2, 0.5, 1.0, 2.0, 3.0, 5.0),
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
# 发送队列待处理数
|
|||
|
|
send_queue_pending = Gauge(
|
|||
|
|
"woc_send_queue_pending",
|
|||
|
|
"Send queue pending count",
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
# WeChat 运行态（1=running, 0=not running）
|
|||
|
|
wechat_running = Gauge(
|
|||
|
|
"woc_wechat_running",
|
|||
|
|
"WeChat process running state (1=running, 0=not)",
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
# 熔断器状态（0=closed, 1=open, 2=half_open）
|
|||
|
|
circuit_state = Gauge(
|
|||
|
|
"woc_circuit_state",
|
|||
|
|
"Circuit breaker state (0=closed, 1=open, 2=half_open)",
|
|||
|
|
["name"],
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
# 会话缓存命中
|
|||
|
|
session_cache_hit = Counter(
|
|||
|
|
"woc_session_cache_hit_total",
|
|||
|
|
"Session cache hit count",
|
|||
|
|
["result"], # hit / miss
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
# 自适应等待超时计数
|
|||
|
|
adaptive_wait_timeout = Counter(
|
|||
|
|
"woc_adaptive_wait_timeout_total",
|
|||
|
|
"Adaptive wait timeout count",
|
|||
|
|
["kind"],
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
# 首次发送耗时（无缓存）
|
|||
|
|
send_duration_first = Histogram(
|
|||
|
|
"woc_send_duration_first_seconds",
|
|||
|
|
"First send duration (no session cache) in seconds",
|
|||
|
|
buckets=(1.0, 3.0, 5.0, 10.0, 30.0, 60.0),
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
# 缓存命中后的发送耗时
|
|||
|
|
send_duration_cached = Histogram(
|
|||
|
|
"woc_send_duration_cached_seconds",
|
|||
|
|
"Cached send duration (session cache hit) in seconds",
|
|||
|
|
buckets=(0.5, 1.0, 2.0, 5.0, 10.0, 30.0),
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
|
|||
|
|
def set_circuit_state(name: str, state_int: int) -> None:
|
|||
|
|
"""更新熔断器状态指标。
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
name: 熔断器名称
|
|||
|
|
state_int: 0=closed, 1=open, 2=half_open
|
|||
|
|
"""
|
|||
|
|
circuit_state.labels(name=name).set(state_int)
|