diff options
author | Chris <ccbrown112@gmail.com> | 2017-10-06 08:12:10 -0700 |
---|---|---|
committer | Joram Wilander <jwawilander@gmail.com> | 2017-10-06 11:12:10 -0400 |
commit | 363568b4eb3209adb1b88ceb0d8e455e6d4a1073 (patch) | |
tree | 419fb74d8e1a2e91f9f77aa5b873c005bcfbff48 /store/sqlstore/tokens_store.go | |
parent | 12501673d0c70120eebeac633e5072b2e7a2174d (diff) | |
download | chat-363568b4eb3209adb1b88ceb0d8e455e6d4a1073.tar.gz chat-363568b4eb3209adb1b88ceb0d8e455e6d4a1073.tar.bz2 chat-363568b4eb3209adb1b88ceb0d8e455e6d4a1073.zip |
reduce store boiler plate (#7585)
Diffstat (limited to 'store/sqlstore/tokens_store.go')
-rw-r--r-- | store/sqlstore/tokens_store.go | 44 |
1 files changed, 6 insertions, 38 deletions
diff --git a/store/sqlstore/tokens_store.go b/store/sqlstore/tokens_store.go index 31e39f1fe..ccb58cef1 100644 --- a/store/sqlstore/tokens_store.go +++ b/store/sqlstore/tokens_store.go @@ -34,54 +34,27 @@ func (s SqlTokenStore) CreateIndexesIfNotExists() { } func (s SqlTokenStore) Save(token *model.Token) store.StoreChannel { - - storeChannel := make(store.StoreChannel, 1) - - go func() { - result := store.StoreResult{} - + return store.Do(func(result *store.StoreResult) { if result.Err = token.IsValid(); result.Err != nil { - storeChannel <- result - close(storeChannel) return } if err := s.GetMaster().Insert(token); err != nil { result.Err = model.NewAppError("SqlTokenStore.Save", "store.sql_recover.save.app_error", nil, "", http.StatusInternalServerError) } - - storeChannel <- result - close(storeChannel) - }() - - return storeChannel + }) } func (s SqlTokenStore) Delete(token string) store.StoreChannel { - - storeChannel := make(store.StoreChannel, 1) - - go func() { - result := store.StoreResult{} - + return store.Do(func(result *store.StoreResult) { if _, err := s.GetMaster().Exec("DELETE FROM Tokens WHERE Token = :Token", map[string]interface{}{"Token": token}); err != nil { result.Err = model.NewAppError("SqlTokenStore.Delete", "store.sql_recover.delete.app_error", nil, "", http.StatusInternalServerError) } - - storeChannel <- result - close(storeChannel) - }() - - return storeChannel + }) } func (s SqlTokenStore) GetByToken(tokenString string) store.StoreChannel { - - storeChannel := make(store.StoreChannel, 1) - - go func() { - result := store.StoreResult{} - + return store.Do(func(result *store.StoreResult) { token := model.Token{} if err := s.GetReplica().SelectOne(&token, "SELECT * FROM Tokens WHERE Token = :Token", map[string]interface{}{"Token": tokenString}); err != nil { @@ -93,12 +66,7 @@ func (s SqlTokenStore) GetByToken(tokenString string) store.StoreChannel { } result.Data = &token - - storeChannel <- result - close(storeChannel) - }() - - return storeChannel + }) } func (s SqlTokenStore) Cleanup() { |