Use key.ClusterName when storing a key in memory

gravitational · Mar 12, 2021 · 8e4efc1 · 8e4efc1
1 parent 948f574
commit 8e4efc1
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/lib/client/keystore.go b/lib/client/keystore.go
@@ -769,6 +769,9 @@ func NewMemLocalKeyStore(dirPath string) (*MemLocalKeyStore, error) {
 // AddKey writes a key to the underlying key store.
 func (s *MemLocalKeyStore) AddKey(proxy string, username string, key *Key) error {
 	s.inMem[keyIndex{proxyHost: proxy, username: username}] = key
+	if key.ClusterName != "" {
+		s.inMem[keyIndex{proxyHost: proxy, username: username, clusterName: key.ClusterName}] = key
+	}
 	return nil
 }
 
@@ -785,7 +788,6 @@ func (s *MemLocalKeyStore) GetKey(proxy, username string, opts ...KeyOption) (*K
 			return nil, trace.Wrap(err)
 		}
 	}
-
 	return entry, nil
 }