debian-mirror-gitlab/app/graphql/resolvers/work_items_resolver.rb
2022-10-11 01:57:18 +05:30

82 lines
2.4 KiB
Ruby

# frozen_string_literal: true
module Resolvers
class WorkItemsResolver < BaseResolver
include SearchArguments
include LooksAhead
type Types::WorkItemType.connection_type, null: true
argument :iid, GraphQL::Types::String,
required: false,
description: 'IID of the issue. For example, "1".'
argument :iids, [GraphQL::Types::String],
required: false,
description: 'List of IIDs of work items. For example, `["1", "2"]`.'
argument :sort, Types::WorkItemSortEnum,
description: 'Sort work items by this criteria.',
required: false,
default_value: :created_desc
argument :state, Types::IssuableStateEnum,
required: false,
description: 'Current state of this work item.'
argument :types, [Types::IssueTypeEnum],
as: :issue_types,
description: 'Filter work items by the given work item types.',
required: false
def resolve_with_lookahead(**args)
return WorkItem.none if resource_parent.nil? || !resource_parent.work_items_feature_flag_enabled?
finder = ::WorkItems::WorkItemsFinder.new(current_user, prepare_finder_params(args))
Gitlab::Graphql::Loaders::IssuableLoader.new(resource_parent, finder).batching_find_all { |q| apply_lookahead(q) }
end
private
def preloads
{
last_edited_by: :last_edited_by
}
end
# Allows to apply lookahead for fields
# selected from WidgetInterface
override :node_selection
def node_selection
selected_fields = super
return unless selected_fields
selected_fields.selection(:widgets)
end
def unconditional_includes
[
{
project: [:project_feature, :group]
},
:author
]
end
def prepare_finder_params(args)
params = super(args)
params[:iids] ||= [params.delete(:iid)].compact if params[:iid]
params
end
def resource_parent
# The project could have been loaded in batch by `BatchLoader`.
# At this point we need the `id` of the project to query for work items, so
# make sure it's loaded and not `nil` before continuing.
strong_memoize(:resource_parent) do
object.respond_to?(:sync) ? object.sync : object
end
end
end
end
Resolvers::WorkItemsResolver.prepend_mod_with('Resolvers::WorkItemsResolver')