Deuxfleurs/alps
24
1
Fork 0
Code Issues Pull requests 1 Projects Releases Wiki Activity

Add synchronization to connection pool

Browse source
This commit is contained in:
Simon Ser 2019-12-02 17:40:53 +01:00
parent 85f8530fd3
commit 594bb2ec19
No known key found for this signature in database
GPG key ID: 0FDE7BE0E88F5E48
1 changed files with 11 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
conn_pool.go
View file

@ -4,6 +4,7 @@ import (
"crypto/rand"
"encoding/base64"
"errors"
"sync"
imapclient "github.com/emersion/go-imap/client"
)
@ -21,7 +22,7 @@ var ErrSessionExpired = errors.New("session expired")
// TODO: expiration timer
type ConnPool struct {
// TODO: add synchronization
locker sync.Mutex
conns map[string]*imapclient.Client
}
@ -32,6 +33,9 @@ func NewConnPool() *ConnPool {
}
func (pool *ConnPool) Get(token string) (*imapclient.Client, error) {
pool.locker.Lock()
defer pool.locker.Unlock()
conn, ok := pool.conns[token]
if !ok {
return nil, ErrSessionExpired
@ -40,6 +44,9 @@ func (pool *ConnPool) Get(token string) (*imapclient.Client, error) {
}
func (pool *ConnPool) Put(conn *imapclient.Client) (token string, err error) {
pool.locker.Lock()
defer pool.locker.Unlock()
for {
var err error
token, err = generateToken()
@ -57,7 +64,10 @@ func (pool *ConnPool) Put(conn *imapclient.Client) (token string, err error) {
go func() {
<-conn.LoggedOut()
pool.locker.Lock()
delete(pool.conns, token)
pool.locker.Unlock()
}()
return token, nil