Add source field to RequestUrlCollectorStream (#92)

aicers · Jan 5, 2024 · 8b44497 · 8b44497
1 parent bdabae7
commit 8b44497
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/src/publish.rs b/src/publish.rs
@@ -561,7 +561,10 @@ mod tests {
         assert_eq!(req_data, bincode::serialize(&crusher_req).unwrap());
 
         // send/recv url collector stream request
-        let url_collector_req = RequestUrlCollectorStream { start: 0 };
+        let url_collector_req = RequestUrlCollectorStream {
+            start: 0,
+            source: Some("hello".to_string()),
+        };
         super::send_stream_request(
             &mut channel.client.send,
             super::stream::RequestStreamRecord::Http,

diff --git a/src/publish/stream.rs b/src/publish/stream.rs
@@ -130,4 +130,5 @@ pub struct RequestCrusherStream {
 #[allow(clippy::module_name_repetitions)]
 pub struct RequestUrlCollectorStream {
     pub start: i64,
+    pub source: Option<String>,
 }