forked from AkkomaGang/akkoma
58a4f350a8
This patch refactors gun pooling to use Elixir process registry and simplifies adapter option insertion. Having the pool use process registry instead of a GenServer has a number of advantages: - Simpler code: the initial implementation adds about half the lines of code it deletes - Concurrency: unlike a GenServer, ETS-based registry can handle multiple checkout/checkin requests at the same time - Precise and easy idle connection clousure: current proposal for closing idle connections in the GenServer-based pool needs to filter through all connections once a minute and compare their last active time with closing time. With Elixir process registry this can be done by just using `Process.send_after`/`Process.cancel_timer` in the worker process. - Lower memory footprint: In my tests `gun-memory-leak` branch uses about 290mb on peak load (250 connections) and 235mb on idle (5-10 connections). Registry-based pool uses 210mb on idle and 240mb on peak load
98 lines
2.9 KiB
Elixir
98 lines
2.9 KiB
Elixir
# Pleroma: A lightweight social networking server
|
|
# Copyright © 2017-2020 Pleroma Authors <https://pleroma.social/>
|
|
# SPDX-License-Identifier: AGPL-3.0-only
|
|
|
|
defmodule Pleroma.HTTP do
|
|
@moduledoc """
|
|
Wrapper for `Tesla.request/2`.
|
|
"""
|
|
|
|
alias Pleroma.HTTP.AdapterHelper
|
|
alias Pleroma.HTTP.Request
|
|
alias Pleroma.HTTP.RequestBuilder, as: Builder
|
|
alias Tesla.Client
|
|
alias Tesla.Env
|
|
|
|
require Logger
|
|
|
|
@type t :: __MODULE__
|
|
@type method() :: :get | :post | :put | :delete | :head
|
|
|
|
@doc """
|
|
Performs GET request.
|
|
|
|
See `Pleroma.HTTP.request/5`
|
|
"""
|
|
@spec get(Request.url() | nil, Request.headers(), keyword()) ::
|
|
nil | {:ok, Env.t()} | {:error, any()}
|
|
def get(url, headers \\ [], options \\ [])
|
|
def get(nil, _, _), do: nil
|
|
def get(url, headers, options), do: request(:get, url, "", headers, options)
|
|
|
|
@spec head(Request.url(), Request.headers(), keyword()) :: {:ok, Env.t()} | {:error, any()}
|
|
def head(url, headers \\ [], options \\ []), do: request(:head, url, "", headers, options)
|
|
|
|
@doc """
|
|
Performs POST request.
|
|
|
|
See `Pleroma.HTTP.request/5`
|
|
"""
|
|
@spec post(Request.url(), String.t(), Request.headers(), keyword()) ::
|
|
{:ok, Env.t()} | {:error, any()}
|
|
def post(url, body, headers \\ [], options \\ []),
|
|
do: request(:post, url, body, headers, options)
|
|
|
|
@doc """
|
|
Builds and performs http request.
|
|
|
|
# Arguments:
|
|
`method` - :get, :post, :put, :delete, :head
|
|
`url` - full url
|
|
`body` - request body
|
|
`headers` - a keyworld list of headers, e.g. `[{"content-type", "text/plain"}]`
|
|
`options` - custom, per-request middleware or adapter options
|
|
|
|
# Returns:
|
|
`{:ok, %Tesla.Env{}}` or `{:error, error}`
|
|
|
|
"""
|
|
@spec request(method(), Request.url(), String.t(), Request.headers(), keyword()) ::
|
|
{:ok, Env.t()} | {:error, any()}
|
|
def request(method, url, body, headers, options) when is_binary(url) do
|
|
uri = URI.parse(url)
|
|
adapter_opts = AdapterHelper.options(uri, options[:adapter] || [])
|
|
|
|
case AdapterHelper.get_conn(uri, adapter_opts) do
|
|
{:ok, adapter_opts} ->
|
|
options = put_in(options[:adapter], adapter_opts)
|
|
params = options[:params] || []
|
|
request = build_request(method, headers, options, url, body, params)
|
|
|
|
adapter = Application.get_env(:tesla, :adapter)
|
|
client = Tesla.client([Tesla.Middleware.FollowRedirects], adapter)
|
|
|
|
response = request(client, request)
|
|
|
|
AdapterHelper.after_request(adapter_opts)
|
|
|
|
response
|
|
|
|
err ->
|
|
err
|
|
end
|
|
end
|
|
|
|
@spec request(Client.t(), keyword()) :: {:ok, Env.t()} | {:error, any()}
|
|
def request(client, request), do: Tesla.request(client, request)
|
|
|
|
defp build_request(method, headers, options, url, body, params) do
|
|
Builder.new()
|
|
|> Builder.method(method)
|
|
|> Builder.headers(headers)
|
|
|> Builder.opts(options)
|
|
|> Builder.url(url)
|
|
|> Builder.add_param(:body, :body, body)
|
|
|> Builder.add_param(:query, :query, params)
|
|
|> Builder.convert_to_keyword()
|
|
end
|
|
end
|