Add timeout option to install

k8ssandra · Mar 22, 2024 · f80fae9 · f80fae9
1 parent 37e0b3f
commit f80fae9
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/pkg/helmutil/fetch.go b/pkg/helmutil/fetch.go
@@ -5,6 +5,7 @@ import (
  "os"
  "path/filepath"
  "strings"
+ "time"
 
  "github.com/k8ssandra/k8ssandra-client/pkg/util"
  "helm.sh/helm/v3/pkg/action"
@@ -129,13 +130,16 @@ func ListInstallations(cfg *action.Configuration) ([]*release.Release, error) {
  return listAction.Run()
 }
 
-func Install(cfg *action.Configuration, releaseName, path, namespace string, values map[string]interface{}, devel bool, skipCRDs bool) (*release.Release, error) {
+func Install(cfg *action.Configuration, releaseName, path, namespace string, values map[string]interface{}, devel bool, skipCRDs bool, timeout time.Duration) (*release.Release, error) {
  installAction := action.NewInstall(cfg)
  installAction.ReleaseName = releaseName
  installAction.Namespace = namespace
  installAction.CreateNamespace = true
  installAction.Atomic = true
  installAction.Wait = true
+ if timeout > 0 {
+ installAction.Timeout = timeout
+ }
  if skipCRDs {
  installAction.SkipCRDs = true
  }