mirror of
https://github.com/discourse/discourse.git
synced 2026-08-06 13:08:40 +08:00
4e264b4d4dAdmin dashboard sections normally execute in a bounded thread pool. `rack-mini-profiler`'s flamegraph mode uses `stackprof` which only samples the request thread, so asynchronous code and SQLs executed in the thread pool do not appear in the flamegraph. This commit gives `AdminDashboardSectionLoader` an opt-out from parallel execution and uses it for authorized flamegraph and async-flamegraph requests. Ordinary dashboard requests continue using the existing thread pool, preserving concurrency.
111 lines
2.9 KiB
Ruby
Vendored
111 lines
2.9 KiB
Ruby
Vendored
# frozen_string_literal: true
|
|
|
|
class AdminDashboardSectionLoader
|
|
def self.build(section_ids:, current_user:, start_date:, end_date:, parallel: true)
|
|
new(
|
|
section_ids: section_ids,
|
|
current_user: current_user,
|
|
start_date: start_date,
|
|
end_date: end_date,
|
|
).build(parallel: parallel)
|
|
end
|
|
|
|
def self.pool_size
|
|
desired =
|
|
AdminDashboardSectionConfiguration::KNOWN_SECTIONS.size +
|
|
DiscoursePluginRegistry.admin_dashboard_sections.size
|
|
|
|
available = [ActiveRecord::Base.connection_pool.size - 1, 1].max
|
|
[desired, available].min
|
|
end
|
|
|
|
def self.thread_pool
|
|
@thread_pool ||=
|
|
Scheduler::ThreadPool.new(min_threads: 0, max_threads: pool_size, idle_time: 30)
|
|
end
|
|
|
|
def initialize(section_ids:, current_user:, start_date:, end_date:)
|
|
@section_ids = section_ids
|
|
@current_user = current_user
|
|
@start_date = start_date
|
|
@end_date = end_date
|
|
end
|
|
|
|
def build(parallel: true)
|
|
if parallel
|
|
build_in_parallel
|
|
else
|
|
section_ids.map { |id| build_section(id) }
|
|
end
|
|
end
|
|
|
|
private
|
|
|
|
attr_reader :section_ids, :current_user, :start_date, :end_date
|
|
|
|
def build_in_parallel
|
|
results = Queue.new
|
|
|
|
section_ids.each do |id|
|
|
self.class.thread_pool.post do
|
|
ActiveRecord::Base.with_connection(prevent_permanent_checkout: true) do
|
|
results << { id: id, data: section_data(id, current_user) }
|
|
end
|
|
rescue StandardError => e
|
|
results << { id: id, error: e }
|
|
end
|
|
end
|
|
|
|
results_by_id = {}
|
|
|
|
section_ids.size.times do
|
|
result = results.pop
|
|
|
|
result = failed_result(result) if result[:error]
|
|
|
|
results_by_id[result[:id]] = result
|
|
end
|
|
|
|
section_ids.map { |id| results_by_id.fetch(id) }
|
|
end
|
|
|
|
def build_section(id)
|
|
{ id: id, data: section_data(id, current_user) }
|
|
rescue StandardError => e
|
|
failed_result(id: id, error: e)
|
|
end
|
|
|
|
def failed_result(result)
|
|
Discourse.warn_exception(
|
|
result[:error],
|
|
message: "Failed to build admin dashboard section",
|
|
env: {
|
|
section_id: result[:id],
|
|
},
|
|
)
|
|
|
|
{ id: result[:id], data: nil, error: true }
|
|
end
|
|
|
|
def section_data(id, user)
|
|
case id
|
|
when "highlights"
|
|
AdminDashboardHighlights.build(start_date: start_date, end_date: end_date)
|
|
when "traffic"
|
|
AdminDashboardSiteTraffic.build(
|
|
start_date: start_date,
|
|
end_date: end_date,
|
|
guardian: user.guardian,
|
|
)
|
|
when "engagement"
|
|
AdminDashboardEngagement.build(start_date: start_date, end_date: end_date, current_user: user)
|
|
when "reports"
|
|
AdminDashboard::Reports::Section.build(guardian: user.guardian)
|
|
when "search"
|
|
AdminDashboardSearch.build(start_date: start_date, end_date: end_date)
|
|
else
|
|
section = DiscoursePluginRegistry.admin_dashboard_sections.find { |s| s[:id] == id }
|
|
section&.dig(:loader)&.call(start_date: start_date, end_date: end_date, current_user: user)
|
|
end
|
|
end
|
|
end
|
