Skip to content

Commit ecbc9af

Browse files
committed
Check for full format!("{}.", T::PREFIX) in gRPC keys
1 parent 2f2341c commit ecbc9af

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

src/config_extension_ext.rs

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -162,10 +162,11 @@ impl ConfigExtensionExt for SessionConfig {
162162
let mut found_some = false;
163163
for (k, v) in flight_metadata.0.iter() {
164164
let key = k.as_str().trim_start_matches(FLIGHT_METADATA_CONFIG_PREFIX);
165-
if key.starts_with(T::PREFIX) {
165+
let prefix = format!("{}.", T::PREFIX);
166+
if key.starts_with(&prefix) {
166167
found_some = true;
167168
result.set(
168-
&key.trim_start_matches(T::PREFIX).trim_start_matches("."),
169+
&key.trim_start_matches(&prefix),
169170
v.to_str().map_err(|err| {
170171
internal_datafusion_err!("Cannot parse header value: {err}")
171172
})?,

0 commit comments

Comments
 (0)