瀏覽代碼

guess aws cluster id based on cluster name

Alexander Belanger 4 年之前
父節點
當前提交
d0a6af596d
共有 2 個文件被更改,包括 9 次插入2 次删除
  1. 1 1
      internal/kubernetes/config.go
  2. 8 1
      internal/models/integrations/aws.go

+ 1 - 1
internal/kubernetes/config.go

@@ -329,7 +329,7 @@ func (conf *OutOfClusterConfig) CreateRawConfigFromCluster() (*api.Config, error
 			return nil, err
 			return nil, err
 		}
 		}
 
 
-		tok, err := awsAuth.GetBearerToken(conf.getTokenCache, conf.setTokenCache)
+		tok, err := awsAuth.GetBearerToken(conf.getTokenCache, conf.setTokenCache, cluster.Name)
 
 
 		if err != nil {
 		if err != nil {
 			return nil, err
 			return nil, err

+ 8 - 1
internal/models/integrations/aws.go

@@ -104,6 +104,7 @@ func (a *AWSIntegration) PopulateAWSArn() error {
 func (a *AWSIntegration) GetBearerToken(
 func (a *AWSIntegration) GetBearerToken(
 	getTokenCache GetTokenCacheFunc,
 	getTokenCache GetTokenCacheFunc,
 	setTokenCache SetTokenCacheFunc,
 	setTokenCache SetTokenCacheFunc,
+	clusterID string,
 ) (string, error) {
 ) (string, error) {
 	cache, err := getTokenCache()
 	cache, err := getTokenCache()
 
 
@@ -126,9 +127,15 @@ func (a *AWSIntegration) GetBearerToken(
 		return "", err
 		return "", err
 	}
 	}
 
 
+	clusterIDGuess := string(a.AWSClusterID)
+
+	if clusterIDGuess == "" {
+		clusterIDGuess = clusterID
+	}
+
 	tok, err := generator.GetWithOptions(&token.GetTokenOptions{
 	tok, err := generator.GetWithOptions(&token.GetTokenOptions{
 		Session:   sess,
 		Session:   sess,
-		ClusterID: string(a.AWSClusterID),
+		ClusterID: clusterID,
 	})
 	})
 
 
 	if err != nil {
 	if err != nil {