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"] }
@@ -18,4 +19,7 @@ prost-types = { version = "0.9" }
 google-api-proto = { version = "1.0.0-alpha", features = ["google-cloud-translation-v3", "google-cloud-speech-v1"] }
 tokio-stream = "0.1.8"
 futures-util = "0.3"
-async-stream = "*"
+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,28 +38,31 @@ async fn main() -> eyre::Result<()> {
 
     let mut client = SpeechClient::new(channel);
 
-    let outbound = async_stream::stream! {
-        let request = StreamingRecognizeRequest {
-            streaming_request: Some(StreamingRequest::StreamingConfig(
-                StreamingRecognitionConfig {
-                    config: Some(RecognitionConfig {
-                        encoding: AudioEncoding::Flac.into(), // matching current example file
-                        sample_rate_hertz: 44_100,            // matching current example file
-                        audio_channel_count: 2,
-                        language_code: "en-US".to_string(),   // we only support en-US to start with
-                        model: "video".to_string(),           // dictate does not set this option
-                        use_enhanced: true,                   // dictate does not set this option
-                        profanity_filter: true,               // used by Dictate, so we also use it here
-                        enable_word_time_offsets: true, // important so we can get the spoken word time ranges
-                        max_alternatives: 1,            // make sure the default is used
-                        ..Default::default()
-                    }),
-                    single_utterance: false,
-                    interim_results: false,
-                },
-            )),
-        };
-        yield request;
+
+    let (sender, receiver) = tokio::sync::mpsc::unbounded_channel();
+
+    sender.send(StreamingRecognizeRequest {
+        streaming_request: Some(StreamingRequest::StreamingConfig(
+            StreamingRecognitionConfig {
+                config: Some(RecognitionConfig {
+                    encoding: AudioEncoding::Flac.into(), // matching current example file
+                    sample_rate_hertz: 44_100,            // matching current example file
+                    audio_channel_count: 2,
+                    language_code: "en-US".to_string(),   // we only support en-US to start with
+                    model: "video".to_string(),           // dictate does not set this option
+                    use_enhanced: true,                   // dictate does not set this option
+                    profanity_filter: true,               // used by Dictate, so we also use it here
+                    enable_word_time_offsets: true, // important so we can get the spoken word time ranges
+                    max_alternatives: 1,            // make sure the default is used
+                    ..Default::default()
+                }),
+                single_utterance: false,
+                interim_results: false,
+            },
+        )),
+    })?;
+
+    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;