@@ -4,7 +4,7 @@ use chrono::{TimeZone, Utc};
4
4
use futures:: { StreamExt , TryStreamExt } ;
5
5
use rdkafka:: {
6
6
consumer:: { Consumer , StreamConsumer } ,
7
- message:: { Headers , OwnedHeaders } ,
7
+ message:: { Header , Headers , OwnedHeaders } ,
8
8
producer:: { FutureProducer , FutureRecord } ,
9
9
util:: Timeout ,
10
10
ClientConfig , Message , TopicPartitionList ,
@@ -49,7 +49,10 @@ pub async fn produce(
49
49
for ( topic_name, partition_index, record) in records {
50
50
let mut headers = OwnedHeaders :: new ( ) ;
51
51
for ( k, v) in record. headers {
52
- headers = headers. add ( & k, & v) ;
52
+ headers = headers. insert ( Header {
53
+ key : & k,
54
+ value : Some ( & v) ,
55
+ } ) ;
53
56
}
54
57
55
58
let mut f_record = FutureRecord :: to ( & topic_name)
@@ -104,8 +107,8 @@ pub async fn consume(
104
107
. map ( |headers| {
105
108
( 0 ..headers. count ( ) )
106
109
. map ( |i| {
107
- let ( k , v ) = headers. get ( i) . unwrap ( ) ;
108
- ( k . to_owned ( ) , v . to_vec ( ) )
110
+ let header = headers. get ( i) ;
111
+ ( header . key . to_owned ( ) , header . value . unwrap ( ) . to_vec ( ) )
109
112
} )
110
113
. collect ( )
111
114
} )
0 commit comments