YakumoLabs/gatus
4
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(websocket): add support for custom headers in WS check (#1180)

Browse Source 
feat(websocket): add support for custom headers in QueryWebSocket function
This commit is contained in:
Olexandr Dovgun
2025-08-02 21:06:46 +03:00
committed by GitHub
parent 8c5ad54e71
commit d27c63ded7
4 changed files with 22 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
config/endpoint/endpoint.go
View File

@@ -399,7 +399,16 @@ func (e *Endpoint) call(result *Result) {
} else if endpointType == TypeICMP {
result.Connected, result.Duration = client.Ping(strings.TrimPrefix(e.URL, "icmp://"), e.ClientConfig)
} else if endpointType == TypeWS {
result.Connected, result.Body, err = client.QueryWebSocket(e.URL, e.getParsedBody(), e.ClientConfig)
wsHeaders := map[string]string{}
if e.Headers != nil {
for k, v := range e.Headers {
wsHeaders[k] = v
}
}
if _, exists := wsHeaders["User-Agent"]; !exists {
wsHeaders["User-Agent"] = GatusUserAgent
}
result.Connected, result.Body, err = client.QueryWebSocket(e.URL, e.getParsedBody(), wsHeaders, e.ClientConfig)
if err != nil {
result.AddError(err.Error())
return