commit: 959dc6e6fc95b33700fb7e08689afb701b2877f2
parent 99fd9c5e38ad08973f435f1a67d6af60d004c578
Author: Mark Felder <feld@feld.me>
Date: Thu, 22 Apr 2021 10:11:08 -0500
Cleanup and ensure we obey custom Repo timeout
Diffstat:
1 file changed, 10 insertions(+), 1 deletion(-)
diff --git a/lib/pleroma/utils.ex b/lib/pleroma/utils.ex
@@ -11,6 +11,8 @@ defmodule Pleroma.Utils do
eperm epipe erange erofs espipe esrch estale etxtbsy exdev
)a
+ @repo_timeout Pleroma.Config.get([Pleroma.Repo, :timeout], 15_000)
+
def compile_dir(dir) when is_binary(dir) do
dir
|> File.ls!()
@@ -64,13 +66,20 @@ defmodule Pleroma.Utils do
def posix_error_message(_), do: ""
+ @doc """
+ Returns [timeout: integer] suitable for passing as an option to Repo functions.
+
+ This function detects if the execution was triggered from IEx shell, Mix task, or
+ ./bin/pleroma_ctl and sets the timeout to :infinity, else returns the default timeout value.
+ """
+ @spec query_timeout() :: [timeout: integer]
def query_timeout do
{parent, _, _, _} = Process.info(self(), :current_stacktrace) |> elem(1) |> Enum.fetch!(2)
cond do
parent |> to_string |> String.starts_with?("Elixir.Mix.Task") -> [timeout: :infinity]
parent == :erl_eval -> [timeout: :infinity]
- true -> [timeout: 15_000]
+ true -> [timeout: @repo_timeout]
end
end
end