rancher/steve
1
0
Fork 0
mirror of https://github.com/rancher/steve.git synced 2025-09-13 05:49:16 +00:00
Code Issues Projects Releases Wiki Activity

Fix restart of steve aggregation

Browse Source
This commit is contained in:
Darren Shepherd
2021-08-16 22:05:50 -07:00
parent bcbcef36b3
commit e70fca2a42
2 changed files with 14 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
pkg/aggregation/server.go
View File

@@ -60,12 +60,24 @@ func ListenAndServe(ctx context.Context, url string, caCert []byte, token string
func serve(ctx context.Context, dialer websocket.Dialer, url string, headers http.Header, handler http.Handler) error {
url = strings.Replace(url, "http://", "ws://", 1)
url = strings.Replace(url, "https://", "wss://", 1)
conn, _, err := dialer.DialContext(ctx, url, headers)
// ensure we clean up everything on exit
ctx, cancel := context.WithCancel(ctx)
defer cancel()
dialCtx, dialCancel := context.WithTimeout(ctx, 5*time.Second)
defer dialCancel()
conn, _, err := dialer.DialContext(dialCtx, url, headers)
if err != nil {
return err
}
defer conn.Close()
go func() {
<-ctx.Done()
conn.Close()
}()
listener := NewListener("steve")
server := http.Server{
Handler: handler,