Add CONTENT-TYPE header to server response (#31)

Author: fafhrd91

examples/helloworld/Cargo.toml

@@ -13,7 +13,7 @@ name = "server"
 path = "src/server.rs"
 
 [dependencies]
-ntex = "3.0.0-pre.2"
+ntex = { version = "3.0.0-pre.9", features = ["tokio"] }
 ntex-grpc = "1"
 ntex-h2 = "3"
 
@@ -23,4 +23,4 @@ env_logger = "0.11"
 num_cpus = "1"
 
 [dev-dependencies]
-ntex = { version = "3.0.0-pre.2", features = ["openssl", "tokio"] }
+ntex = { version = "3.0.0-pre.9", features = ["openssl", "tokio"] }

ntex-grpc/CHANGES.md

@@ -1,5 +1,9 @@
 # Changes
 
+## [1.2.0] - 2025-12-22
+
+* Add "CONTENT-TYPE" header to server response
+
 ## [1.1.0] - 2025-12-17
 
 * Upgrade to ntex-service v4

ntex-grpc/Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "ntex-grpc"
-version = "1.1.0"
+version = "1.2.0"
 license = "MIT OR Apache-2.0"
 authors = ["Nikolay Kim <fafhrd91@gmail.com>"]
 description = "GRPC Client/Server framework"

ntex-grpc/src/server/service.rs

@@ -2,7 +2,7 @@ use std::{cell::RefCell, rc::Rc};
 
 use ntex_bytes::{Buf, BufMut, ByteString, BytesMut};
 use ntex_h2::{self as h2, StreamRef, frame::Reason, frame::StreamId};
-use ntex_http::{HeaderMap, HeaderValue, StatusCode};
+use ntex_http::{HeaderMap, HeaderValue, StatusCode, header::CONTENT_TYPE};
 use ntex_io::{Filter, Io, IoBoxed};
 use ntex_service::{Service, ServiceCtx, ServiceFactory, cfg::SharedCfg};
 use ntex_util::{HashMap, time::Millis, time::timeout_checked};
@@ -18,6 +18,7 @@ const ERR_DATA_DECODE: HeaderValue =
 const ERR_DECODE_TIMEOUT: HeaderValue =
     HeaderValue::from_static("Cannot decode grpc-timeout header");
 const ERR_DEADLINE: HeaderValue = HeaderValue::from_static("Deadline exceeded");
+const HDR_APP_GRPC: HeaderValue = HeaderValue::from_static("application/grpc");
 
 const MILLIS_IN_HOUR: u64 = 60 * 60 * 1000;
 const MILLIS_IN_MINUTE: u64 = 60 * 1000;
@@ -187,17 +188,13 @@ where
                     path.split_to(n)
                 } else {
                     // not found
-                    let _ =
-                        stream.send_response(StatusCode::NOT_FOUND, HeaderMap::default(), true);
+                    let _ = stream.send_response(StatusCode::NOT_FOUND, hdrs(), true);
                     return Ok(());
                 };
 
                 // stream eof, cannot do anything
                 if eof {
-                    if stream
-                        .send_response(StatusCode::OK, HeaderMap::default(), false)
-                        .is_ok()
-                    {
+                    if stream.send_response(StatusCode::OK, hdrs(), false).is_ok() {
                         send_error(&stream, GrpcStatus::InvalidArgument, ERR_DECODE);
                     }
                     return Ok(());
@@ -241,10 +238,7 @@ where
                     let _compressed = data.get_u8();
                     let len = data.get_u32();
                     if (len as usize) > data.len() {
-                        if stream
-                            .send_response(StatusCode::OK, HeaderMap::default(), false)
-                            .is_ok()
-                        {
+                        if stream.send_response(StatusCode::OK, hdrs(), false).is_ok() {
                             send_error(&stream, GrpcStatus::InvalidArgument, ERR_DATA_DECODE);
                         }
                         return Ok(());
@@ -264,10 +258,7 @@ where
                         name: inflight.name,
                         headers: inflight.headers,
                     };
-                    if stream
-                        .send_response(StatusCode::OK, HeaderMap::default(), false)
-                        .is_err()
-                    {
+                    if stream.send_response(StatusCode::OK, hdrs(), false).is_err() {
                         return Ok(());
                     }
                     drop(streams);
@@ -333,6 +324,12 @@ where
     }
 }
 
+fn hdrs() -> HeaderMap {
+    let mut hdrs = HeaderMap::default();
+    hdrs.insert(CONTENT_TYPE, HDR_APP_GRPC);
+    hdrs
+}
+
 fn send_error(stream: &StreamRef, st: GrpcStatus, msg: HeaderValue) {
     let mut trailers = HeaderMap::default();
     trailers.insert(consts::GRPC_STATUS, st.into());