summaryrefslogtreecommitdiff
path: root/app/graphql/resolvers/project_pipeline_resolver.rb
blob: ce4b6ac6b0ccc91a39565f558ad715857d0c1aae (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
# frozen_string_literal: true

module Resolvers
  class ProjectPipelineResolver < BaseResolver
    type ::Types::Ci::PipelineType, null: true

    alias_method :project, :object

    argument :iid, GraphQL::Types::ID,
             required: false,
             description: 'IID of the Pipeline. For example, "1".'

    argument :sha, GraphQL::Types::String,
             required: false,
             description: 'SHA of the Pipeline. For example, "dyd0f15ay83993f5ab66k927w28673882x99100b".'

    def ready?(iid: nil, sha: nil)
      unless iid.present? ^ sha.present?
        raise Gitlab::Graphql::Errors::ArgumentError, 'Provide one of an IID or SHA'
      end

      super
    end

    def resolve(iid: nil, sha: nil)
      if iid
        BatchLoader::GraphQL.for(iid).batch(key: project) do |iids, loader, args|
          finder = ::Ci::PipelinesFinder.new(project, current_user, iids: iids)

          finder.execute.each { |pipeline| loader.call(pipeline.iid.to_s, pipeline) }
        end
      else
        BatchLoader::GraphQL.for(sha).batch(key: project) do |shas, loader, args|
          finder = ::Ci::PipelinesFinder.new(project, current_user, sha: shas)

          finder.execute.each { |pipeline| loader.call(pipeline.sha.to_s, pipeline) }
        end
      end
    end
  end
end