Formatting
This commit is contained in:
parent
d8e00da078
commit
464dba46a8
|
@ -1,4 +1,4 @@
|
||||||
use std::{io, env, fmt, error::Error};
|
use std::{env, error::Error, fmt, io};
|
||||||
use toml::{self, de};
|
use toml::{self, de};
|
||||||
|
|
||||||
#[derive(Debug)]
|
#[derive(Debug)]
|
||||||
|
|
82
src/main.rs
82
src/main.rs
|
@ -14,11 +14,11 @@ use prometheus::{
|
||||||
};
|
};
|
||||||
use reqwest::header::USER_AGENT;
|
use reqwest::header::USER_AGENT;
|
||||||
use serde::Deserialize;
|
use serde::Deserialize;
|
||||||
use vrchat_prometheus_adapter::WsError;
|
|
||||||
use std::collections::HashMap;
|
use std::collections::HashMap;
|
||||||
use std::sync::OnceLock;
|
use std::sync::OnceLock;
|
||||||
use std::{env, fs};
|
use std::{env, fs};
|
||||||
use url::Url;
|
use url::Url;
|
||||||
|
use vrchat_prometheus_adapter::WsError;
|
||||||
|
|
||||||
static PLAYER_COUNT: OnceLock<GaugeVec> = OnceLock::new();
|
static PLAYER_COUNT: OnceLock<GaugeVec> = OnceLock::new();
|
||||||
static VRCDN_VIEWERS: OnceLock<GaugeVec> = OnceLock::new();
|
static VRCDN_VIEWERS: OnceLock<GaugeVec> = OnceLock::new();
|
||||||
|
@ -89,46 +89,56 @@ struct VrcWorldData {
|
||||||
}
|
}
|
||||||
|
|
||||||
fn register_metrics() {
|
fn register_metrics() {
|
||||||
let _ = PLAYER_COUNT.set(
|
PLAYER_COUNT
|
||||||
register_gauge_vec!(
|
.set(
|
||||||
"vrchat_playercount",
|
register_gauge_vec!(
|
||||||
"Current number viewers according to VRCDN's API.",
|
"vrchat_playercount",
|
||||||
&["instance", "world", "name", "group"],
|
"Current number viewers according to VRCDN's API.",
|
||||||
|
&["instance", "world", "name", "group"],
|
||||||
|
)
|
||||||
|
.unwrap(),
|
||||||
)
|
)
|
||||||
.unwrap(),
|
.unwrap();
|
||||||
);
|
VRCDN_VIEWERS
|
||||||
let _ = VRCDN_VIEWERS.set(
|
.set(
|
||||||
register_gauge_vec!(
|
register_gauge_vec!(
|
||||||
"vrcdn_viewers",
|
"vrcdn_viewers",
|
||||||
"Current number viewers according to VRCDN's API.",
|
"Current number viewers according to VRCDN's API.",
|
||||||
&["region", "group"],
|
&["region", "group"],
|
||||||
|
)
|
||||||
|
.unwrap(),
|
||||||
)
|
)
|
||||||
.unwrap(),
|
.unwrap();
|
||||||
);
|
WORLD_VISITS
|
||||||
let _ = WORLD_VISITS.set(
|
.set(
|
||||||
register_int_counter_vec!(
|
register_int_counter_vec!(
|
||||||
"vrchat_world_visits",
|
"vrchat_world_visits",
|
||||||
"Number of times a world has been visited.",
|
"Number of times a world has been visited.",
|
||||||
&["world", "name"],
|
&["world", "name"],
|
||||||
|
)
|
||||||
|
.unwrap(),
|
||||||
)
|
)
|
||||||
.unwrap(),
|
.unwrap();
|
||||||
);
|
WORLD_OCCUPANTS
|
||||||
let _ = WORLD_OCCUPANTS.set(
|
.set(
|
||||||
register_gauge_vec!(
|
register_gauge_vec!(
|
||||||
"vrchat_world_occupants",
|
"vrchat_world_occupants",
|
||||||
"Occupants currently in the world",
|
"Occupants currently in the world",
|
||||||
&["world", "name", "type"]
|
&["world", "name", "type"]
|
||||||
|
)
|
||||||
|
.unwrap(),
|
||||||
)
|
)
|
||||||
.unwrap(),
|
.unwrap();
|
||||||
);
|
WORLD_FAVORITES
|
||||||
let _ = WORLD_FAVORITES.set(
|
.set(
|
||||||
register_int_counter_vec!(
|
register_int_counter_vec!(
|
||||||
"vrchat_world_favorites",
|
"vrchat_world_favorites",
|
||||||
"Number of times a world has been favorited.",
|
"Number of times a world has been favorited.",
|
||||||
&["world", "name"],
|
&["world", "name"],
|
||||||
|
)
|
||||||
|
.unwrap(),
|
||||||
)
|
)
|
||||||
.unwrap(),
|
.unwrap();
|
||||||
);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn load_config() -> Result<Config, WsError> {
|
async fn load_config() -> Result<Config, WsError> {
|
||||||
|
|
Loading…
Reference in a new issue