/
http_websocket.go
93 lines (83 loc) · 2.56 KB
/
http_websocket.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
package cdsclient
import (
"context"
"encoding/json"
"fmt"
"net/url"
"runtime/pprof"
"strings"
"time"
"github.com/gorilla/websocket"
"github.com/ovh/cds/sdk"
"github.com/ovh/cds/sdk/log"
)
func (c *client) RequestWebsocket(ctx context.Context, path string, msgToSend <-chan []sdk.WebsocketFilter, msgReceived chan<- sdk.WebsocketEvent) error {
wsContext, wsContextCancel := context.WithCancel(ctx)
defer wsContextCancel()
// Checks that current session_token is still valid
// If not, challenge a new one against the authenticationToken
if !c.config.HasValidSessionToken() && c.config.BuitinConsumerAuthenticationToken != "" {
resp, err := c.AuthConsumerSignin(sdk.ConsumerBuiltin, sdk.AuthConsumerSigninRequest{"token": c.config.BuitinConsumerAuthenticationToken})
if err != nil {
return err
}
c.config.SessionToken = resp.Token
}
labels := pprof.Labels("path", path, "method", "GET")
wsContext = pprof.WithLabels(wsContext, labels)
pprof.SetGoroutineLabels(wsContext)
uHost, err := url.Parse(c.config.Host)
if err != nil {
return sdk.WrapError(err, "wrong Host configuration")
}
urlWebsocket := url.URL{
Scheme: strings.Replace(uHost.Scheme, "http", "ws", -1),
Host: uHost.Host,
Path: uHost.Path + "/ws",
}
headers := make(map[string][]string)
date := sdk.FormatDateRFC5322(time.Now())
headers["Date"] = []string{date}
headers["X-CDS-RemoteTime"] = []string{date}
auth := "Bearer " + c.config.SessionToken
headers["Authorization"] = []string{auth}
con, _, err := c.httpWebsocketClient.Dial(urlWebsocket.String(), headers)
if err != nil {
return sdk.WithStack(err)
}
defer con.Close() // nolint
// Message to send
sdk.GoRoutine(wsContext, fmt.Sprintf("RequestWebsocket-%s-%s", c.config.User, sdk.UUID()), func(ctx context.Context) {
for {
select {
case <-ctx.Done():
log.Warning(wsContext, "Leaving....")
return
case m := <-msgToSend:
if err := con.WriteJSON(m); err != nil {
log.Error(wsContext, "ws: unable to send message: %v", err)
}
}
}
})
for {
if ctx.Err() != nil {
return ctx.Err()
}
_, message, err := con.ReadMessage()
if err != nil {
if websocket.IsCloseError(err, websocket.CloseGoingAway, websocket.CloseAbnormalClosure) {
log.Warning(ctx, "websocket error: %v", err)
return err
}
log.Error(ctx, "ws: unable to read message: %v", err)
continue
}
var wsEvent sdk.WebsocketEvent
if err := json.Unmarshal(message, &wsEvent); err != nil {
log.Error(ctx, "ws: unable to unmarshal message: %s : %v", string(message), err)
continue
}
msgReceived <- wsEvent
}
}