parent
affc910372
commit
9d9c26b833
@ -1,29 +0,0 @@
|
|||||||
# Pleroma: A lightweight social networking server
|
|
||||||
# Copyright © 2017-2021 Pleroma Authors <https://pleroma.social/>
|
|
||||||
# SPDX-License-Identifier: AGPL-3.0-only
|
|
||||||
|
|
||||||
defmodule Pleroma.Gun do
|
|
||||||
@callback open(charlist(), pos_integer(), map()) :: {:ok, pid()}
|
|
||||||
@callback info(pid()) :: map()
|
|
||||||
@callback close(pid()) :: :ok
|
|
||||||
@callback await_up(pid, pos_integer()) :: {:ok, atom()} | {:error, atom()}
|
|
||||||
@callback connect(pid(), map()) :: reference()
|
|
||||||
@callback await(pid(), reference()) :: {:response, :fin, 200, []}
|
|
||||||
@callback set_owner(pid(), pid()) :: :ok
|
|
||||||
|
|
||||||
defp api, do: Pleroma.Config.get([Pleroma.Gun], Pleroma.Gun.API)
|
|
||||||
|
|
||||||
def open(host, port, opts), do: api().open(host, port, opts)
|
|
||||||
|
|
||||||
def info(pid), do: api().info(pid)
|
|
||||||
|
|
||||||
def close(pid), do: api().close(pid)
|
|
||||||
|
|
||||||
def await_up(pid, timeout \\ 5_000), do: api().await_up(pid, timeout)
|
|
||||||
|
|
||||||
def connect(pid, opts), do: api().connect(pid, opts)
|
|
||||||
|
|
||||||
def await(pid, ref), do: api().await(pid, ref)
|
|
||||||
|
|
||||||
def set_owner(pid, owner), do: api().set_owner(pid, owner)
|
|
||||||
end
|
|
@ -1,50 +0,0 @@
|
|||||||
# Pleroma: A lightweight social networking server
|
|
||||||
# Copyright © 2017-2021 Pleroma Authors <https://pleroma.social/>
|
|
||||||
# SPDX-License-Identifier: AGPL-3.0-only
|
|
||||||
|
|
||||||
defmodule Pleroma.Tesla.Middleware.ConnectionPool do
|
|
||||||
@moduledoc """
|
|
||||||
Middleware to get/release connections from `Pleroma.Gun.ConnectionPool`
|
|
||||||
"""
|
|
||||||
|
|
||||||
@behaviour Tesla.Middleware
|
|
||||||
|
|
||||||
alias Pleroma.Gun.ConnectionPool
|
|
||||||
|
|
||||||
@impl Tesla.Middleware
|
|
||||||
def call(%Tesla.Env{url: url, opts: opts} = env, next, _) do
|
|
||||||
uri = URI.parse(url)
|
|
||||||
|
|
||||||
# Avoid leaking connections when the middleware is called twice
|
|
||||||
# with body_as: :chunks. We assume only the middleware can set
|
|
||||||
# opts[:adapter][:conn]
|
|
||||||
if opts[:adapter][:conn] do
|
|
||||||
ConnectionPool.release_conn(opts[:adapter][:conn])
|
|
||||||
end
|
|
||||||
|
|
||||||
case ConnectionPool.get_conn(uri, opts[:adapter]) do
|
|
||||||
{:ok, conn_pid} ->
|
|
||||||
adapter_opts = Keyword.merge(opts[:adapter], conn: conn_pid, close_conn: false)
|
|
||||||
opts = Keyword.put(opts, :adapter, adapter_opts)
|
|
||||||
env = %{env | opts: opts}
|
|
||||||
|
|
||||||
case Tesla.run(env, next) do
|
|
||||||
{:ok, env} ->
|
|
||||||
unless opts[:adapter][:body_as] == :chunks do
|
|
||||||
ConnectionPool.release_conn(conn_pid)
|
|
||||||
{_, res} = pop_in(env.opts[:adapter][:conn])
|
|
||||||
{:ok, res}
|
|
||||||
else
|
|
||||||
{:ok, env}
|
|
||||||
end
|
|
||||||
|
|
||||||
err ->
|
|
||||||
ConnectionPool.release_conn(conn_pid)
|
|
||||||
err
|
|
||||||
end
|
|
||||||
|
|
||||||
err ->
|
|
||||||
err
|
|
||||||
end
|
|
||||||
end
|
|
||||||
end
|
|
Loading…
Reference in new issue