Merge pull request #1 from rafaelcaricio/with-channels

Trying with channels

Cargo.lock

@@ -284,6 +284,7 @@ dependencies = [
  "tokio",
  "tokio-stream",
  "tonic",
+ "tracing",
  "tracing-subscriber",
 ]
 

Cargo.toml

@@ -6,6 +6,7 @@ edition = "2021"
 [dependencies]
 eyre = "0.6.6"
 log = "0.4"
+tracing = "0.1"
 tracing-subscriber = { version = "0.3", features = ["env-filter"] }
 tokio = { version = "1.15", features = ["macros", "rt-multi-thread", "fs"] }
 google-authz = { version = "1.0.0-alpha.2", features = ["tonic"] }
@@ -19,3 +20,6 @@ google-api-proto = { version = "1.0.0-alpha", features = ["google-cloud-translat
 tokio-stream = "0.1.8"
 futures-util = "0.3"
 async-stream = "*"
+#glib = "0.15.4"
+#gst = { package = "gstreamer", version = "0.18.3" }
+#gstreamer-base = "0.18.0"

src/main.rs

@@ -5,14 +5,16 @@ use google_api_proto::google::cloud::speech::v1::{
     StreamingRecognitionConfig, StreamingRecognizeRequest,
 };
 use google_authz::{Credentials, GoogleAuthz};
-use log::{debug, info};
+use log::{debug, info, warn};
 use tokio::io::AsyncReadExt;
+use tokio_stream::wrappers::UnboundedReceiverStream;
 use tonic::transport::Channel;
+use tracing::Instrument;
 
 #[tokio::main]
 async fn main() -> eyre::Result<()> {
     tracing_subscriber::fmt::init();
-    //console_subscriber::init();
+    // console_subscriber::init();
 
     debug!("starting...");
 
@@ -36,8 +38,10 @@ async fn main() -> eyre::Result<()> {
 
     let mut client = SpeechClient::new(channel);
 
-    let outbound = async_stream::stream! {
-        let request = StreamingRecognizeRequest {
+
+    let (sender, receiver) = tokio::sync::mpsc::unbounded_channel();
+
+    sender.send(StreamingRecognizeRequest {
         streaming_request: Some(StreamingRequest::StreamingConfig(
             StreamingRecognitionConfig {
                 config: Some(RecognitionConfig {
@@ -56,8 +60,9 @@ async fn main() -> eyre::Result<()> {
                 interim_results: false,
             },
         )),
-        };
-        yield request;
+    })?;
+
+    tokio::spawn(async move {
         let file = tokio::fs::File::open("some-audio.flac").await.unwrap();
         let mut audio_file = tokio::io::BufReader::new(file);
         // read file chunk
@@ -69,18 +74,18 @@ async fn main() -> eyre::Result<()> {
                     BytesMut::from(&buffer.as_slice()[..n]).freeze(),
                 )),
             };
-            yield request;
-            // debug!("added a buffer to the sender queue: {} bytes", n);
+            sender.send(request).unwrap();
+            //debug!("added a buffer to the sender queue: {} bytes", n);
             tokio::time::sleep(std::time::Duration::from_millis(100)).await;
         }
-    };
+    });
 
     let response = client
-        .streaming_recognize(tonic::Request::new(outbound))
+        .streaming_recognize(UnboundedReceiverStream::new(receiver))
         .await?;
     let mut inbound = response.into_inner();
 
-    while let Some(response) = inbound.message().await? {
+    while let Some(response) = inbound.message().instrument(tracing::info_span!("transcription-results")).await? {
         let mut num_results = 0;
         for res in &response.results {
             num_results = num_results + 1;