File: user_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 (45 lines) | stat: -rw-r--r-- 1,032 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
35
36
37
38
39
40
41
42
43
44
45
# frozen_string_literal: true

module Resolvers
  class UserResolver < BaseResolver
    include Gitlab::Graphql::Authorize::AuthorizeResource

    description 'Retrieve a single user'

    type Types::UserType, null: true

    argument :id, Types::GlobalIDType[User],
      required: false,
      description: 'ID of the User.'

    argument :username, GraphQL::Types::String,
      required: false,
      description: 'Username of the User.'

    validates exactly_one_of: [:id, :username]

    def resolve(id: nil, username: nil)
      authorize!

      if id
        GitlabSchema.object_from_id(id, expected_type: User)
      else
        batch_load(username)
      end
    end

    private

    def batch_load(username)
      BatchLoader::GraphQL.for(username).batch do |usernames, loader|
        User.by_username(usernames).each do |user|
          loader.call(username, user)
        end
      end
    end

    def authorize!
      raise_resource_not_available_error! unless context[:current_user].present?
    end
  end
end