liviosoares / go-watson-sdk Goto Github PK

diff --git a/watson/speech_to_text/speech_to_text.go b/watson/speech_to_text/speech_to_text.go
index 55847b7..a801ba3 100644
--- a/watson/speech_to_text/speech_to_text.go
+++ b/watson/speech_to_text/speech_to_text.go
@@ -19,8 +19,11 @@ package speech_to_text
 import (
    "encoding/json"
    "errors"
+   "fmt"
    "io"
    "net/url"
+   "sync"
+   "time"

    "github.com/liviosoares/go-watson-sdk/watson"
    "github.com/liviosoares/go-watson-sdk/watson/authorization"
@@ -224,10 +227,38 @@ type stream struct {
    stopped        bool
 }

+type counter struct {
+   sync.RWMutex
+   count uint64
+}
+
+func (c *counter) increment() {
+   c.Lock()
+   c.count += 1
+   c.Unlock()
+}
+
+func (c *counter) value() uint64 {
+   c.RLock()
+   defer c.RUnlock()
+
+   return c.count
+}
+
+func (c *counter) String() string {
+   c.RLock()
+   defer c.RUnlock()
+
+   return fmt.Sprintf("[c:%d]", c.count)
+}
+
+var chunkCounter = new(counter)
+
 func (s *stream) Write(p []byte) (n int, err error) {
    if s.stopped {
        return 0, errors.New("cannot write to stopped stream")
    }
+   fmt.Printf("writing chunk of size: %d\n", len(p))
    if !s.started {
        m := make(map[string]interface{})
        for k, v := range s.options {
@@ -242,6 +273,8 @@ func (s *stream) Write(p []byte) (n int, err error) {
            }
        }
        m["interim_results"] = true
+       sendTime := time.Now()
+       fmt.Printf("chunkCount: %s sendTime: %d\n", chunkCounter, sendTime.Unix())
        err := websocket.JSON.Send(s.ws, m)
        if err != nil {
            return 0, err
@@ -249,6 +282,8 @@ func (s *stream) Write(p []byte) (n int, err error) {
        s.started = true
        var state StateReply
        err = websocket.JSON.Receive(s.ws, &state)
+       fmt.Printf("time to receive chunk: %s %s\n", chunkCounter, time.Since(sendTime))
+       chunkCounter.increment()
        if err != nil {
            return 0, err
        }