Add last_update information on global /stats route

This commit is contained in:
Quentin de Quelen 2019-11-20 11:21:55 +01:00
parent a0caf0d6d7
commit d01a3944c1

View File

@ -1,6 +1,7 @@
use std::collections::HashMap; use chrono::{DateTime, Utc};
use pretty_bytes::converter::convert; use pretty_bytes::converter::convert;
use serde::Serialize; use serde::Serialize;
use std::collections::HashMap;
use sysinfo::{NetworkExt, Pid, ProcessExt, ProcessorExt, System, SystemExt}; use sysinfo::{NetworkExt, Pid, ProcessExt, ProcessorExt, System, SystemExt};
use tide::{Context, Response}; use tide::{Context, Response};
use walkdir::WalkDir; use walkdir::WalkDir;
@ -55,6 +56,7 @@ pub async fn index_stat(ctx: Context<Data>) -> SResult<Response> {
#[serde(rename_all = "camelCase")] #[serde(rename_all = "camelCase")]
struct StatsResult { struct StatsResult {
database_size: u64, database_size: u64,
last_update: Option<DateTime<Utc>>,
indexes: HashMap<String, IndexStatsResponse>, indexes: HashMap<String, IndexStatsResponse>,
} }
@ -63,14 +65,15 @@ pub async fn get_stats(ctx: Context<Data>) -> SResult<Response> {
let mut index_list = HashMap::new(); let mut index_list = HashMap::new();
if let Ok(indexes_set) = ctx.state().db.indexes_uids() {
for index_uid in indexes_set {
let db = &ctx.state().db; let db = &ctx.state().db;
let env = &db.env; let env = &db.env;
let index = db.open_index(&index_uid).unwrap();
let reader = env.read_txn().map_err(ResponseError::internal)?; let reader = env.read_txn().map_err(ResponseError::internal)?;
let indexes_set = ctx.state().db.indexes_uids();
for index_uid in indexes_set {
let index = db.open_index(&index_uid).unwrap();
let number_of_documents = index let number_of_documents = index
.main .main
.number_of_documents(&reader) .number_of_documents(&reader)
@ -95,7 +98,6 @@ pub async fn get_stats(ctx: Context<Data>) -> SResult<Response> {
}; };
index_list.insert(index_uid, response); index_list.insert(index_uid, response);
} }
}
let database_size = WalkDir::new(ctx.state().db_path.clone()) let database_size = WalkDir::new(ctx.state().db_path.clone())
.into_iter() .into_iter()
@ -104,8 +106,14 @@ pub async fn get_stats(ctx: Context<Data>) -> SResult<Response> {
.filter(|metadata| metadata.is_file()) .filter(|metadata| metadata.is_file())
.fold(0, |acc, m| acc + m.len()); .fold(0, |acc, m| acc + m.len());
let last_update = ctx
.state()
.last_update(&reader)
.map_err(ResponseError::internal)?;
let response = StatsResult { let response = StatsResult {
database_size, database_size,
last_update,
indexes: index_list, indexes: index_list,
}; };