mirror of
https://github.com/discourse/discourse.git
synced 2024-11-23 06:29:30 +08:00
4eee1320b0
At the time of writing, this is how the `TopicPosterSerializer` looks like: ``` class TopicPosterSerializer < ApplicationSerializer attributes :extras, :description has_one :user, serializer: PosterSerializer has_one :primary_group, serializer: PrimaryGroupSerializer has_one :flair_group, serializer: FlairGroupSerializer end ``` Within `PosterSerializer`, the `primary_group` and `flair_group` association is requested on the `user` object. However, the associations have not been loaded on the `user` object at this point leading to the N+1 queries problem. One may wonder why the associations have not been loaded when the `TopicPosterSerializer` has `has_one :primary_group` and `has_one :flair_group`. It turns out that `TopicPoster` is just a struct containing the `user`, `primary_group` and `flair_group` objects. The `primary_group` and `flair_group` ActiveRecord objects are loaded seperately in `UserLookup` and not preloaded when querying for the users. This is done for performance reason so that we are able to load the `primary_group` and `flair_group` records in a single query without duplication.
70 lines
1.7 KiB
Ruby
70 lines
1.7 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
class UserLookup
|
|
def self.lookup_columns
|
|
@user_lookup_columns ||= %i{id username name uploaded_avatar_id primary_group_id flair_group_id admin moderator trust_level}
|
|
end
|
|
|
|
def self.group_lookup_columns
|
|
@group_lookup_columns ||= %i{id name flair_icon flair_upload_id flair_bg_color flair_color}
|
|
end
|
|
|
|
def initialize(user_ids = [])
|
|
@user_ids = user_ids.tap(&:compact!).tap(&:uniq!).tap(&:flatten!)
|
|
end
|
|
|
|
# Lookup a user by id
|
|
def [](user_id)
|
|
users[user_id]
|
|
end
|
|
|
|
def primary_groups
|
|
@primary_groups ||= users.values.each_with_object({}) do |user, hash|
|
|
if user.primary_group_id
|
|
group = groups[user.primary_group_id]
|
|
set_user_group_preload(user, group, :primary_group)
|
|
hash[user.id] = group
|
|
end
|
|
end
|
|
end
|
|
|
|
def flair_groups
|
|
@flair_groups ||= users.values.each_with_object({}) do |user, hash|
|
|
if user.flair_group_id
|
|
group = groups[user.flair_group_id]
|
|
set_user_group_preload(user, group, :flair_group)
|
|
hash[user.id] = group
|
|
end
|
|
end
|
|
end
|
|
|
|
private
|
|
|
|
def set_user_group_preload(user, group, group_association_name)
|
|
association = user.association(group_association_name)
|
|
association.target = group
|
|
end
|
|
|
|
def users
|
|
@users ||= User
|
|
.where(id: @user_ids)
|
|
.select(self.class.lookup_columns)
|
|
.index_by(&:id)
|
|
end
|
|
|
|
def groups
|
|
@group_lookup ||= begin
|
|
group_ids = users.values.map { |u| [u.primary_group_id, u.flair_group_id] }
|
|
group_ids.flatten!
|
|
group_ids.uniq!
|
|
group_ids.compact!
|
|
|
|
Group
|
|
.includes(:flair_upload)
|
|
.where(id: group_ids)
|
|
.select(self.class.group_lookup_columns)
|
|
.index_by(&:id)
|
|
end
|
|
end
|
|
end
|