91d922d0b8
added env variable SPOOLMAN_METRICS_ENABLED for enabled database collector set metrics path as /metrics checked Null values at vendors, init weight and price removed some logs
87 lines
2.8 KiB
Python
87 lines
2.8 KiB
Python
"""Prometheus metrics collectors."""
|
|
import logging
|
|
from typing import Callable
|
|
|
|
import sqlalchemy
|
|
from prometheus_client import REGISTRY, Gauge, make_asgi_app
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
from sqlalchemy.orm import contains_eager
|
|
|
|
from spoolman.database import models
|
|
|
|
registry = REGISTRY
|
|
|
|
PREFIX = "spoolman"
|
|
|
|
SPOOL_PRICE = Gauge("%s_spool_price" % PREFIX, "Total Spool price", ["spool_id", "filament_id"])
|
|
SPOOL_USED_WEIGHT = Gauge("%s_spool_weight_used" % PREFIX, "Spool Used Weight", ["spool_id", "filament_id"])
|
|
FILAMENT_INFO = Gauge(
|
|
"%s_filament_info" % PREFIX,
|
|
"Filament information",
|
|
["filament_id", "vendor", "name", "material", "color"],
|
|
)
|
|
FILAMENT_DENSITY = Gauge("%s_filament_density" % PREFIX, "Density of filament", ["filament_id"])
|
|
FILAMENT_DIAMETER = Gauge("%s_filament_diameter" % PREFIX, "Diameter of filament", ["filament_id"])
|
|
FILAMENT_WEIGHT = Gauge("%s_filament_weight" % PREFIX, "Net weight of filament", ["filament_id"])
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
def make_metrics_app() -> Callable:
|
|
"""Start ASGI prometheus app with global registry."""
|
|
logger.info("Start metrics app")
|
|
return make_asgi_app(registry=registry)
|
|
|
|
|
|
metrics_app = make_asgi_app()
|
|
|
|
|
|
async def spool_metrics(db: AsyncSession) -> None:
|
|
"""Get metrics by Spools from DB and write to prometheus.
|
|
|
|
Args:
|
|
db: async db session
|
|
"""
|
|
stmt = sqlalchemy.select(models.Spool).where(
|
|
sqlalchemy.or_(
|
|
models.Spool.archived.is_(False),
|
|
models.Spool.archived.is_(None),
|
|
),
|
|
)
|
|
rows = await db.execute(stmt)
|
|
result = list(rows.unique().scalars().all())
|
|
for row in result:
|
|
if row.price is not None:
|
|
SPOOL_PRICE.labels(str(row.id), str(row.filament_id)).set(row.price)
|
|
SPOOL_USED_WEIGHT.labels(str(row.id), str(row.filament_id)).set(row.used_weight)
|
|
|
|
|
|
async def filament_metrics(db: AsyncSession) -> None:
|
|
"""Get metrics and info by Filaments from DB and write to prometheus.
|
|
|
|
Args:
|
|
db: async db session
|
|
"""
|
|
stmt = (
|
|
sqlalchemy.select(models.Filament)
|
|
.options(contains_eager(models.Filament.vendor))
|
|
.join(models.Filament.vendor, isouter=True)
|
|
)
|
|
rows = await db.execute(stmt)
|
|
result = list(rows.unique().scalars().all())
|
|
for row in result:
|
|
vendor_name = "-"
|
|
if row.vendor is not None:
|
|
vendor_name = row.vendor.name
|
|
FILAMENT_INFO.labels(
|
|
str(row.id),
|
|
vendor_name,
|
|
row.name,
|
|
row.material,
|
|
row.color_hex,
|
|
).set(1)
|
|
FILAMENT_DENSITY.labels(str(row.id)).set(row.density)
|
|
FILAMENT_DIAMETER.labels(str(row.id)).set(row.diameter)
|
|
if row.weight is not None:
|
|
FILAMENT_WEIGHT.labels(str(row.id)).set(row.weight)
|