github/kubeshark
1
0
Fork 0
mirror of https://github.com/kubeshark/kubeshark.git synced 2025-08-07 19:45:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

✨ Call POST /scripts/done after the posting the scripts to mark the end

Browse Source
This commit is contained in:
M. Mert Yildiran 2023-02-07 12:25:41 +03:00
parent a6dd98d241
commit 2a20dc173c
No known key found for this signature in database
GPG Key ID: DA5D6DCBB758A461
2 changed files with 22 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
cmd/tapRunner.go
View File

@ -450,6 +450,8 @@ func postHubStarted(ctx context.Context, kubernetesProvider *kubernetes.Provider
connector.PostScript(script) connector.PostScript(script)
} }
connector.PostScriptDone()
// Hub proxy URL // Hub proxy URL
url := kubernetes.GetLocalhostOnPort(config.Config.Tap.Proxy.Hub.SrcPort) url := kubernetes.GetLocalhostOnPort(config.Config.Tap.Proxy.Hub.SrcPort)
log.Info().Str("url", url).Msg(fmt.Sprintf(utils.Green, "Hub is available at:")) log.Info().Str("url", url).Msg(fmt.Sprintf(utils.Green, "Hub is available at:"))

20
internal/connect/hub.go
View File

@ -228,3 +228,23 @@ func (connector *Connector) PostScript(script *configStructs.Script) {
} }
} }
} }
func (connector *Connector) PostScriptDone() {
postScripDonetUrl := fmt.Sprintf("%s/scripts/done", connector.url)
ok := false
var err error
for !ok {
var resp *http.Response
if resp, err = utils.Post(postScripDonetUrl, "application/json", nil, connector.client); err != nil || resp.StatusCode != http.StatusOK {
if _, ok := err.(*url.Error); ok {
break
}
log.Debug().Err(err).Msg("Failed sending the POST script done to Hub:")
} else {
ok = true
log.Debug().Msg("Reported POST script done to Hub:")
}
time.Sleep(time.Second)
}
}