Silverfish/proton-bridge
1
0
Fork 1
mirror of https://github.com/ProtonMail/proton-bridge.git synced 2025-12-18 16:17:03 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: use clientsLocker mutex

Browse Source
This commit is contained in:
James Houlahan
2020-04-08 13:11:05 +02:00
parent abca7284dd
commit d787d8b223
1 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
pkg/pmapi/clientmanager.go
View File

@ -108,6 +108,9 @@ func (cm *ClientManager) GetRoundTripper() (rt http.RoundTripper) {
// GetClient returns a client for the given userID. // GetClient returns a client for the given userID.
// If the client does not exist already, it is created. // If the client does not exist already, it is created.
func (cm *ClientManager) GetClient(userID string) Client { func (cm *ClientManager) GetClient(userID string) Client {
cm.clientsLocker.Lock()
defer cm.clientsLocker.Unlock()
if client, ok := cm.clients[userID]; ok { if client, ok := cm.clients[userID]; ok {
return client return client
} }
@ -119,6 +122,9 @@ func (cm *ClientManager) GetClient(userID string) Client {
// GetAnonymousClient returns an anonymous client. It replaces any anonymous client that was already created. // GetAnonymousClient returns an anonymous client. It replaces any anonymous client that was already created.
func (cm *ClientManager) GetAnonymousClient() Client { func (cm *ClientManager) GetAnonymousClient() Client {
cm.clientsLocker.Lock()
defer cm.clientsLocker.Unlock()
if client, ok := cm.clients[""]; ok { if client, ok := cm.clients[""]; ok {
client.DeleteAuth() client.DeleteAuth()
} }