metrics.py

from prometheus_client.metrics import Counter, Gauge, Histogram, Info
from variables import DEPOSIT_MODULES_WHITELIST, PROMETHEUS_PREFIX, PUBLIC_ENV_VARS

GAS_FEE = Gauge(
    'gas_fee',
    'Gas fee',
    ['type', 'module_id'],
    namespace=PROMETHEUS_PREFIX,
)

TX_SEND = Counter('transactions_send', 'Amount of send transaction from bot.', ['status'], namespace=PROMETHEUS_PREFIX)

# Initialize metrics
TX_SEND.labels('success').inc(0)
TX_SEND.labels('failure').inc(0)

MODULE_TX_SEND = Counter(
    'transactions',
    'Amount of send transactions from depositor bot.',
    ['status', 'module_id'],
    namespace=PROMETHEUS_PREFIX,
)

DEPOSIT_MESSAGES = Gauge(
    'deposit_messages',
    'Guardians deposit messages',
    ['address', 'module_id', 'version', 'transport', 'chain_id'],
    namespace=PROMETHEUS_PREFIX,
)
PAUSE_MESSAGES = Gauge(
    'pause_messages',
    'Guardians pause messages',
    ['address', 'module_id', 'version', 'transport', 'chain_id'],
    namespace=PROMETHEUS_PREFIX,
)
PING_MESSAGES = Gauge(
    'ping_messages',
    'Guardians ping messages',
    ['address', 'version', 'transport', 'chain_id'],
    namespace=PROMETHEUS_PREFIX,
)
UNVET_MESSAGES = Gauge(
    'unvet_messages',
    'Guardian unvet messages',
    ['address', 'module_id', 'version', 'transport', 'chain_id'],
    namespace=PROMETHEUS_PREFIX,
)

CURRENT_QUORUM_SIZE = Gauge(
    'quorum_size',
    'Current quorum size',
    ['type'],
    namespace=PROMETHEUS_PREFIX,
)

DEPOSITABLE_ETHER = Gauge(
    'depositable_ether',
    'Depositable Ether',
    namespace=PROMETHEUS_PREFIX,
)

POSSIBLE_DEPOSITS_AMOUNT = Gauge(
    'possible_deposits_amount',
    'Possible deposits amount.',
    ['module_id'],
    namespace=PROMETHEUS_PREFIX,
)

IS_DEPOSITABLE = Gauge(
    'is_depositable',
    'Represents is_depositable check.',
    ['module_id'],
    namespace=PROMETHEUS_PREFIX,
)

QUORUM = Gauge(
    'quorum',
    'Represents if quorum could be collected.',
    ['module_id'],
    namespace=PROMETHEUS_PREFIX,
)

CAN_DEPOSIT = Gauge(
    'can_deposit',
    'Represents can_deposit check.',
    ['module_id'],
    namespace=PROMETHEUS_PREFIX,
)

GAS_OK = Gauge(
    'is_gas_ok',
    'Represents is_gas_ok check.',
    ['module_id'],
    namespace=PROMETHEUS_PREFIX,
)

DEPOSIT_AMOUNT_OK = Gauge(
    'is_deposit_amount_ok',
    'Represents is_deposit_amount_ok check.',
    ['module_id'],
    namespace=PROMETHEUS_PREFIX,
)

ETH_RPC_REQUESTS_DURATION = Histogram(
    'eth_rpc_requests_duration',
    'Duration of requests to ETH1 RPC',
    namespace=PROMETHEUS_PREFIX,
)

ETH_RPC_REQUESTS = Counter(
    'eth_rpc_requests',
    'Total count of requests to ETH1 RPC',
    ['method', 'code', 'domain'],
    namespace=PROMETHEUS_PREFIX,
)

ONCHAIN_TRANSPORT_ETH_RPC_REQUESTS = Counter(
    'onchain_transport_rpc_requests',
    'Total count of requests to onchain transport RPC',
    ['method', 'code', 'domain'],
    namespace=PROMETHEUS_PREFIX,
)

UNEXPECTED_EXCEPTIONS = Counter(
    'unexpected_exceptions',
    'Total count of unexpected exceptions',
    ['type'],
    namespace=PROMETHEUS_PREFIX,
)

# TODO unify ACCOUNT_BALANCE and GUARDIAN_BALANCE
ACCOUNT_BALANCE = Gauge(
    'account_balance',
    'Account balance',
    ['address', 'chain_id'],
    namespace=PROMETHEUS_PREFIX,
)

GUARDIAN_BALANCE = Gauge(
    'guardian_balance',
    'Balance of the guardian',
    ['address', 'chain_id'],
    namespace=PROMETHEUS_PREFIX,
)

MODULES = Gauge('modules', 'Modules gauge', ['module_id'], namespace=PROMETHEUS_PREFIX)

for module_id in DEPOSIT_MODULES_WHITELIST:
    MODULES.labels(module_id).set(1)

INFO = Info(name='build', documentation='Info metric', namespace=PROMETHEUS_PREFIX)
CONVERTED_PUBLIC_ENV = {k: str(v) for k, v in PUBLIC_ENV_VARS.items()}
INFO.info(CONVERTED_PUBLIC_ENV)