Mercurial > gemma
view auth/connection.go @ 148:0116aae1071b
Made ConnectionPool an interface and use current in-memory implementation.
author | Sascha L. Teichmann <sascha.teichmann@intevation.de> |
---|---|
date | Mon, 02 Jul 2018 11:00:19 +0200 |
parents | 0c56c56a1c44 |
children | 3349bfc2a047 |
line wrap: on
line source
package auth import ( "database/sql" "errors" "log" "sync" "time" ) var ErrNoSuchToken = errors.New("No such token") type ConnectionPool interface { Delete(token string) bool Add(token string, session *Session) *Connection Renew(token string) (string, error) Do(token string, fn func(*sql.DB) error) error Session(token string) *Session } var ConnPool ConnectionPool = NewInMemoryConnectionPool() const ( maxOpen = 16 maxDBIdle = time.Minute * 5 ) type Connection struct { session *Session access time.Time db *sql.DB refCount int mu sync.Mutex } func (c *Connection) set(session *Session) { c.session = session c.touch() } func (c *Connection) touch() { c.mu.Lock() c.access = time.Now() c.mu.Unlock() } func (c *Connection) last() time.Time { c.mu.Lock() access := c.access c.mu.Unlock() return access } func (c *Connection) close() { if c.db != nil { if err := c.db.Close(); err != nil { log.Printf("warn: %v\n", err) } c.db = nil } }