minor healthcheck/docker fixes
This commit is contained in:
parent
cecaecb92e
commit
3003f4cf78
2 changed files with 27 additions and 13 deletions
|
@ -36,9 +36,9 @@ services:
|
|||
links:
|
||||
- tsdb
|
||||
|
||||
prune:
|
||||
task_prune:
|
||||
image: ghcr.io/genudine/saerro/tasks:latest
|
||||
command: prune
|
||||
command: /app prune
|
||||
pull_policy: always
|
||||
restart: "no"
|
||||
environment:
|
||||
|
|
|
@ -71,25 +71,39 @@ impl Health {
|
|||
}
|
||||
}
|
||||
|
||||
/// Is the websocket connection to the Nanite Systems manifold up?
|
||||
async fn websocket<'ctx>(&self, ctx: &Context<'ctx>) -> UpDown {
|
||||
/// Is the websocket processing jobs?
|
||||
async fn ingest<'ctx>(&self, ctx: &Context<'ctx>) -> UpDown {
|
||||
let pool = ctx.data::<Pool<Postgres>>().unwrap();
|
||||
|
||||
match query("SELECT count(*) FROM analytics WHERE time > now() - interval '5 minutes'")
|
||||
.fetch_one(pool)
|
||||
.await
|
||||
{
|
||||
Ok(i) => {
|
||||
let num: i64 = i.get(0);
|
||||
if num == 0 {
|
||||
UpDown::Down
|
||||
let events_resp =
|
||||
query("SELECT count(*) FROM analytics WHERE time > now() - interval '5 minutes'")
|
||||
.fetch_one(pool)
|
||||
.await;
|
||||
|
||||
match events_resp {
|
||||
Ok(row) => {
|
||||
let events_row: i64 = row.get(0);
|
||||
|
||||
if events_row == 0 {
|
||||
return UpDown::Down;
|
||||
} else {
|
||||
UpDown::Up
|
||||
return UpDown::Up;
|
||||
}
|
||||
}
|
||||
Err(_) => UpDown::Down,
|
||||
}
|
||||
}
|
||||
|
||||
/// Is the websocket actually turned on?
|
||||
async fn ingest_reachable(&self) -> UpDown {
|
||||
reqwest::get(
|
||||
std::env::var("WEBSOCKET_HEALTHCHECK")
|
||||
.unwrap_or("http://localhost:8999/health".to_string()),
|
||||
)
|
||||
.await
|
||||
.map(|_| UpDown::Up)
|
||||
.unwrap_or(UpDown::Down)
|
||||
}
|
||||
}
|
||||
|
||||
#[derive(Default)]
|
||||
|
|
Loading…
Add table
Reference in a new issue