handle connection reset by peer

rtbrick · Feb 21, 2024 · c2f6f76 · c2f6f76
1 parent 057ac7d
commit c2f6f76
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/pkg/controller/repository.go b/pkg/controller/repository.go
@@ -2,6 +2,7 @@ package controller
 
 import (
 	"encoding/json"
+	"errors"
 	"fmt"
 	"io"
 	"io/ioutil"
@@ -288,10 +289,13 @@ func (r *DefaultRepository) Command(name string, command SocketCommand) ([]byte,
 		count, err := c.Read(buf)
 		received = append(received, buf[:count]...)
 		if err != nil {
-			if err != io.EOF {
-				return nil, err
+			if errors.Is(err, syscall.ECONNRESET) {
+				continue
+			}
+			if err == io.EOF {
+				break
 			}
-			break
+			return nil, err
 		}
 	}