From aafdf61a66e4514c4240a47c9f9c7855e597417c Mon Sep 17 00:00:00 2001 From: Harry C Date: Wed, 20 Dec 2023 11:18:10 +0800 Subject: [PATCH] Try to support Server Sent Event. --- pkg/hub/server.go | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/pkg/hub/server.go b/pkg/hub/server.go index 5b0e268..993489e 100644 --- a/pkg/hub/server.go +++ b/pkg/hub/server.go @@ -344,7 +344,8 @@ func (hs *HubServer) handleAgentSubmit(w http.ResponseWriter, r *http.Request) { } defer submitter.Close() - upResp, err := http.ReadResponse(bufio.NewReader(r.Body), nil) + // To support Server Sent Event, we have to use a short buffer + upResp, err := http.ReadResponse(bufio.NewReaderSize(r.Body, 20), nil) if err != nil { if err := conn.Close(err); err != nil { agentLog.WithError(err).Error("Failed to close connection")