ps2.live/saerro
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

add health checks to gql

Browse source
This commit is contained in:
41666 2022-11-26 01:37:37 -05:00
parent 4d20b20ad8
commit 318ffde050
3 changed files with 93 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

8
services/websocket/src/main.rs
View file

@ -120,14 +120,16 @@ async fn track_class(class_event: ClassEvent) {
fn should_process_event() -> bool {
let mut con = REDIS_CLIENT.get_connection().unwrap();
let role: String = ROLE.parse().unwrap();
let heartbeat_key = format!("heartbeat:{}", PAIR.to_string());
let heartbeat_key = format!("heartbeat:{}:{}", PAIR.to_string(), role);
let _: () = con.set_ex(heartbeat_key, "1", 60).unwrap();
if role == "primary" {
let _: () = con.set_ex(heartbeat_key, "1", 60).unwrap();
return false;
}
match con.get(heartbeat_key) {
let primary_heartbeat_key = format!("heartbeat:{}:primary", PAIR.to_string());
match con.get(primary_heartbeat_key) {
Ok(1) => true,
_ => false,
}