feat: add response header argument in EventOpenFunc #1060

Author: yoogoc

sse.go

@@ -41,7 +41,7 @@ type (
 	// EventOpenFunc is a callback function type used to receive notification
 	// when Resty establishes a connection with the server for the
 	// Server-Sent Events(SSE)
-	EventOpenFunc func(url string)
+	EventOpenFunc func(url string, respHdr http.Header)
 
 	// EventMessageFunc is a callback function type used to receive event details
 	// from the Server-Sent Events(SSE) stream
@@ -453,7 +453,7 @@ func (es *EventSource) Get() error {
 		if err != nil {
 			return err
 		}
-		es.triggerOnOpen()
+		es.triggerOnOpen(res.Header.Clone())
 		if err := es.listenStream(res); err != nil {
 			return err
 		}
@@ -479,11 +479,11 @@ func (es *EventSource) isClosed() bool {
 	return es.closed
 }
 
-func (es *EventSource) triggerOnOpen() {
+func (es *EventSource) triggerOnOpen(hdr http.Header) {
 	es.lock.RLock()
 	defer es.lock.RUnlock()
 	if es.onOpen != nil {
-		es.onOpen(strings.Clone(es.url))
+		es.onOpen(strings.Clone(es.url), hdr)
 	}
 }
 

sse_test.go

@@ -167,8 +167,8 @@ func TestEventSourceOverwriteFuncs(t *testing.T) {
 
 	es.SetURL(ts.URL).
 		OnMessage(messageFunc2, nil).
-		OnOpen(func(url string) {
-			t.Log("from overwrite func", url)
+		OnOpen(func(url string, respHdr http.Header) {
+			t.Log("from overwrite func", url, respHdr)
 		}).
 		OnError(func(err error) {
 			t.Log("from overwrite func", err)
@@ -422,8 +422,8 @@ func createEventSource(t *testing.T, url string, fn EventMessageFunc, rt any) *E
 		SetRetryMaxWaitTime(1000 * time.Millisecond).
 		SetMaxBufSize(1 << 14). // 16kb
 		SetLogger(createLogger()).
-		OnOpen(func(url string) {
-			t.Log("I'm connected:", url)
+		OnOpen(func(url string, respHdr http.Header) {
+			t.Log("I'm connected:", url, respHdr)
 		}).
 		OnError(func(err error) {
 			t.Log("Error occurred:", err)