arch/vrchat-prometheus-adapter
1
0
Fork 0
Code Pull requests Actions Packages Activity

Reset world occupants, register metrics function

Browse source
This commit is contained in:
Gabriel Simmer 2023-10-25 21:17:42 +01:00
parent 0298508f45
commit 2297da8e73
Signed by: arch
SSH key fingerprint: SHA256:m3OEcdtrnBpMX+2BDGh/byv3hrCekCLzDYMdvGEKPPQ
1 changed files with 10 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
src/main.rs
View file

@ -119,12 +119,7 @@ struct VrcWorldData {
instances: Vec<(String, f64)>, instances: Vec<(String, f64)>,
} }
#[tokio::main] fn register_metrics() {
async fn main() -> Result<(), ()> {
let content = fs::read_to_string("config.toml").unwrap();
let mut config: Config = toml::from_str(&content).unwrap();
config.vrchat_token = Some(env::var("VRCHAT_AUTH_TOKEN").unwrap());
let _ = PLAYER_COUNT.set( let _ = PLAYER_COUNT.set(
register_gauge_vec!( register_gauge_vec!(
"vrchat_playercount", "vrchat_playercount",
@ -165,6 +160,14 @@ async fn main() -> Result<(), ()> {
) )
.unwrap(), .unwrap(),
); );
}
#[tokio::main]
async fn main() -> Result<(), ()> {
let content = fs::read_to_string("config.toml").unwrap();
let mut config: Config = toml::from_str(&content).unwrap();
config.vrchat_token = Some(env::var("VRCHAT_AUTH_TOKEN").unwrap());
register_metrics();
let app = Router::new() let app = Router::new()
.route("/", get(homepage)) .route("/", get(homepage))
@ -195,7 +198,7 @@ async fn homepage() -> &'static str {
async fn metrics(config: Config) -> Result<Vec<u8>, WsError> { async fn metrics(config: Config) -> Result<Vec<u8>, WsError> {
PLAYER_COUNT.get().unwrap().reset(); PLAYER_COUNT.get().unwrap().reset();
VRCDN_VIEWERS.get().unwrap().reset(); VRCDN_VIEWERS.get().unwrap().reset();
WORLD_VISITS.get().unwrap().reset(); WORLD_OCCUPANTS.get().unwrap().reset();
let encoder = TextEncoder::new(); let encoder = TextEncoder::new();
let client = reqwest::Client::new(); let client = reqwest::Client::new();
let auth_cookie = format!("auth={}", &config.vrchat_token.unwrap()); let auth_cookie = format!("auth={}", &config.vrchat_token.unwrap());