Migrate "Team.ClearAllCustomRoleAssignments" to Sync by default (#11587)

Этот коммит содержится в:
Rajiv Kushwaha
2019-07-11 19:24:56 +05:30
коммит произвёл George Goldberg
родитель e8f77daa8a
Коммит 76f4fccf8a
5 изменённых файлов: 46 добавлений и 50 удалений

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

@@ -33,8 +33,8 @@ func (a *App) ResetPermissionsSystem() *model.AppError {
} }
// Reset all Custom Role assignments to TeamMembers. // Reset all Custom Role assignments to TeamMembers.
if result := <-a.Srv.Store.Team().ClearAllCustomRoleAssignments(); result.Err != nil { if err := a.Srv.Store.Team().ClearAllCustomRoleAssignments(); err != nil {
return result.Err return err
} }
// Reset all Custom Role assignments to ChannelMembers. // Reset all Custom Role assignments to ChannelMembers.

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

@@ -886,8 +886,8 @@ func (s SqlTeamStore) InvalidateAllTeamIdsForUser(userId string) {
} }
} }
func (s SqlTeamStore) ClearAllCustomRoleAssignments() store.StoreChannel { func (s SqlTeamStore) ClearAllCustomRoleAssignments() *model.AppError {
return store.Do(func(result *store.StoreResult) {
builtInRoles := model.MakeDefaultRoles() builtInRoles := model.MakeDefaultRoles()
lastUserId := strings.Repeat("0", 26) lastUserId := strings.Repeat("0", 26)
lastTeamId := strings.Repeat("0", 26) lastTeamId := strings.Repeat("0", 26)
@@ -897,15 +897,13 @@ func (s SqlTeamStore) ClearAllCustomRoleAssignments() store.StoreChannel {
var err error var err error
if transaction, err = s.GetMaster().Begin(); err != nil { if transaction, err = s.GetMaster().Begin(); err != nil {
result.Err = model.NewAppError("SqlTeamStore.ClearAllCustomRoleAssignments", "store.sql_team.clear_all_custom_role_assignments.open_transaction.app_error", nil, err.Error(), http.StatusInternalServerError) return model.NewAppError("SqlTeamStore.ClearAllCustomRoleAssignments", "store.sql_team.clear_all_custom_role_assignments.open_transaction.app_error", nil, err.Error(), http.StatusInternalServerError)
return
} }
defer finalizeTransaction(transaction) defer finalizeTransaction(transaction)
var teamMembers []*teamMember var teamMembers []*teamMember
if _, err := transaction.Select(&teamMembers, "SELECT * from TeamMembers WHERE (TeamId, UserId) > (:TeamId, :UserId) ORDER BY TeamId, UserId LIMIT 1000", map[string]interface{}{"TeamId": lastTeamId, "UserId": lastUserId}); err != nil { if _, err := transaction.Select(&teamMembers, "SELECT * from TeamMembers WHERE (TeamId, UserId) > (:TeamId, :UserId) ORDER BY TeamId, UserId LIMIT 1000", map[string]interface{}{"TeamId": lastTeamId, "UserId": lastUserId}); err != nil {
result.Err = model.NewAppError("SqlTeamStore.ClearAllCustomRoleAssignments", "store.sql_team.clear_all_custom_role_assignments.select.app_error", nil, err.Error(), http.StatusInternalServerError) return model.NewAppError("SqlTeamStore.ClearAllCustomRoleAssignments", "store.sql_team.clear_all_custom_role_assignments.select.app_error", nil, err.Error(), http.StatusInternalServerError)
return
} }
if len(teamMembers) == 0 { if len(teamMembers) == 0 {
@@ -930,18 +928,16 @@ func (s SqlTeamStore) ClearAllCustomRoleAssignments() store.StoreChannel {
newRolesString := strings.Join(newRoles, " ") newRolesString := strings.Join(newRoles, " ")
if newRolesString != member.Roles { if newRolesString != member.Roles {
if _, err := transaction.Exec("UPDATE TeamMembers SET Roles = :Roles WHERE UserId = :UserId AND TeamId = :TeamId", map[string]interface{}{"Roles": newRolesString, "TeamId": member.TeamId, "UserId": member.UserId}); err != nil { if _, err := transaction.Exec("UPDATE TeamMembers SET Roles = :Roles WHERE UserId = :UserId AND TeamId = :TeamId", map[string]interface{}{"Roles": newRolesString, "TeamId": member.TeamId, "UserId": member.UserId}); err != nil {
result.Err = model.NewAppError("SqlTeamStore.ClearAllCustomRoleAssignments", "store.sql_team.clear_all_custom_role_assignments.update.app_error", nil, err.Error(), http.StatusInternalServerError) return model.NewAppError("SqlTeamStore.ClearAllCustomRoleAssignments", "store.sql_team.clear_all_custom_role_assignments.update.app_error", nil, err.Error(), http.StatusInternalServerError)
return
} }
} }
} }
if err := transaction.Commit(); err != nil { if err := transaction.Commit(); err != nil {
result.Err = model.NewAppError("SqlTeamStore.ClearAllCustomRoleAssignments", "store.sql_team.clear_all_custom_role_assignments.commit_transaction.app_error", nil, err.Error(), http.StatusInternalServerError) return model.NewAppError("SqlTeamStore.ClearAllCustomRoleAssignments", "store.sql_team.clear_all_custom_role_assignments.commit_transaction.app_error", nil, err.Error(), http.StatusInternalServerError)
return
} }
} }
}) return nil
} }
func (s SqlTeamStore) AnalyticsGetTeamCountForScheme(schemeId string) store.StoreChannel { func (s SqlTeamStore) AnalyticsGetTeamCountForScheme(schemeId string) store.StoreChannel {

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

@@ -118,7 +118,7 @@ type TeamStore interface {
GetTeamsByScheme(schemeId string, offset int, limit int) ([]*model.Team, *model.AppError) GetTeamsByScheme(schemeId string, offset int, limit int) ([]*model.Team, *model.AppError)
MigrateTeamMembers(fromTeamId string, fromUserId string) (map[string]string, *model.AppError) MigrateTeamMembers(fromTeamId string, fromUserId string) (map[string]string, *model.AppError)
ResetAllTeamSchemes() *model.AppError ResetAllTeamSchemes() *model.AppError
ClearAllCustomRoleAssignments() StoreChannel ClearAllCustomRoleAssignments() *model.AppError
AnalyticsGetTeamCountForScheme(schemeId string) StoreChannel AnalyticsGetTeamCountForScheme(schemeId string) StoreChannel
GetAllForExportAfter(limit int, afterId string) ([]*model.TeamForExport, *model.AppError) GetAllForExportAfter(limit int, afterId string) ([]*model.TeamForExport, *model.AppError)
GetTeamMembersForExport(userId string) ([]*model.TeamMemberForExport, *model.AppError) GetTeamMembersForExport(userId string) ([]*model.TeamMemberForExport, *model.AppError)

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

@@ -53,15 +53,15 @@ func (_m *TeamStore) AnalyticsTeamCount() (int64, *model.AppError) {
} }
// ClearAllCustomRoleAssignments provides a mock function with given fields: // ClearAllCustomRoleAssignments provides a mock function with given fields:
func (_m *TeamStore) ClearAllCustomRoleAssignments() store.StoreChannel { func (_m *TeamStore) ClearAllCustomRoleAssignments() *model.AppError {
ret := _m.Called() ret := _m.Called()
var r0 store.StoreChannel var r0 *model.AppError
if rf, ok := ret.Get(0).(func() store.StoreChannel); ok { if rf, ok := ret.Get(0).(func() *model.AppError); ok {
r0 = rf() r0 = rf()
} else { } else {
if ret.Get(0) != nil { if ret.Get(0) != nil {
r0 = ret.Get(0).(store.StoreChannel) r0 = ret.Get(0).(*model.AppError)
} }
} }

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

@@ -1516,7 +1516,7 @@ func testTeamStoreClearAllCustomRoleAssignments(t *testing.T, ss store.Store) {
store.Must(ss.Team().SaveMember(m3, -1)) store.Must(ss.Team().SaveMember(m3, -1))
store.Must(ss.Team().SaveMember(m4, -1)) store.Must(ss.Team().SaveMember(m4, -1))
require.Nil(t, (<-ss.Team().ClearAllCustomRoleAssignments()).Err) require.Nil(t, (ss.Team().ClearAllCustomRoleAssignments()))
r1, err := ss.Team().GetMember(m1.TeamId, m1.UserId) r1, err := ss.Team().GetMember(m1.TeamId, m1.UserId)
require.Nil(t, err) require.Nil(t, err)