assert streams are not overwritten

stepancheg · Dec 20, 2016 · 7e2ec88 · 7e2ec88
1 parent 45e3dc9
commit 7e2ec88
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/src/http_client.rs b/src/http_client.rs
@@ -122,7 +122,9 @@ struct ClientInner {
 impl ClientInner {
     fn insert_stream(&mut self, stream: GrpcHttpClientStream) -> StreamId {
         let id = self.session_state.next_stream_id;
-        self.common.streams.insert(id, stream);
+        if let Some(..) = self.common.streams.insert(id, stream) {
+            panic!("inserted stream that already existed");
+        }
         self.session_state.next_stream_id += 2;
         id
     }

diff --git a/src/http_server.rs b/src/http_server.rs
@@ -179,7 +179,9 @@ impl<F : HttpService> ServerInner<F> {
             request_handler: Some(req_tx),
             _marker: marker::PhantomData,
         };
-        self.common.streams.insert(stream_id, stream);
+        if let Some(..) = self.common.streams.insert(stream_id, stream) {
+            panic!("inserted stream that already existed");
+        }
         self.common.streams.get_mut(&stream_id).unwrap()
     }