65 lines
2.1 KiB
Ruby
65 lines
2.1 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 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')
|