fix(sdk): http sse client (#3447)

ovh · Oct 11, 2018 · 9b93888 · 9b93888
1 parent 45c3c5f
commit 9b93888
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 7 deletions.
diff --git a/sdk/cdsclient/client_queue.go b/sdk/cdsclient/client_queue.go
@@ -60,9 +60,9 @@ func (c *client) QueuePolling(ctx context.Context, jobs chan<- sdk.WorkflowNodeJ
 	sdk.GoRoutine(ctx, "RequestSSEGet", func(ctx context.Context) {
 		for ctx.Err() == nil {
 			if err := c.RequestSSEGet(ctx, "/events", chanSSEvt); err != nil {
-				log.Println(err)
+				log.Println("QueuePolling", err)
 			}
-			time.Sleep(2 * time.Second)
+			time.Sleep(1 * time.Second)
 		}
 	})
 

diff --git a/sdk/cdsclient/http_sse.go b/sdk/cdsclient/http_sse.go
@@ -87,12 +87,14 @@ func (c *client) RequestSSEGet(ctx context.Context, path string, evCh chan<- SSE
 		}
 
 		bs, err := br.ReadBytes('\n')
-		if err != nil && err != io.EOF {
-			return err
-		}
 
-		if err == io.EOF {
-			EOF = true
+		if err != nil {
+			switch err {
+			case io.EOF, io.ErrUnexpectedEOF, io.ErrClosedPipe:
+				EOF = true
+			default:
+				return err
+			}
 		}
 
 		if len(bs) < 2 {