Migrate "User.GetSystemAdminProfiles" to Sync by default (#11491)

Этот коммит содержится в:
piperRyan
2019-07-02 11:16:14 -06:00
коммит произвёл Jesús Espino
родитель 0b6e4e2f16
Коммит 45b47ca901
5 изменённых файлов: 38 добавлений и 34 удалений

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

@@ -94,12 +94,11 @@ func (s *Server) DoSecurityUpdateCheck() {
for _, bulletin := range bulletins { for _, bulletin := range bulletins {
if bulletin.AppliesToVersion == model.CurrentVersion { if bulletin.AppliesToVersion == model.CurrentVersion {
if props["SecurityBulletin_"+bulletin.Id] == "" { if props["SecurityBulletin_"+bulletin.Id] == "" {
results := <-s.Store.User().GetSystemAdminProfiles() users, userErr := s.Store.User().GetSystemAdminProfiles()
if results.Err != nil { if userErr != nil {
mlog.Error("Failed to get system admins for security update information from Mattermost.") mlog.Error("Failed to get system admins for security update information from Mattermost.")
return return
} }
users := results.Data.(map[string]*model.User)
resBody, err := http.Get(SECURITY_URL + "/bulletins/" + bulletin.Id) resBody, err := http.Get(SECURITY_URL + "/bulletins/" + bulletin.Id)
if err != nil { if err != nil {

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

@@ -972,33 +972,29 @@ func (us SqlUserStore) GetProfileByGroupChannelIdsForUser(userId string, channel
return usersByChannelId, nil return usersByChannelId, nil
} }
func (us SqlUserStore) GetSystemAdminProfiles() store.StoreChannel { func (us SqlUserStore) GetSystemAdminProfiles() (map[string]*model.User, *model.AppError) {
return store.Do(func(result *store.StoreResult) { query := us.usersQuery.
query := us.usersQuery. Where("Roles LIKE ?", "%system_admin%").
Where("Roles LIKE ?", "%system_admin%"). OrderBy("u.Username ASC")
OrderBy("u.Username ASC")
queryString, args, err := query.ToSql() queryString, args, err := query.ToSql()
if err != nil { if err != nil {
result.Err = model.NewAppError("SqlUserStore.GetSystemAdminProfiles", "store.sql_user.app_error", nil, err.Error(), http.StatusInternalServerError) return nil, model.NewAppError("SqlUserStore.GetSystemAdminProfiles", "store.sql_user.app_error", nil, err.Error(), http.StatusInternalServerError)
return }
}
var users []*model.User var users []*model.User
if _, err := us.GetReplica().Select(&users, queryString, args...); err != nil { if _, err := us.GetReplica().Select(&users, queryString, args...); err != nil {
result.Err = model.NewAppError("SqlUserStore.GetSystemAdminProfiles", "store.sql_user.get_sysadmin_profiles.app_error", nil, err.Error(), http.StatusInternalServerError) return nil, model.NewAppError("SqlUserStore.GetSystemAdminProfiles", "store.sql_user.get_sysadmin_profiles.app_error", nil, err.Error(), http.StatusInternalServerError)
return }
}
userMap := make(map[string]*model.User) userMap := make(map[string]*model.User)
for _, u := range users { for _, u := range users {
u.Sanitize(map[string]bool{}) u.Sanitize(map[string]bool{})
userMap[u.Id] = u userMap[u.Id] = u
} }
result.Data = userMap return userMap, nil
})
} }
func (us SqlUserStore) GetByEmail(email string) (*model.User, *model.AppError) { func (us SqlUserStore) GetByEmail(email string) (*model.User, *model.AppError) {

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

@@ -280,7 +280,7 @@ type UserStore interface {
GetEtagForAllProfiles() StoreChannel GetEtagForAllProfiles() StoreChannel
GetEtagForProfiles(teamId string) StoreChannel GetEtagForProfiles(teamId string) StoreChannel
UpdateFailedPasswordAttempts(userId string, attempts int) StoreChannel UpdateFailedPasswordAttempts(userId string, attempts int) StoreChannel
GetSystemAdminProfiles() StoreChannel GetSystemAdminProfiles() (map[string]*model.User, *model.AppError)
PermanentDelete(userId string) *model.AppError PermanentDelete(userId string) *model.AppError
AnalyticsActiveCount(time int64) StoreChannel AnalyticsActiveCount(time int64) StoreChannel
GetUnreadCount(userId string) (int64, error) GetUnreadCount(userId string) (int64, error)

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

@@ -633,19 +633,28 @@ func (_m *UserStore) GetRecentlyActiveUsersForTeam(teamId string, offset int, li
} }
// GetSystemAdminProfiles provides a mock function with given fields: // GetSystemAdminProfiles provides a mock function with given fields:
func (_m *UserStore) GetSystemAdminProfiles() store.StoreChannel { func (_m *UserStore) GetSystemAdminProfiles() (map[string]*model.User, *model.AppError) {
ret := _m.Called() ret := _m.Called()
var r0 store.StoreChannel var r0 map[string]*model.User
if rf, ok := ret.Get(0).(func() store.StoreChannel); ok { if rf, ok := ret.Get(0).(func() map[string]*model.User); 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).(map[string]*model.User)
} }
} }
return r0 var r1 *model.AppError
if rf, ok := ret.Get(1).(func() *model.AppError); ok {
r1 = rf()
} else {
if ret.Get(1) != nil {
r1 = ret.Get(1).(*model.AppError)
}
}
return r0, r1
} }
// GetTeamGroupUsers provides a mock function with given fields: teamID // GetTeamGroupUsers provides a mock function with given fields: teamID

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

@@ -1450,12 +1450,12 @@ func testUserStoreGetSystemAdminProfiles(t *testing.T, ss store.Store) {
defer func() { require.Nil(t, ss.Bot().PermanentDelete(u3.Id)) }() defer func() { require.Nil(t, ss.Bot().PermanentDelete(u3.Id)) }()
t.Run("all system admin profiles", func(t *testing.T) { t.Run("all system admin profiles", func(t *testing.T) {
result := <-ss.User().GetSystemAdminProfiles() result, userError := ss.User().GetSystemAdminProfiles()
require.Nil(t, result.Err) require.Nil(t, userError)
assert.Equal(t, map[string]*model.User{ assert.Equal(t, map[string]*model.User{
u1.Id: sanitized(u1), u1.Id: sanitized(u1),
u3.Id: sanitized(u3), u3.Id: sanitized(u3),
}, result.Data.(map[string]*model.User)) }, result)
}) })
} }