github/kubeshark
1
0
Fork 0
mirror of https://github.com/kubeshark/kubeshark.git synced 2025-06-26 16:24:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

Improved the console command made it resilient to Websocket breaks and redeployment.

Browse Source
This commit is contained in:
Alon Girmonsky 2024-10-10 22:15:00 -07:00
parent b7b0e3dcee
commit ecc577ccc8
1 changed files with 56 additions and 53 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
cmd/console.go
View File

@ -42,6 +42,7 @@ func init() {
}
func runConsole() {
for {
hubUrl := kubernetes.GetHubUrl()
response, err := http.Get(fmt.Sprintf("%s/echo", hubUrl))
if err != nil || response.StatusCode != 200 {
@ -64,8 +65,9 @@ func runConsole() {
c, _, err := websocket.DefaultDialer.Dial(u.String(), headers)
if err != nil {
log.Error().Err(err).Send()
return
log.Error().Err(err).Msg("Websocket dial error, retrying in 5 seconds...")
time.Sleep(5 * time.Second) // Delay before retrying
continue
}
defer c.Close()
@ -76,8 +78,8 @@ func runConsole() {
for {
_, message, err := c.ReadMessage()
if err != nil {
log.Error().Err(err).Send()
return
log.Error().Err(err).Msg("Error reading websocket message, reconnecting...")
break // Break to reconnect
}
msg := string(message)
@ -93,17 +95,18 @@ func runConsole() {
ticker := time.NewTicker(time.Second)
defer ticker.Stop()
for {
select {
case <-done:
return
log.Warn().Msg(fmt.Sprintf(utils.Yellow, "Connection closed, reconnecting..."))
time.Sleep(5 * time.Second) // Delay before reconnecting
continue // Reconnect after error
case <-interrupt:
log.Warn().Msg(fmt.Sprintf(utils.Yellow, "Received interrupt, exiting..."))
err := c.WriteMessage(websocket.CloseMessage, websocket.FormatCloseMessage(websocket.CloseNormalClosure, ""))
if err != nil {
log.Error().Err(err).Send()
return
continue
}
select {