[speech-to-text] Send close message to the WebSocket once we get the

final results

Author: germanattanasio

src/main/java/com/ibm/watson/developer_cloud/speech_to_text/v1/websocket/WebSocketSpeechToTextClient.java

@@ -42,7 +42,7 @@ public class WebSocketSpeechToTextClient {
    */
   public class WebSocketListener extends WebSocketAdapter {
     private RecognizeDelegate delegate;
-
+    private boolean audioSent = false;
     /**
      * Instantiates a new WebSocket listener.
      * 
@@ -51,6 +51,7 @@ public class WebSocketListener extends WebSocketAdapter {
     public WebSocketListener(RecognizeDelegate delegate) {
       super();
       this.delegate = delegate;
+      audioSent = false;
     }
 
     /*
@@ -69,6 +70,8 @@ public void onTextMessage(WebSocket websocket, String message) {
         } else if (json.has(RESULTS)) {
           SpeechResults transcript = GsonSingleton.getGson().fromJson(message, SpeechResults.class);
           delegate.onMessage(transcript);
+        } else if (audioSent) {
+          websocket.sendClose();
         }
       } catch (JsonParseException e) {
         new RuntimeException("Error parsing the incoming message: " + message);
@@ -170,7 +173,10 @@ public void recognize(InputStream stream, RecognizeOptions options, RecognizeDel
 
       // 4. Send the input stream as binary data
       sendInputStream(ws, stream);
-
+      
+      // 5. Tell the listener that we sent the audio
+      listener.audioSent = true;
+      
       // 5. Send stop message
       ws.sendText(buildStopMessage());
 
@@ -217,9 +223,8 @@ private void sendInputStream(WebSocket ws, InputStream stream) throws IOExceptio
       else
         ws.sendBinary(Arrays.copyOfRange(buffer, 0, read));
 
-      Thread.sleep(10);
+      Thread.sleep(20);
     }
-
     stream.close();
   }
 

src/test/java/com/ibm/watson/developer_cloud/speech_to_text/v1/SpeechToTextIT.java

@@ -204,8 +204,7 @@ public void onError(Exception e) {
       public void onMessage(SpeechResults speechResults) {
         if (speechResults != null && speechResults.isFinal()) {
           asyncResults = speechResults;
-          // System.out.println(speechResults);
-          lock.countDown();
+          //System.out.println(speechResults);
         }
       }