rust/greptimedb
1
0
Fork 0
mirror of https://github.com/GreptimeTeam/greptimedb.git synced 2026-06-02 05:10:40 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore: return 404 if trace not found (#7304)

Browse Source 
* chore: return 404 if trace not found

Signed-off-by: shuiyisong <xixing.sys@gmail.com>

* chore: add test and fix

Signed-off-by: shuiyisong <xixing.sys@gmail.com>

---------

Signed-off-by: shuiyisong <xixing.sys@gmail.com>
This commit is contained in:
shuiyisong
2025-11-26 17:39:28 +08:00
committed by GitHub
parent 6485a26fa3
commit 5472bdfc0f
2 changed files with 51 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
src/servers/src/http/jaeger.rs
View File

@@ -56,6 +56,9 @@ pub const JAEGER_QUERY_TABLE_NAME_KEY: &str = "jaeger_query_table_name";
const REF_TYPE_CHILD_OF: &str = "CHILD_OF";
const SPAN_KIND_TIME_FMTS: [&str; 2] = ["%Y-%m-%d %H:%M:%S%.6f%z", "%Y-%m-%d %H:%M:%S%.9f%z"];
const TRACE_NOT_FOUND_ERROR_CODE: i32 = 404;
const TRACE_NOT_FOUND_ERROR_MSG: &str = "trace not found";
/// JaegerAPIResponse is the response of Jaeger HTTP API.
/// The original version is `structuredResponse` which is defined in https://github.com/jaegertracing/jaeger/blob/main/cmd/query/app/http_handler.go.
#[derive(Default, Debug, Serialize, Deserialize, PartialEq)]
@@ -67,6 +70,22 @@ pub struct JaegerAPIResponse {
pub errors: Vec<JaegerAPIError>,
}
impl JaegerAPIResponse {
pub fn trace_not_found() -> Self {
Self {
data: None,
total: 0,
limit: 0,
offset: 0,
errors: vec![JaegerAPIError {
code: TRACE_NOT_FOUND_ERROR_CODE,
msg: TRACE_NOT_FOUND_ERROR_MSG.to_string(),
trace_id: None,
}],
}
}
}
/// JaegerData is the query result of Jaeger HTTP API.
#[derive(Debug, Serialize, Deserialize, PartialEq)]
#[serde(untagged)]
@@ -474,6 +493,10 @@ pub async fn handle_get_trace(
match covert_to_records(output).await {
Ok(Some(records)) => match traces_from_records(records) {
Ok(traces) if traces.is_empty() => (
HttpStatusCode::NOT_FOUND,
axum::Json(JaegerAPIResponse::trace_not_found()),
),
Ok(traces) => (
HttpStatusCode::OK,
axum::Json(JaegerAPIResponse {
@@ -486,7 +509,10 @@ pub async fn handle_get_trace(
error_response(err)
}
},
Ok(None) => (HttpStatusCode::OK, axum::Json(JaegerAPIResponse::default())),
Ok(None) => (
HttpStatusCode::NOT_FOUND,
axum::Json(JaegerAPIResponse::trace_not_found()),
),
Err(err) => {
error!("Failed to get trace '{}': {:?}", trace_id, err);
error_response(err)