fix: correct the error message for snappy compress (#2956)

Signed-off-by: Zhenchi <[email protected]>
GreptimeTeam · Dec 19, 2023 · ccbd497 · ccbd497
1 parent 29fc2ea
commit ccbd497
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/src/servers/src/error.rs b/src/servers/src/error.rs
@@ -214,6 +214,13 @@ pub enum Error {
         error: snap::Error,
     },
 
+    #[snafu(display("Failed to compress prometheus remote request"))]
+    CompressPromRemoteRequest {
+        location: Location,
+        #[snafu(source)]
+        error: snap::Error,
+    },
+
     #[snafu(display("Invalid prometheus remote request, msg: {}", msg))]
     InvalidPromRemoteRequest { msg: String, location: Location },
 
@@ -444,6 +451,7 @@ impl ErrorExt for Error {
             | InvalidOpentsdbJsonRequest { .. }
             | DecodePromRemoteRequest { .. }
             | DecodeOtlpRequest { .. }
+            | CompressPromRemoteRequest { .. }
             | DecompressPromRemoteRequest { .. }
             | InvalidPromRemoteRequest { .. }
             | InvalidFlightTicket { .. }

diff --git a/src/servers/src/prom_store.rs b/src/servers/src/prom_store.rs
@@ -370,7 +370,7 @@ pub fn snappy_compress(buf: &[u8]) -> Result<Vec<u8>> {
     let mut encoder = Encoder::new();
     encoder
         .compress_vec(buf)
-        .context(error::DecompressPromRemoteRequestSnafu)
+        .context(error::CompressPromRemoteRequestSnafu)
 }
 
 /// Mock timeseries for test, it is both used in servers and frontend crate