mirror of
https://github.com/discourse/discourse.git
synced 2024-12-16 23:13:53 +08:00
40 lines
901 B
Ruby
40 lines
901 B
Ruby
class PrimaryGroupLookup
|
|
def initialize(user_ids = [])
|
|
@user_ids = user_ids.tap(&:compact!).tap(&:uniq!).tap(&:flatten!)
|
|
end
|
|
|
|
# Lookup primary group for a given user id
|
|
def [](user_id)
|
|
users[user_id]
|
|
end
|
|
|
|
private
|
|
|
|
def self.lookup_columns
|
|
@lookup_columns ||= %i{id name flair_url flair_bg_color flair_color}
|
|
end
|
|
|
|
def users
|
|
@users ||= user_lookup_hash
|
|
end
|
|
|
|
def user_lookup_hash
|
|
users_with_primary_group = User.where(id: @user_ids)
|
|
.where.not(primary_group_id: nil)
|
|
.select(:id, :primary_group_id)
|
|
|
|
group_lookup = {}
|
|
group_ids = users_with_primary_group.map(&:primary_group_id)
|
|
group_ids.uniq!
|
|
|
|
Group.where(id: group_ids).select(self.class.lookup_columns)
|
|
.each { |g| group_lookup[g.id] = g }
|
|
|
|
hash = {}
|
|
users_with_primary_group.each do |u|
|
|
hash[u.id] = group_lookup[u.primary_group_id]
|
|
end
|
|
hash
|
|
end
|
|
end
|