Use the new iter() fn for message headers
Signed-off-by: Jim Crossley <jim@crossleys.org>
This commit is contained in:
parent
f4c85cd335
commit
991cdbd01f
|
@ -20,16 +20,13 @@ pub struct ConsumerRecordDeserializer {
|
||||||
|
|
||||||
impl ConsumerRecordDeserializer {
|
impl ConsumerRecordDeserializer {
|
||||||
fn get_kafka_headers(message: &impl Message) -> Result<HashMap<String, Vec<u8>>> {
|
fn get_kafka_headers(message: &impl Message) -> Result<HashMap<String, Vec<u8>>> {
|
||||||
let mut hm = HashMap::new();
|
match message.headers() {
|
||||||
let headers = message
|
None => Err(crate::message::Error::WrongEncoding {}),
|
||||||
.headers()
|
Some(headers) => Ok(headers
|
||||||
// TODO create an error variant for invalid headers
|
.iter()
|
||||||
.ok_or(crate::message::Error::WrongEncoding {})?;
|
.map(|h| (h.key.to_string(), Vec::from(h.value.unwrap())))
|
||||||
for i in 0..headers.count() {
|
.collect()),
|
||||||
let header = headers.get(i);
|
|
||||||
hm.insert(header.key.to_string(), Vec::from(header.value.unwrap()));
|
|
||||||
}
|
}
|
||||||
Ok(hm)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn new(message: &impl Message) -> Result<ConsumerRecordDeserializer> {
|
pub fn new(message: &impl Message) -> Result<ConsumerRecordDeserializer> {
|
||||||
|
|
Loading…
Reference in New Issue