[MM-16513] Migrate Token.Delete to Sync by default (#11365)

* Made changes to TokenStore interface, Delete()

* Modified user.go

* Refactored app/user.go
Этот коммит содержится в:
Deepak Sah
2019-06-27 16:44:17 +05:30
коммит произвёл Jesús Espino
родитель eb8b475497
Коммит 77ed6015fd
4 изменённых файлов: 11 добавлений и 16 удалений

Просмотреть файл

@@ -1360,11 +1360,7 @@ func (a *App) GetPasswordRecoveryToken(token string) (*model.Token, *model.AppEr
} }
func (a *App) DeleteToken(token *model.Token) *model.AppError { func (a *App) DeleteToken(token *model.Token) *model.AppError {
if result := <-a.Srv.Store.Token().Delete(token.Token); result.Err != nil { return a.Srv.Store.Token().Delete(token.Token)
return result.Err
}
return nil
} }
func (a *App) UpdateUserRoles(userId string, newRoles string, sendWebSocketEvent bool) (*model.User, *model.AppError) { func (a *App) UpdateUserRoles(userId string, newRoles string, sendWebSocketEvent bool) (*model.User, *model.AppError) {

Просмотреть файл

@@ -43,12 +43,11 @@ func (s SqlTokenStore) Save(token *model.Token) *model.AppError {
return nil return nil
} }
func (s SqlTokenStore) Delete(token string) store.StoreChannel { func (s SqlTokenStore) Delete(token string) *model.AppError {
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 {
if _, err := s.GetMaster().Exec("DELETE FROM Tokens WHERE Token = :Token", map[string]interface{}{"Token": token}); err != nil { return model.NewAppError("SqlTokenStore.Delete", "store.sql_recover.delete.app_error", nil, "", http.StatusInternalServerError)
result.Err = model.NewAppError("SqlTokenStore.Delete", "store.sql_recover.delete.app_error", nil, "", http.StatusInternalServerError) }
} return nil
})
} }
func (s SqlTokenStore) GetByToken(tokenString string) (*model.Token, *model.AppError) { func (s SqlTokenStore) GetByToken(tokenString string) (*model.Token, *model.AppError) {

Просмотреть файл

@@ -450,7 +450,7 @@ type LicenseStore interface {
type TokenStore interface { type TokenStore interface {
Save(recovery *model.Token) *model.AppError Save(recovery *model.Token) *model.AppError
Delete(token string) StoreChannel Delete(token string) *model.AppError
GetByToken(token string) (*model.Token, *model.AppError) GetByToken(token string) (*model.Token, *model.AppError)
Cleanup() Cleanup()
RemoveAllTokensByType(tokenType string) StoreChannel RemoveAllTokensByType(tokenType string) StoreChannel

Просмотреть файл

@@ -19,15 +19,15 @@ func (_m *TokenStore) Cleanup() {
} }
// Delete provides a mock function with given fields: token // Delete provides a mock function with given fields: token
func (_m *TokenStore) Delete(token string) store.StoreChannel { func (_m *TokenStore) Delete(token string) *model.AppError {
ret := _m.Called(token) ret := _m.Called(token)
var r0 store.StoreChannel var r0 *model.AppError
if rf, ok := ret.Get(0).(func(string) store.StoreChannel); ok { if rf, ok := ret.Get(0).(func(string) *model.AppError); ok {
r0 = rf(token) r0 = rf(token)
} else { } else {
if ret.Get(0) != nil { if ret.Get(0) != nil {
r0 = ret.Get(0).(store.StoreChannel) r0 = ret.Get(0).(*model.AppError)
} }
} }