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

⚡ Change consts config field to env

Browse Source
This commit is contained in:
M. Mert Yildiran 2023-03-03 17:32:19 +03:00
parent 72a18871b1
commit 9a95fa364c
No known key found for this signature in database
GPG Key ID: DA5D6DCBB758A461
3 changed files with 9 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
cmd/tapRunner.go
View File

@ -448,7 +448,7 @@ func postHubStarted(ctx context.Context, kubernetesProvider *kubernetes.Provider
}
// Scripting
connector.PostConsts(config.Config.Scripting.Consts)
connector.PostEnv(config.Config.Scripting.Env)
scripts, err := config.Config.Scripting.GetScripts()
if err != nil {

2
config/configStructs/scriptingConfig.go
View File

@ -10,7 +10,7 @@ import (
)
type ScriptingConfig struct {
Consts map[string]interface{} `yaml:"consts"`
Env map[string]interface{} `yaml:"env"`
Source string `yaml:"source" default:""`
}

14
internal/connect/hub.go
View File

@ -181,27 +181,27 @@ func (connector *Connector) PostLicense(license string) {
}
}
func (connector *Connector) PostConsts(consts map[string]interface{}) {
if len(consts) == 0 {
func (connector *Connector) PostEnv(env map[string]interface{}) {
if len(env) == 0 {
return
}
postConstsUrl := fmt.Sprintf("%s/scripts/consts", connector.url)
postEnvUrl := fmt.Sprintf("%s/scripts/env", connector.url)
if constsMarshalled, err := json.Marshal(consts); err != nil {
log.Error().Err(err).Msg("Failed to marshal the consts:")
if envMarshalled, err := json.Marshal(env); err != nil {
log.Error().Err(err).Msg("Failed to marshal the env:")
} else {
ok := false
for !ok {
var resp *http.Response
if resp, err = utils.Post(postConstsUrl, "application/json", bytes.NewBuffer(constsMarshalled), connector.client); err != nil || resp.StatusCode != http.StatusOK {
if resp, err = utils.Post(postEnvUrl, "application/json", bytes.NewBuffer(envMarshalled), connector.client); err != nil || resp.StatusCode != http.StatusOK {
if _, ok := err.(*url.Error); ok {
break
}
log.Warn().Err(err).Msg("Failed sending the constants to Hub:")
} else {
ok = true
log.Info().Interface("consts", consts).Msg("Reported constants to Hub:")
log.Info().Interface("env", env).Msg("Reported constants to Hub:")
}
time.Sleep(DefaultSleep)
}