app/graphql/types/ci/job_need_union.rb


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21

# frozen_string_literal: true

module Types
  module Ci
    class JobNeedUnion < GraphQL::Schema::Union
      TypeNotSupportedError = Class.new(StandardError)

      possible_types Types::Ci::JobType, Types::Ci::BuildNeedType

      def self.resolve_type(object, context)
        if object.is_a?(::Ci::BuildNeed)
          Types::Ci::BuildNeedType
        elsif object.is_a?(CommitStatus)
          Types::Ci::JobType
        else
          raise TypeNotSupportedError
        end
      end
    end
  end
end