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

✨ Watch scripts inside tap command

Browse Source
This commit is contained in:
M. Mert Yildiran
2023-03-07 20:21:28 +03:00
parent ca35177b44
commit cedb7bc8bc
3 changed files with 21 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
cmd/scripts.go
View File

@@ -51,10 +51,14 @@ func runScripts() {
runProxy(false)
}
files := make(map[string]int64)
connector = connect.NewConnector(kubernetes.GetLocalhostOnPort(config.Config.Tap.Proxy.Hub.SrcPort), connect.DefaultRetries, connect.DefaultTimeout)
watchScripts(true)
}
func watchScripts(block bool) {
files := make(map[string]int64)
scripts, err := config.Config.Scripting.GetScripts()
if err != nil {
log.Error().Err(err).Send()
@@ -76,7 +80,9 @@ func runScripts() {
log.Error().Err(err).Send()
return
}
defer watcher.Close()
if block {
defer watcher.Close()
}
go func() {
for {
@@ -138,7 +144,9 @@ func runScripts() {
log.Info().Str("directory", config.Config.Scripting.Source).Msg("Watching files against changes:")
ctx, cancel := context.WithCancel(context.Background())
defer cancel()
utils.WaitForTermination(ctx, cancel)
if block {
ctx, cancel := context.WithCancel(context.Background())
defer cancel()
utils.WaitForTermination(ctx, cancel)
}
}