feat(graphql): add tracing support for managers

This commit is contained in:
Paul Chavard 2023-02-15 17:44:49 +01:00
parent 2c729ff396
commit 74c6d45b74

View file

@ -1,10 +1,6 @@
class API::V2::GraphqlController < API::V2::BaseController class API::V2::GraphqlController < API::V2::BaseController
def execute def execute
result = API::V2::Schema.execute(query, result = tracing? ? instrumented { perform_query } : perform_query
variables: variables,
context: context,
operation_name: params[:operationName])
render json: result render json: result
rescue GraphQL::ParseError, JSON::ParserError => exception rescue GraphQL::ParseError, JSON::ParserError => exception
handle_parse_error(exception) handle_parse_error(exception)
@ -18,6 +14,13 @@ class API::V2::GraphqlController < API::V2::BaseController
private private
def perform_query
API::V2::Schema.execute(query,
variables: variables,
context: context,
operation_name: params[:operationName])
end
def append_info_to_payload(payload) def append_info_to_payload(payload)
super super
@ -124,4 +127,27 @@ class API::V2::GraphqlController < API::V2::BaseController
data: nil data: nil
}, status: 500 }, status: 500
end end
def tracing?
params[:tracing].present? && (Rails.env.development? || manager?)
end
def manager?
administrateur_signed_in? && AdministrateursProcedure.exists?(administrateur: current_administrateur, manager: true)
end
def instrumented
events = []
ActiveSupport::Notifications.subscribed(-> (_name, start, finish, _id, payload) { events << { start: Time.zone.at(start), duration: (finish - start) * 1000, sql: payload[:sql] } }, "sql.active_record", monotonic: true) do
result = yield
result.merge(extensions: {
tracing: {
startTime: events.first[:start],
endTime: events.last[:start],
duration: events.sum { _1[:duration] },
events: events.sort_by { _1[:duration] }.reverse
}
})
end
end
end end