Migrate User.GetProfiles to sync by default (#11512)

Этот коммит содержится в:
Rodrigo Villablanca Vásquez
2019-07-05 13:57:01 -04:00
коммит произвёл Jesús Espino
родитель 25f3bf4a2b
Коммит aa14c9bbdb
7 изменённых файлов: 54 добавлений и 59 удалений

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

@@ -1017,11 +1017,10 @@ func (a *App) AddChannelMember(userId string, channel *model.Channel, userReques
func (a *App) AddDirectChannels(teamId string, user *model.User) *model.AppError { func (a *App) AddDirectChannels(teamId string, user *model.User) *model.AppError {
var profiles []*model.User var profiles []*model.User
options := &model.UserGetOptions{InTeamId: teamId, Page: 0, PerPage: 100} options := &model.UserGetOptions{InTeamId: teamId, Page: 0, PerPage: 100}
result := <-a.Srv.Store.User().GetProfiles(options) profiles, err := a.Srv.Store.User().GetProfiles(options)
if result.Err != nil { if err != nil {
return model.NewAppError("AddDirectChannels", "api.user.add_direct_channels_and_forget.failed.error", map[string]interface{}{"UserId": user.Id, "TeamId": teamId, "Error": result.Err.Error()}, "", http.StatusInternalServerError) return model.NewAppError("AddDirectChannels", "api.user.add_direct_channels_and_forget.failed.error", map[string]interface{}{"UserId": user.Id, "TeamId": teamId, "Error": err.Error()}, "", http.StatusInternalServerError)
} }
profiles = result.Data.([]*model.User)
var preferences model.Preferences var preferences model.Preferences

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

@@ -301,8 +301,7 @@ func (me *LoadTestProvider) PostsCommand(a *App, args *model.CommandArgs, messag
var usernames []string var usernames []string
options := &model.UserGetOptions{InTeamId: args.TeamId, Page: 0, PerPage: 1000} options := &model.UserGetOptions{InTeamId: args.TeamId, Page: 0, PerPage: 1000}
if result := <-a.Srv.Store.User().GetProfiles(options); result.Err == nil { if profileUsers, err := a.Srv.Store.User().GetProfiles(options); err == nil {
profileUsers := result.Data.([]*model.User)
usernames = make([]string, len(profileUsers)) usernames = make([]string, len(profileUsers))
i := 0 i := 0
for _, userprof := range profileUsers { for _, userprof := range profileUsers {

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

@@ -464,11 +464,7 @@ func (a *App) GetUsersEtag(restrictionsHash string) string {
} }
func (a *App) GetUsersInTeam(options *model.UserGetOptions) ([]*model.User, *model.AppError) { func (a *App) GetUsersInTeam(options *model.UserGetOptions) ([]*model.User, *model.AppError) {
result := <-a.Srv.Store.User().GetProfiles(options) return a.Srv.Store.User().GetProfiles(options)
if result.Err != nil {
return nil, result.Err
}
return result.Data.([]*model.User), nil
} }
func (a *App) GetUsersNotInTeam(teamId string, groupConstrained bool, offset int, limit int, viewRestrictions *model.ViewUsersRestrictions) ([]*model.User, *model.AppError) { func (a *App) GetUsersNotInTeam(teamId string, groupConstrained bool, offset int, limit int, viewRestrictions *model.ViewUsersRestrictions) ([]*model.User, *model.AppError) {

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

@@ -482,40 +482,37 @@ func (s SqlUserStore) GetEtagForProfiles(teamId string) string {
return fmt.Sprintf("%v.%v", model.CurrentVersion, updateAt) return fmt.Sprintf("%v.%v", model.CurrentVersion, updateAt)
} }
func (us SqlUserStore) GetProfiles(options *model.UserGetOptions) store.StoreChannel { func (us SqlUserStore) GetProfiles(options *model.UserGetOptions) ([]*model.User, *model.AppError) {
isPostgreSQL := us.DriverName() == model.DATABASE_DRIVER_POSTGRES isPostgreSQL := us.DriverName() == model.DATABASE_DRIVER_POSTGRES
return store.Do(func(result *store.StoreResult) { query := us.usersQuery.
query := us.usersQuery. Join("TeamMembers tm ON ( tm.UserId = u.Id AND tm.DeleteAt = 0 )").
Join("TeamMembers tm ON ( tm.UserId = u.Id AND tm.DeleteAt = 0 )"). Where("tm.TeamId = ?", options.InTeamId).
Where("tm.TeamId = ?", options.InTeamId). OrderBy("u.Username ASC").
OrderBy("u.Username ASC"). Offset(uint64(options.Page * options.PerPage)).Limit(uint64(options.PerPage))
Offset(uint64(options.Page * options.PerPage)).Limit(uint64(options.PerPage))
query = applyViewRestrictionsFilter(query, options.ViewRestrictions, true) query = applyViewRestrictionsFilter(query, options.ViewRestrictions, true)
query = applyRoleFilter(query, options.Role, isPostgreSQL) query = applyRoleFilter(query, options.Role, isPostgreSQL)
if options.Inactive { if options.Inactive {
query = query.Where("u.DeleteAt != 0") query = query.Where("u.DeleteAt != 0")
} }
queryString, args, err := query.ToSql() queryString, args, err := query.ToSql()
if err != nil { if err != nil {
result.Err = model.NewAppError("SqlUserStore.GetProfiles", "store.sql_user.app_error", nil, err.Error(), http.StatusInternalServerError) return nil, model.NewAppError("SqlUserStore.GetProfiles", "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.GetProfiles", "store.sql_user.get_profiles.app_error", nil, err.Error(), http.StatusInternalServerError) return nil, model.NewAppError("SqlUserStore.GetProfiles", "store.sql_user.get_profiles.app_error", nil, err.Error(), http.StatusInternalServerError)
return }
}
for _, u := range users { for _, u := range users {
u.Sanitize(map[string]bool{}) u.Sanitize(map[string]bool{})
} }
result.Data = users
}) return users, nil
} }
func (us SqlUserStore) InvalidateProfilesInChannelCacheByUser(userId string) { func (us SqlUserStore) InvalidateProfilesInChannelCacheByUser(userId string) {

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

@@ -270,7 +270,7 @@ type UserStore interface {
GetProfilesWithoutTeam(offset int, limit int, viewRestrictions *model.ViewUsersRestrictions) StoreChannel GetProfilesWithoutTeam(offset int, limit int, viewRestrictions *model.ViewUsersRestrictions) StoreChannel
GetProfilesByUsernames(usernames []string, viewRestrictions *model.ViewUsersRestrictions) StoreChannel GetProfilesByUsernames(usernames []string, viewRestrictions *model.ViewUsersRestrictions) StoreChannel
GetAllProfiles(options *model.UserGetOptions) ([]*model.User, *model.AppError) GetAllProfiles(options *model.UserGetOptions) ([]*model.User, *model.AppError)
GetProfiles(options *model.UserGetOptions) StoreChannel GetProfiles(options *model.UserGetOptions) ([]*model.User, *model.AppError)
GetProfileByIds(userIds []string, options *UserGetByIdsOpts, allowFromCache bool) ([]*model.User, *model.AppError) GetProfileByIds(userIds []string, options *UserGetByIdsOpts, allowFromCache bool) ([]*model.User, *model.AppError)
GetProfileByGroupChannelIdsForUser(userId string, channelIds []string) (map[string][]*model.User, *model.AppError) GetProfileByGroupChannelIdsForUser(userId string, channelIds []string) (map[string][]*model.User, *model.AppError)
InvalidatProfileCacheForUser(userId string) InvalidatProfileCacheForUser(userId string)

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

@@ -535,19 +535,28 @@ func (_m *UserStore) GetProfileByIds(userIds []string, options *store.UserGetByI
} }
// GetProfiles provides a mock function with given fields: options // GetProfiles provides a mock function with given fields: options
func (_m *UserStore) GetProfiles(options *model.UserGetOptions) store.StoreChannel { func (_m *UserStore) GetProfiles(options *model.UserGetOptions) ([]*model.User, *model.AppError) {
ret := _m.Called(options) ret := _m.Called(options)
var r0 store.StoreChannel var r0 []*model.User
if rf, ok := ret.Get(0).(func(*model.UserGetOptions) store.StoreChannel); ok { if rf, ok := ret.Get(0).(func(*model.UserGetOptions) []*model.User); ok {
r0 = rf(options) r0 = rf(options)
} else { } else {
if ret.Get(0) != nil { if ret.Get(0) != nil {
r0 = ret.Get(0).(store.StoreChannel) r0 = ret.Get(0).([]*model.User)
} }
} }
return r0 var r1 *model.AppError
if rf, ok := ret.Get(1).(func(*model.UserGetOptions) *model.AppError); ok {
r1 = rf(options)
} else {
if ret.Get(1) != nil {
r1 = ret.Get(1).(*model.AppError)
}
}
return r0, r1
} }
// GetProfilesByUsernames provides a mock function with given fields: usernames, viewRestrictions // GetProfilesByUsernames provides a mock function with given fields: usernames, viewRestrictions

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

@@ -555,14 +555,13 @@ func testUserStoreGetProfiles(t *testing.T, ss store.Store) {
store.Must(ss.Team().SaveMember(&model.TeamMember{TeamId: teamId, UserId: u5.Id}, -1)) store.Must(ss.Team().SaveMember(&model.TeamMember{TeamId: teamId, UserId: u5.Id}, -1))
t.Run("get page 0, perPage 100", func(t *testing.T) { t.Run("get page 0, perPage 100", func(t *testing.T) {
result := <-ss.User().GetProfiles(&model.UserGetOptions{ actual, err := ss.User().GetProfiles(&model.UserGetOptions{
InTeamId: teamId, InTeamId: teamId,
Page: 0, Page: 0,
PerPage: 100, PerPage: 100,
}) })
require.Nil(t, result.Err) require.Nil(t, err)
actual := result.Data.([]*model.User)
require.Equal(t, []*model.User{ require.Equal(t, []*model.User{
sanitized(u1), sanitized(u1),
sanitized(u2), sanitized(u2),
@@ -573,26 +572,24 @@ func testUserStoreGetProfiles(t *testing.T, ss store.Store) {
}) })
t.Run("get page 0, perPage 1", func(t *testing.T) { t.Run("get page 0, perPage 1", func(t *testing.T) {
result := <-ss.User().GetProfiles(&model.UserGetOptions{ actual, err := ss.User().GetProfiles(&model.UserGetOptions{
InTeamId: teamId, InTeamId: teamId,
Page: 0, Page: 0,
PerPage: 1, PerPage: 1,
}) })
require.Nil(t, result.Err) require.Nil(t, err)
actual := result.Data.([]*model.User)
require.Equal(t, []*model.User{sanitized(u1)}, actual) require.Equal(t, []*model.User{sanitized(u1)}, actual)
}) })
t.Run("get unknown team id", func(t *testing.T) { t.Run("get unknown team id", func(t *testing.T) {
result := <-ss.User().GetProfiles(&model.UserGetOptions{ actual, err := ss.User().GetProfiles(&model.UserGetOptions{
InTeamId: "123", InTeamId: "123",
Page: 0, Page: 0,
PerPage: 100, PerPage: 100,
}) })
require.Nil(t, result.Err) require.Nil(t, err)
actual := result.Data.([]*model.User)
require.Equal(t, []*model.User{}, actual) require.Equal(t, []*model.User{}, actual)
}) })
@@ -610,28 +607,26 @@ func testUserStoreGetProfiles(t *testing.T, ss store.Store) {
}) })
t.Run("filter to system_admin role", func(t *testing.T) { t.Run("filter to system_admin role", func(t *testing.T) {
result := <-ss.User().GetProfiles(&model.UserGetOptions{ actual, err := ss.User().GetProfiles(&model.UserGetOptions{
InTeamId: teamId, InTeamId: teamId,
Page: 0, Page: 0,
PerPage: 10, PerPage: 10,
Role: "system_admin", Role: "system_admin",
}) })
require.Nil(t, result.Err) require.Nil(t, err)
actual := result.Data.([]*model.User)
require.Equal(t, []*model.User{ require.Equal(t, []*model.User{
sanitized(u4), sanitized(u4),
}, actual) }, actual)
}) })
t.Run("filter to inactive", func(t *testing.T) { t.Run("filter to inactive", func(t *testing.T) {
result := <-ss.User().GetProfiles(&model.UserGetOptions{ actual, err := ss.User().GetProfiles(&model.UserGetOptions{
InTeamId: teamId, InTeamId: teamId,
Page: 0, Page: 0,
PerPage: 10, PerPage: 10,
Inactive: true, Inactive: true,
}) })
require.Nil(t, result.Err) require.Nil(t, err)
actual := result.Data.([]*model.User)
require.Equal(t, []*model.User{ require.Equal(t, []*model.User{
sanitized(u5), sanitized(u5),
}, actual) }, actual)