Reduce amount of subscription data stored in Elixir registry

absinthe-graphql · Dec 15, 2020 · aa4760e · aa4760e
1 parent 72ff9f9
commit aa4760e
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 5 deletions.
diff --git a/lib/absinthe/phase/subscription/subscribe_self.ex b/lib/absinthe/phase/subscription/subscribe_self.ex
@@ -26,7 +26,7 @@ defmodule Absinthe.Phase.Subscription.SubscribeSelf do
       subscription_id = get_subscription_id(config, blueprint, options)
 
       for field_key <- field_keys,
-          do: Absinthe.Subscription.subscribe(pubsub, field_key, subscription_id, blueprint)
+          do: Absinthe.Subscription.subscribe(pubsub, field_key, subscription_id, blueprint, options)
 
       {:replace, blueprint,
        [

diff --git a/lib/absinthe/subscription.ex b/lib/absinthe/subscription.ex
@@ -106,14 +106,15 @@ defmodule Absinthe.Subscription do
   defp fetch_fields(_, _), do: []
 
   @doc false
-  def subscribe(pubsub, field_key, doc_id, doc) do
+  def subscribe(pubsub, field_key, doc_id, doc, options) do
     registry = pubsub |> registry_name
 
     doc_value = {
       doc_id,
       %{
-        initial_phases: doc.initial_phases,
-        source: doc.source
+        schema: doc.schema,
+        source: doc.source,
+        options: options
       }
     }
 

diff --git a/lib/absinthe/subscription/local.ex b/lib/absinthe/subscription/local.ex
@@ -37,8 +37,10 @@ defmodule Absinthe.Subscription.Local do
   defp run_docset(pubsub, docs_and_topics, mutation_result) do
     for {topic, key_strategy, doc} <- docs_and_topics do
       try do
+        pipeline = Absinthe.Pipeline.for_document(doc.schema, doc.options)
+
         pipeline =
-          doc.initial_phases
+          pipeline
           |> Pipeline.replace(
             Phase.Telemetry,
             {Phase.Telemetry, event: [:subscription, :publish, :start]}