mirror of
https://github.com/GreptimeTeam/greptimedb.git
synced 2026-07-04 04:50:37 +00:00
fix: add owned query frontend key conversion
Signed-off-by: jeremyhi <fengjiachun@gmail.com>
This commit is contained in:
@@ -73,6 +73,20 @@ impl From<&QueryFrontendRequest> for QueryKey {
|
||||
}
|
||||
}
|
||||
|
||||
impl From<QueryFrontendRequest> for QueryKey {
|
||||
fn from(request: QueryFrontendRequest) -> Self {
|
||||
Self {
|
||||
db: request.db,
|
||||
read_preference: request.read_preference,
|
||||
query: request.query,
|
||||
start: request.start,
|
||||
end: request.end,
|
||||
step: request.step,
|
||||
lookback: request.lookback,
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
@@ -163,6 +177,16 @@ mod tests {
|
||||
assert_eq!("db", request.key().db());
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn owned_request_converts_into_key() {
|
||||
let request = test_request();
|
||||
let borrowed_key = request.key();
|
||||
let owned_key = request.into_key();
|
||||
|
||||
assert_eq!(borrowed_key, owned_key);
|
||||
assert_eq!("db", owned_key.db());
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn debug_redacts_query_text() {
|
||||
let request = QueryFrontendRequest {
|
||||
|
||||
@@ -54,6 +54,11 @@ impl QueryFrontendRequest {
|
||||
pub fn key(&self) -> QueryKey {
|
||||
QueryKey::from(self)
|
||||
}
|
||||
|
||||
/// Converts this request into its exact identity [`QueryKey`].
|
||||
pub fn into_key(self) -> QueryKey {
|
||||
QueryKey::from(self)
|
||||
}
|
||||
}
|
||||
|
||||
impl fmt::Debug for QueryFrontendRequest {
|
||||
|
||||
Reference in New Issue
Block a user