Contents

# frozen_string_literal: true
class GraphqlChannel < ActionCable::Channel::Base
  QueryType = GraphQL::ObjectType.define do
    name "Query"
    field :value, types.Int, resolve: Proc.new { 3 }
  end

  SubscriptionType = GraphQL::ObjectType.define do
    name "Subscription"
    field :payload, PayloadType do
      argument :id, !types.ID
    end
  end

  PayloadType = GraphQL::ObjectType.define do
    name "Payload"
    field :value, types.Int
  end

  GraphQLSchema = GraphQL::Schema.define do
    query(QueryType)
    subscription(SubscriptionType)
    use GraphQL::Subscriptions::ActionCableSubscriptions
  end

  def subscribed
    @subscription_ids = []
  end

  def execute(data)
    query = data["query"]
    variables = data["variables"] || {}
    operation_name = data["operationName"]
    context = {
      # Make sure the channel is in the context
      channel: self,
    }

    result = GraphQLSchema.execute({
      query: query,
      context: context,
      variables: variables,
      operation_name: operation_name
    })

    payload = {
      result: result.to_h,
      more: result.subscription?,
    }

    # Track the subscription here so we can remove it
    # on unsubscribe.
    if result.context[:subscription_id]
      @subscription_ids << result.context[:subscription_id]
    end

    transmit(payload)
  end

  def make_trigger(data)
    GraphQLSchema.subscriptions.trigger("payload", {"id" => data["id"]}, ExamplePayload.new(data["value"]))
  end

  def unsubscribed
    @subscription_ids.each { |sid|
      GraphQLSchema.subscriptions.delete_subscription(sid)
    }
  end

  # This is to make sure that GlobalID is used to load and dump this object
  class ExamplePayload
    include GlobalID::Identification
    def initialize(value)
      @value = value
    end

    def self.find(value)
      self.new(value)
    end

    attr_reader :value
    alias :id :value
  end
end

Version data entries

4 entries across 4 versions & 1 rubygems

Version	Path
graphql-1.8.0.pre1	spec/dummy/app/channels/graphql_channel.rb
graphql-1.7.6	spec/dummy/app/channels/graphql_channel.rb
graphql-1.7.5	spec/dummy/app/channels/graphql_channel.rb
graphql-1.7.4	spec/dummy/app/channels/graphql_channel.rb