sui_bridge/sui_bridge_watchdog/
metrics.rs

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
// Copyright (c) Mysten Labs, Inc.
// SPDX-License-Identifier: Apache-2.0

use prometheus::{
    register_int_gauge_vec_with_registry, register_int_gauge_with_registry, IntGauge, IntGaugeVec,
    Registry,
};

#[derive(Clone, Debug)]
pub struct WatchdogMetrics {
    pub eth_vault_balance: IntGauge,
    pub usdt_vault_balance: IntGauge,
    pub wbtc_vault_balance: IntGauge,
    pub lbtc_vault_balance: IntGauge,
    pub total_supplies: IntGaugeVec,
    pub eth_bridge_paused: IntGauge,
    pub sui_bridge_paused: IntGauge,
}

impl WatchdogMetrics {
    pub fn new(registry: &Registry) -> Self {
        Self {
            eth_vault_balance: register_int_gauge_with_registry!(
                "bridge_eth_vault_balance",
                "Current balance of eth vault",
                registry,
            )
            .unwrap(),
            usdt_vault_balance: register_int_gauge_with_registry!(
                "bridge_usdt_vault_balance",
                "Current balance of usdt eth vault",
                registry,
            )
            .unwrap(),
            wbtc_vault_balance: register_int_gauge_with_registry!(
                "bridge_wbtc_vault_balance",
                "Current balance of wbtc eth vault",
                registry,
            )
            .unwrap(),
            lbtc_vault_balance: register_int_gauge_with_registry!(
                "bridge_lbtc_vault_balance",
                "Current balance of lbtc eth vault",
                registry,
            )
            .unwrap(),
            total_supplies: register_int_gauge_vec_with_registry!(
                "bridge_total_supplies",
                "Current total supplies of coins on Sui based on Treasury Cap",
                &["token_name"],
                registry,
            )
            .unwrap(),
            eth_bridge_paused: register_int_gauge_with_registry!(
                "bridge_eth_bridge_paused",
                "Whether the eth bridge is paused",
                registry,
            )
            .unwrap(),
            sui_bridge_paused: register_int_gauge_with_registry!(
                "bridge_sui_bridge_paused",
                "Whether the sui bridge is paused",
                registry,
            )
            .unwrap(),
        }
    }

    pub fn new_for_testing() -> Self {
        let registry = Registry::new();
        Self::new(&registry)
    }
}