Use PartitionSupervisor (#6)

lib/kino_proxy/application.ex

@@ -3,9 +3,10 @@ defmodule KinoProxy.Application do
 
  @impl true
  def start(_type, _args) do
- children = [KinoProxy.Client]
- opts = [strategy: :one_for_one, name: KinoProxy.Supervisor]
+ children = [
+ {PartitionSupervisor, child_spec: KinoProxy.Client, name: KinoProxy.PartitionSupervisor}
+ ]
 
- Supervisor.start_link(children, opts)
+ Supervisor.start_link(children, strategy: :one_for_one, name: KinoProxy.Supervisor)
  end
 end

lib/kino_proxy/client.ex

@@ -9,14 +9,17 @@ defmodule KinoProxy.Client do
  # by the owner of the HTTP request.
 
  use GenServer
- @name Kino.Proxy
 
  def start_link(opts \\ []) do
- GenServer.start_link(__MODULE__, opts, name: @name)
+ GenServer.start_link(__MODULE__, opts)
  end
 
  def listen(fun) when is_function(fun, 1) do
- GenServer.cast(@name, {:listen, fun})
+ GenServer.cast(partition_name(), {:listen, fun})
+ end
+
+ def get_pid() do
+ GenServer.whereis(partition_name())
  end
 
  # GenServer callbacks
@@ -43,4 +46,8 @@ defmodule KinoProxy.Client do
  def handle_cast({:listen, fun}, state) do
  {:noreply, %{state | fun: fun}}
  end
+
+ defp partition_name() do
+ {:via, PartitionSupervisor, {KinoProxy.PartitionSupervisor, Kino.Proxy}}
+ end
 end

test/support/endpoint.ex

@@ -13,7 +13,7 @@ defmodule KinoProxy.Endpoint do
  script_name = [id, "proxy"]
  conn = %{conn | path_info: path_info, script_name: conn.script_name ++ script_name}
 
- if pid = GenServer.whereis(Kino.Proxy) do
+ if pid = KinoProxy.Client.get_pid() do
  {conn, _reason} = Kino.Proxy.serve(pid, conn)
  conn
  else