axum-extra: Pre-allocate right sized buffer when encoding protobuf response (#3325)

jhgg · web-flow · commit 7b04b174b9de · 2025-04-26T09:37:32.000+02:00
diff --git a/axum-extra/src/protobuf.rs b/axum-extra/src/protobuf.rs
@@ -128,7 +128,7 @@ where
     T: Message + Default,
 {
     fn into_response(self) -> Response {
-        let mut buf = BytesMut::with_capacity(128);
+        let mut buf = BytesMut::with_capacity(self.0.encoded_len());
         match &self.0.encode(&mut buf) {
             Ok(()) => buf.into_response(),
             Err(err) => (StatusCode::INTERNAL_SERVER_ERROR, err.to_string()).into_response(),

Original file line number	Diff line number	Diff line change
`@@ -128,7 +128,7 @@ where`
`128`	`128`	`T: Message + Default,`
`129`	`129`	`{`
`130`	`130`	`fn into_response(self) -> Response {`
`131`		`- let mut buf = BytesMut::with_capacity(128);`
	`131`	`+ let mut buf = BytesMut::with_capacity(self.0.encoded_len());`
`132`	`132`	`match &self.0.encode(&mut buf) {`
`133`	`133`	`Ok(()) => buf.into_response(),`
`134`	`134`	`Err(err) => (StatusCode::INTERNAL_SERVER_ERROR, err.to_string()).into_response(),`