mirror of
https://github.com/fastapi-practices/fastapi-best-architecture.git
synced 2026-10-12 20:27:15 +00:00
38 lines
1.3 KiB
Python
38 lines
1.3 KiB
Python
from typing import Any
|
|
|
|
from prometheus_client import Gauge
|
|
from sqlalchemy.pool import QueuePool
|
|
|
|
from backend.core.conf import settings
|
|
|
|
_PROMETHEUS_SQLALCHEMY_POOL_CONNECTIONS_GAUGE = Gauge(
|
|
name='fba_sqlalchemy_pool_connections',
|
|
documentation='SQLAlchemy 连接池状态',
|
|
labelnames=['app_name', 'source', 'state'],
|
|
)
|
|
|
|
|
|
def observe_sqlalchemy_pool_connections(
|
|
*_event_args: Any,
|
|
pool: QueuePool,
|
|
source: str = 'default',
|
|
) -> None:
|
|
"""监听 SQLAlchemy 连接池状态"""
|
|
total_size = pool.size()
|
|
checked_out_size = pool.checkedout()
|
|
overflow_size = pool.overflow()
|
|
idle_size = max(total_size + overflow_size - checked_out_size, 0)
|
|
|
|
_PROMETHEUS_SQLALCHEMY_POOL_CONNECTIONS_GAUGE.labels(
|
|
app_name=settings.GRAFANA_PROMETHEUS_APP_NAME, source=source, state='size'
|
|
).set(total_size)
|
|
_PROMETHEUS_SQLALCHEMY_POOL_CONNECTIONS_GAUGE.labels(
|
|
app_name=settings.GRAFANA_PROMETHEUS_APP_NAME, source=source, state='checked_out'
|
|
).set(checked_out_size)
|
|
_PROMETHEUS_SQLALCHEMY_POOL_CONNECTIONS_GAUGE.labels(
|
|
app_name=settings.GRAFANA_PROMETHEUS_APP_NAME, source=source, state='idle'
|
|
).set(idle_size)
|
|
_PROMETHEUS_SQLALCHEMY_POOL_CONNECTIONS_GAUGE.labels(
|
|
app_name=settings.GRAFANA_PROMETHEUS_APP_NAME, source=source, state='overflow'
|
|
).set(overflow_size)
|