vaultAuth by switching

Lucretius · Sep 21, 2021 · 74d7799 · 74d7799
1 parent a50e05a
commit 74d7799
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/main.go b/main.go
@@ -46,10 +46,12 @@ func main() {
 
 	for {
 		if snapshotter.TokenExpiration.Before(time.Now()) {
-			if c.VaultAuthMethod == "k8s" {
+			switch c.VaultAuthMethod {
+			case "k8s":
 				snapshotter.SetClientTokenFromK8sAuth(c)
+			default:
+				snapshotter.SetClientTokenFromAppRole(c)
 			}
-			snapshotter.SetClientTokenFromAppRole(c)
 		}
 		leader, err := snapshotter.API.Sys().Leader()
 		if err != nil {