itsscb/tower-http
1
0
Fork 0
mirror of https://github.com/tower-rs/tower-http.git synced 2026-03-13 16:48:26 +00:00
Code Issues Packages Projects Releases Wiki Activity

Avoid backing up in broadcast channel in tonic example (#94)

Browse Source
This commit is contained in:
Mark Ingram 2021-05-19 16:02:40 +01:00 committed by GitHub
parent a58e5f0735
commit 526368ca31
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
examples/tonic-key-value-store/src/main.rs
View File

@ -153,7 +153,10 @@ async fn serve_forever(listener: TcpListener) -> Result<(), Box<dyn std::error::
// Build our database for holding the key/value pairs
let db = Arc::new(RwLock::new(HashMap::new()));
let (tx, _rx) = broadcast::channel(1024);
let (tx, rx) = broadcast::channel(1024);
// Drop the first receiver to avoid retaining messages in the channel
drop(rx);
// Build our tonic `Service`
let service = key_value_store_server::KeyValueStoreServer::new(ServerImpl { db, tx });
@ -216,9 +219,8 @@ impl key_value_store_server::KeyValueStore for ServerImpl {
let SetRequest { key, value } = request.into_inner();
let value = Bytes::from(value);
self.tx
.send(SubscribeReply { key: key.clone() })
.expect("failed to send");
// SendError is only possible when there are no subscribers - so can safely be ignored here
let _send = self.tx.send(SubscribeReply { key: key.clone() });
self.db.write().unwrap().insert(key, value);