feat: add server-sent events client #658 (#931)

go-resty · Dec 29, 2024 · af72a4d · af72a4d
1 parent 8422694
commit af72a4d
Show file tree

Hide file tree

Showing 5 changed files with 1,058 additions and 2 deletions.
diff --git a/client.go b/client.go
@@ -58,6 +58,7 @@ var (
 	ErrUnsupportedRequestBodyKind = errors.New("resty: unsupported request body kind")
 
 	hdrUserAgentKey       = http.CanonicalHeaderKey("User-Agent")
+	hdrAcceptKey          = http.CanonicalHeaderKey("Accept")
 	hdrAcceptEncodingKey  = http.CanonicalHeaderKey("Accept-Encoding")
 	hdrContentTypeKey     = http.CanonicalHeaderKey("Content-Type")
 	hdrContentLengthKey   = http.CanonicalHeaderKey("Content-Length")

diff --git a/resty_test.go b/resty_test.go
@@ -34,7 +34,6 @@ import (
 
 var (
 	hdrLocationKey = http.CanonicalHeaderKey("Location")
-	hdrAcceptKey   = http.CanonicalHeaderKey("Accept")
 )
 
 //‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾

diff --git a/retry.go b/retry.go
@@ -115,7 +115,10 @@ func (b *backoffWithJitter) NextWaitDuration(c *Client, res *Response, err error
 		b.max = maxInt
 	}
 
-	retryStrategyFunc := c.RetryStrategy()
+	var retryStrategyFunc RetryStrategyFunc
+	if c != nil {
+		retryStrategyFunc = c.RetryStrategy()
+	}
 	if res == nil || retryStrategyFunc == nil {
 		return b.balanceMinMax(b.defaultStrategy(attempt)), nil
 	}
-Original file line number
+Diff line change
@@ Expand Up / @@ -34,7 +34,6 @@ import ( @@
     var (
     	hdrLocationKey = http.CanonicalHeaderKey("Location")
-    	hdrAcceptKey   = http.CanonicalHeaderKey("Accept")
     )
     //‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾
@@ Expand Down @@