Current section

Files

Jump to
thousand_island lib thousand_island acceptor.ex
Raw

lib/thousand_island/acceptor.ex

defmodule ThousandIsland.Acceptor do
@moduledoc false
use Task, restart: :transient
alias ThousandIsland.{AcceptorSupervisor, Connection, Listener, Server, ServerConfig}
def start_link(arg) do
Task.start_link(__MODULE__, :run, [arg])
end
def run({server_pid, parent_pid, %ServerConfig{} = config}) do
listener_pid = Server.listener_pid(server_pid)
{:ok, listener_socket} = Listener.listener_socket(listener_pid)
acceptor_info = %{
acceptor_id: unique_id(),
listener_socket: listener_socket,
connection_sup_pid: AcceptorSupervisor.connection_sup_pid(parent_pid),
server_config: config
}
telemetry(:start, %{}, acceptor_info)
accept(acceptor_info)
end
defp accept(
%{
acceptor_id: acceptor_id,
listener_socket: listener_socket,
connection_sup_pid: connection_sup_pid,
server_config: %ServerConfig{transport_module: transport_module} = server_config
} = acceptor_info
) do
start = System.monotonic_time()
case transport_module.accept(listener_socket) do
{:ok, socket} ->
wakeup = System.monotonic_time()
Connection.start(connection_sup_pid, socket, acceptor_id, server_config)
complete = System.monotonic_time()
wait_time = wakeup - start
startup_time = complete - wakeup
telemetry(:accept, %{wait_time: wait_time, startup_time: startup_time}, acceptor_info)
accept(acceptor_info)
{:error, reason} ->
wakeup = System.monotonic_time()
wait_time = wakeup - start
telemetry(:shutdown, %{wait_time: wait_time, shutdown_reason: reason}, acceptor_info)
# Reasons other than closed are unexpected, so raise up and let our supervisor take care of things
if !(reason in [:closed]) do
raise "Unexpected error in accept: #{inspect(reason)}"
end
end
end
defp telemetry(subevent, measurement, acceptor_info) do
acceptor_info = Map.take(acceptor_info, [:acceptor_id, :server_config])
:telemetry.execute([:acceptor] ++ [subevent], measurement, acceptor_info)
end
defp unique_id, do: Base.encode16(:crypto.strong_rand_bytes(6))
end