auth
2014-09-01
Child:5bf0a5fd1d01
auth/memstore.go
Rough out tokens and begin the memstore. Rough out the Token type for working with OAuth2 access and refresh tokens. Rough out the TokenStore interface that dictates how Tokens will be stored and retrieved. Write tests for the successful (in the working-as-intended sense) calls to TokenStore. Begin a Memstore type that stores data in memory. Implement the TokenStore interface for Memstore.
| paddy@28 | 1 package auth |
| paddy@28 | 2 |
| paddy@28 | 3 import "sync" |
| paddy@28 | 4 |
| paddy@28 | 5 type Memstore struct { |
| paddy@28 | 6 tokens map[string]Token |
| paddy@28 | 7 refreshTokenLookup map[string]string |
| paddy@28 | 8 profileTokenLookup map[string][]string |
| paddy@28 | 9 tokenLock sync.RWMutex |
| paddy@28 | 10 } |
| paddy@28 | 11 |
| paddy@28 | 12 func NewMemstore() *Memstore { |
| paddy@28 | 13 return &Memstore{ |
| paddy@28 | 14 tokens: map[string]Token{}, |
| paddy@28 | 15 refreshTokenLookup: map[string]string{}, |
| paddy@28 | 16 profileTokenLookup: map[string][]string{}, |
| paddy@28 | 17 } |
| paddy@28 | 18 } |
| paddy@28 | 19 |
| paddy@28 | 20 func (m *Memstore) lookupTokenByRefresh(token string) (string, error) { |
| paddy@28 | 21 m.tokenLock.RLock() |
| paddy@28 | 22 defer m.tokenLock.RUnlock() |
| paddy@28 | 23 t, ok := m.refreshTokenLookup[token] |
| paddy@28 | 24 if !ok { |
| paddy@28 | 25 return "", ErrTokenNotFound |
| paddy@28 | 26 } |
| paddy@28 | 27 return t, nil |
| paddy@28 | 28 } |
| paddy@28 | 29 |
| paddy@28 | 30 func (m *Memstore) lookupTokensByProfileID(id string) ([]string, error) { |
| paddy@28 | 31 m.tokenLock.RLock() |
| paddy@28 | 32 defer m.tokenLock.RUnlock() |
| paddy@28 | 33 return m.profileTokenLookup[id], nil |
| paddy@28 | 34 } |