ensureConnectionCache should only create a cache if one does not exist.

#699
turbot · Nov 2, 2023 · bcdd15b · bcdd15b
1 parent 8adf4d8
commit bcdd15b
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/plugin/plugin.go b/plugin/plugin.go
@@ -236,15 +236,21 @@ func (p *Plugin) shutdown() {
 }
 
 func (p *Plugin) ensureConnectionCache(connectionName string) (*connectionmanager.ConnectionCache, error) {
+	p.connectionCacheMapLock.Lock()
+	defer p.connectionCacheMapLock.Unlock()
+
+	// check if we already have a cache
+	if cache, ok := p.connectionCacheMap[connectionName]; ok {
+		return cache, nil
+	}
+
+	// add to map of connection caches
 	maxCost := int64(100000 / len(p.ConnectionMap))
 	connectionCache, err := connectionmanager.NewConnectionCache(connectionName, maxCost)
 	if err != nil {
 		return nil, err
 	}
 
-	p.connectionCacheMapLock.Lock()
-	defer p.connectionCacheMapLock.Unlock()
-	// add to map of connection caches
 	p.connectionCacheMap[connectionName] = connectionCache
 	return connectionCache, nil
 }