use std::sync::Arc; use hyper::{Body, Response, StatusCode}; use serde::Serialize; use garage_util::data::*; use garage_util::error::Error as GarageError; use garage_rpc::ring::Ring; use garage_table::util::*; use garage_model::garage::Garage; use garage_model::k2v::counter_table::{BYTES, CONFLICTS, ENTRIES, VALUES}; use crate::error::*; use crate::k2v::range::read_range; pub async fn handle_read_index( garage: Arc, bucket_id: Uuid, prefix: Option, start: Option, end: Option, limit: Option, reverse: Option, ) -> Result, Error> { let reverse = reverse.unwrap_or(false); let ring: Arc = garage.system.ring.borrow().clone(); let (partition_keys, more, next_start) = read_range( &garage.k2v.counter_table.table, &bucket_id, &prefix, &start, &end, limit, Some((DeletedFilter::NotDeleted, ring.layout.node_id_vec.clone())), EnumerationOrder::from_reverse(reverse), ) .await?; let s_entries = ENTRIES.to_string(); let s_conflicts = CONFLICTS.to_string(); let s_values = VALUES.to_string(); let s_bytes = BYTES.to_string(); let resp = ReadIndexResponse { prefix, start, end, limit, reverse, partition_keys: partition_keys .into_iter() .map(|part| { let vals = part.filtered_values(&ring); ReadIndexResponseEntry { pk: part.sk, entries: *vals.get(&s_entries).unwrap_or(&0), conflicts: *vals.get(&s_conflicts).unwrap_or(&0), values: *vals.get(&s_values).unwrap_or(&0), bytes: *vals.get(&s_bytes).unwrap_or(&0), } }) .collect::>(), more, next_start, }; let resp_json = serde_json::to_string_pretty(&resp).map_err(GarageError::from)?; Ok(Response::builder() .status(StatusCode::OK) .body(Body::from(resp_json))?) } #[derive(Serialize)] struct ReadIndexResponse { prefix: Option, start: Option, end: Option, limit: Option, reverse: bool, #[serde(rename = "partitionKeys")] partition_keys: Vec, more: bool, #[serde(rename = "nextStart")] next_start: Option, } #[derive(Serialize)] struct ReadIndexResponseEntry { pk: String, entries: i64, conflicts: i64, values: i64, bytes: i64, }