app/graphql/resolvers/ci/runner_resolver.rb


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

# frozen_string_literal: true

module Resolvers
  module Ci
    class RunnerResolver < BaseResolver
      include LooksAhead

      type Types::Ci::RunnerType, null: true
      extras [:lookahead]
      description 'Runner information.'

      argument :id,
               type: ::Types::GlobalIDType[::Ci::Runner],
               required: true,
               description: 'Runner ID.'

      def resolve_with_lookahead(id:)
        find_runner(id: id)
      end

      private

      def find_runner(id:)
        runner_id = GitlabSchema.parse_gid(id, expected_type: ::Ci::Runner).model_id.to_i
        preload_tag_list = lookahead.selects?(:tag_list)

        BatchLoader::GraphQL.for(runner_id).batch(key: { preload_tag_list: preload_tag_list }) do |ids, loader, batch|
          results = ::Ci::Runner.id_in(ids)
          results = results.with_tags if batch[:key][:preload_tag_list]

          results.each { |record| loader.call(record.id, record) }
        end
      end
    end
  end
end