File: runner_managers_resolver.rb

package info (click to toggle)
gitlab 17.6.5-19
  • links: PTS, VCS
  • area: main
  • in suites: sid
  • size: 629,368 kB
  • sloc: ruby: 1,915,304; javascript: 557,307; sql: 60,639; xml: 6,509; sh: 4,567; makefile: 1,239; python: 406
file content (34 lines) | stat: -rw-r--r-- 1,153 bytes parent folder | download
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
# frozen_string_literal: true

module Resolvers
  module Ci
    class RunnerManagersResolver < BaseResolver
      type Types::Ci::RunnerManagerType.connection_type, null: true

      argument :system_id, ::GraphQL::Types::String,
        required: false,
        description: 'Filter runner managers by system ID.'

      argument :status, ::Types::Ci::RunnerStatusEnum,
        required: false,
        description: 'Filter runner managers by status.'

      def resolve(**args)
        BatchLoader::GraphQL.for(object.id).batch(key: args[:system_id]) do |runner_ids, loader|
          runner_managers =
            ::Ci::RunnerManagersFinder
              .new(runner: runner_ids, params: args.slice(:system_id, :status))
              .execute
          ::Preloaders::RunnerManagerPolicyPreloader.new(runner_managers, current_user).execute

          runner_managers_by_runner_id = runner_managers.group_by(&:runner_id)

          runner_ids.each do |runner_id|
            runner_managers = Array.wrap(runner_managers_by_runner_id[runner_id])
            loader.call(runner_id, runner_managers)
          end
        end
      end
    end
  end
end