Explorar o código

Merge branch 'beta.3.github-actions-ci' of https://github.com/porter-dev/porter into sean-testing

Sean Rhee %!s(int64=5) %!d(string=hai) anos
pai
achega
104baf36eb
Modificáronse 1 ficheiros con 15 adicións e 1 borrados
  1. 15 1
      internal/repository/gorm/cluster.go

+ 15 - 1
internal/repository/gorm/cluster.go

@@ -224,6 +224,16 @@ func (repo *ClusterRepository) UpdateCluster(
 func (repo *ClusterRepository) UpdateClusterTokenCache(
 	tokenCache *ints.ClusterTokenCache,
 ) (*models.Cluster, error) {
+	if tok := tokenCache.Token; len(tok) > 0 {
+		cipherData, err := repository.Encrypt(tok, repo.key)
+
+		if err != nil {
+			return nil, err
+		}
+
+		tokenCache.Token = cipherData
+	}
+
 	cluster := &models.Cluster{}
 
 	if err := repo.db.Where("id = ?", tokenCache.ClusterID).First(&cluster).Error; err != nil {
@@ -233,7 +243,11 @@ func (repo *ClusterRepository) UpdateClusterTokenCache(
 	cluster.TokenCache.Token = tokenCache.Token
 	cluster.TokenCache.Expiry = tokenCache.Expiry
 
-	return repo.UpdateCluster(cluster)
+	if err := repo.db.Save(cluster).Error; err != nil {
+		return nil, err
+	}
+
+	return cluster, nil
 }
 
 // DeleteCluster removes a cluster from the db