custom JSON library support

config/config.exs

@@ -1,5 +1,7 @@
 import Config
 
+config :absinthe_graphql_ws, :json_library, Jason
+
 if config_env() == :test do
   import_config("test.exs")
 end

lib/absinthe/graphql_ws/message/complete.ex

@@ -5,7 +5,9 @@ defmodule Absinthe.GraphqlWS.Message.Complete do
   message, no Complete message will be emitted.
   """
 
+  alias Absinthe.GraphqlWS.Util
+
   def new(id, payload \\ %{}) do
-    Jason.encode!(%{id: id, type: "complete", payload: payload})
+    Util.json_library().encode!(%{id: id, type: "complete", payload: payload})
   end
 end

lib/absinthe/graphql_ws/message/connection_ack.ex

@@ -3,7 +3,9 @@ defmodule Absinthe.GraphqlWS.Message.ConnectionAck do
   Must be sent in response to a ConnectionInit message from the client.
   """
 
+  alias Absinthe.GraphqlWS.Util
+
   def new(payload \\ %{}) do
-    Jason.encode!(%{type: "connection_ack", payload: payload})
+    Util.json_library().encode!(%{type: "connection_ack", payload: payload})
   end
 end

lib/absinthe/graphql_ws/message/error.ex

@@ -4,7 +4,9 @@ defmodule Absinthe.GraphqlWS.Message.Error do
   the actual execution, usually due to validation errors.
   """
 
+  alias Absinthe.GraphqlWS.Util
+
   def new(id, payload \\ %{}) do
-    Jason.encode!(%{id: id, type: "error", payload: payload})
+    Util.json_library().encode!(%{id: id, type: "error", payload: payload})
   end
 end

lib/absinthe/graphql_ws/message/next.ex

@@ -5,7 +5,9 @@ defmodule Absinthe.GraphqlWS.Message.Next do
   will follow indicating stream completion.
   """
 
+  alias Absinthe.GraphqlWS.Util
+
   def new(id, payload \\ %{}) do
-    Jason.encode!(%{id: id, type: "next", payload: payload})
+    Util.json_library().encode!(%{id: id, type: "next", payload: payload})
   end
 end

lib/absinthe/graphql_ws/message/pong.ex

@@ -3,7 +3,9 @@ defmodule Absinthe.GraphqlWS.Message.Pong do
   Reply to a Ping message.
   """
 
+  alias Absinthe.GraphqlWS.Util
+
   def new(payload \\ %{}) do
-    Jason.encode!(%{type: "pong", payload: payload})
+    Util.json_library().encode!(%{type: "pong", payload: payload})
   end
 end

lib/absinthe/graphql_ws/transport.ex

@@ -11,8 +11,7 @@ defmodule Absinthe.GraphqlWS.Transport do
   their codebase, but is documented to help understand the intentions of the code.
   """
 
-  alias Absinthe.GraphqlWS.Message
-  alias Absinthe.GraphqlWS.Socket
+  alias Absinthe.GraphqlWS.{Message, Socket, Util}
   alias Phoenix.Socket.Broadcast
   require Logger
 
@@ -48,7 +47,7 @@ defmodule Absinthe.GraphqlWS.Transport do
   """
   @spec handle_in({binary(), [opcode: :text]}, socket()) :: reply_inbound()
   def handle_in({text, [opcode: :text]}, socket) do
-    Jason.decode(text)
+    Util.json_library().decode(text)
     |> case do
       {:ok, json} ->
         handle_inbound(json, socket)

lib/absinthe/graphql_ws/util.ex

@@ -62,4 +62,9 @@ defmodule Absinthe.GraphqlWS.Util do
     absinthe = %{socket.absinthe | opts: opts}
     %{socket | absinthe: absinthe}
   end
+
+  @doc """
+  Retrieves the JSON library module used to encode and decode transport messages
+  """
+  def json_library, do: Application.get_env(:absinthe_graphql_ws, :json_library)
 end