MM-16482 Migrate UserAccessToken.Delete to Sync by default (#11356)

* MM-16482 Migrate UserAccessToken.Delete to Sync by default

* err mistake

* [Requested Change] Removed else block

* gofmt change

* gofmt change
Этот коммит содержится в:
Micah Thompson
2019-06-24 17:25:19 -04:00
коммит произвёл Jesús Espino
родитель c07b7046ca
Коммит eb9415a668
5 изменённых файлов: 29 добавлений и 34 удалений

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

@@ -312,8 +312,8 @@ func (a *App) RevokeUserAccessToken(token *model.UserAccessToken) *model.AppErro
var session *model.Session var session *model.Session
session, _ = a.Srv.Store.Session().Get(token.Token) session, _ = a.Srv.Store.Session().Get(token.Token)
if result := <-a.Srv.Store.UserAccessToken().Delete(token.Id); result.Err != nil { if err := a.Srv.Store.UserAccessToken().Delete(token.Id); err != nil {
return result.Err return err
} }
if session == nil { if session == nil {

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

@@ -48,29 +48,26 @@ func (s SqlUserAccessTokenStore) Save(token *model.UserAccessToken) (*model.User
return token, nil return token, nil
} }
func (s SqlUserAccessTokenStore) Delete(tokenId string) store.StoreChannel { func (s SqlUserAccessTokenStore) Delete(tokenId string) *model.AppError {
return store.Do(func(result *store.StoreResult) { transaction, err := s.GetMaster().Begin()
transaction, err := s.GetMaster().Begin() if err != nil {
if err != nil { return model.NewAppError("SqlUserAccessTokenStore.Delete", "store.sql_user_access_token.delete.app_error", nil, err.Error(), http.StatusInternalServerError)
result.Err = model.NewAppError("SqlUserAccessTokenStore.Delete", "store.sql_user_access_token.delete.app_error", nil, err.Error(), http.StatusInternalServerError) }
} else {
defer finalizeTransaction(transaction)
if extrasResult := s.deleteSessionsAndTokensById(transaction, tokenId); extrasResult.Err != nil {
*result = extrasResult
}
if result.Err == nil { defer finalizeTransaction(transaction)
if err := transaction.Commit(); err != nil {
// don't need to rollback here since the transaction is already closed if err := s.deleteSessionsAndTokensById(transaction, tokenId); err == nil {
result.Err = model.NewAppError("SqlUserAccessTokenStore.Delete", "store.sql_user_access_token.delete.app_error", nil, err.Error(), http.StatusInternalServerError) if err := transaction.Commit(); err != nil {
} // don't need to rollback here since the transaction is already closed
} return model.NewAppError("SqlUserAccessTokenStore.Delete", "store.sql_user_access_token.delete.app_error", nil, err.Error(), http.StatusInternalServerError)
} }
}) }
return nil
} }
func (s SqlUserAccessTokenStore) deleteSessionsAndTokensById(transaction *gorp.Transaction, tokenId string) store.StoreResult { func (s SqlUserAccessTokenStore) deleteSessionsAndTokensById(transaction *gorp.Transaction, tokenId string) *model.AppError {
result := store.StoreResult{}
query := "" query := ""
if s.DriverName() == model.DATABASE_DRIVER_POSTGRES { if s.DriverName() == model.DATABASE_DRIVER_POSTGRES {
@@ -80,21 +77,19 @@ func (s SqlUserAccessTokenStore) deleteSessionsAndTokensById(transaction *gorp.T
} }
if _, err := transaction.Exec(query, map[string]interface{}{"Id": tokenId}); err != nil { if _, err := transaction.Exec(query, map[string]interface{}{"Id": tokenId}); err != nil {
result.Err = model.NewAppError("SqlUserAccessTokenStore.deleteSessionsById", "store.sql_user_access_token.delete.app_error", nil, "id="+tokenId+", err="+err.Error(), http.StatusInternalServerError) return model.NewAppError("SqlUserAccessTokenStore.deleteSessionsById", "store.sql_user_access_token.delete.app_error", nil, "id="+tokenId+", err="+err.Error(), http.StatusInternalServerError)
return result
} }
return s.deleteTokensById(transaction, tokenId) return s.deleteTokensById(transaction, tokenId)
} }
func (s SqlUserAccessTokenStore) deleteTokensById(transaction *gorp.Transaction, tokenId string) store.StoreResult { func (s SqlUserAccessTokenStore) deleteTokensById(transaction *gorp.Transaction, tokenId string) *model.AppError {
result := store.StoreResult{}
if _, err := transaction.Exec("DELETE FROM UserAccessTokens WHERE Id = :Id", map[string]interface{}{"Id": tokenId}); err != nil { if _, err := transaction.Exec("DELETE FROM UserAccessTokens WHERE Id = :Id", map[string]interface{}{"Id": tokenId}); err != nil {
result.Err = model.NewAppError("SqlUserAccessTokenStore.deleteTokensById", "store.sql_user_access_token.delete.app_error", nil, "", http.StatusInternalServerError) return model.NewAppError("SqlUserAccessTokenStore.deleteTokensById", "store.sql_user_access_token.delete.app_error", nil, "", http.StatusInternalServerError)
} }
return result return nil
} }
func (s SqlUserAccessTokenStore) DeleteAllForUser(userId string) store.StoreChannel { func (s SqlUserAccessTokenStore) DeleteAllForUser(userId string) store.StoreChannel {

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

@@ -518,7 +518,7 @@ type JobStore interface {
type UserAccessTokenStore interface { type UserAccessTokenStore interface {
Save(token *model.UserAccessToken) (*model.UserAccessToken, *model.AppError) Save(token *model.UserAccessToken) (*model.UserAccessToken, *model.AppError)
Delete(tokenId string) StoreChannel Delete(tokenId string) *model.AppError
DeleteAllForUser(userId string) StoreChannel DeleteAllForUser(userId string) StoreChannel
Get(tokenId string) (*model.UserAccessToken, *model.AppError) Get(tokenId string) (*model.UserAccessToken, *model.AppError)
GetAll(offset int, limit int) ([]*model.UserAccessToken, *model.AppError) GetAll(offset int, limit int) ([]*model.UserAccessToken, *model.AppError)

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

@@ -14,15 +14,15 @@ type UserAccessTokenStore struct {
} }
// Delete provides a mock function with given fields: tokenId // Delete provides a mock function with given fields: tokenId
func (_m *UserAccessTokenStore) Delete(tokenId string) store.StoreChannel { func (_m *UserAccessTokenStore) Delete(tokenId string) *model.AppError {
ret := _m.Called(tokenId) ret := _m.Called(tokenId)
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(tokenId) r0 = rf(tokenId)
} else { } else {
if ret.Get(0) != nil { if ret.Get(0) != nil {
r0 = ret.Get(0).(store.StoreChannel) r0 = ret.Get(0).(*model.AppError)
} }
} }

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

@@ -63,8 +63,8 @@ func testUserAccessTokenSaveGetDelete(t *testing.T, ss store.Store) {
t.Fatal("received incorrect number of tokens after save") t.Fatal("received incorrect number of tokens after save")
} }
if result := <-ss.UserAccessToken().Delete(uat.Id); result.Err != nil { if err = ss.UserAccessToken().Delete(uat.Id); err != nil {
t.Fatal(result.Err) t.Fatal(err)
} }
if _, err = ss.Session().Get(s1.Token); err == nil { if _, err = ss.Session().Get(s1.Token); err == nil {