limepotato/akkoma
1
0
Fork 0
Code Pull requests Activity

Use :ets.match_delete to delete old captchas

Browse source
This commit is contained in:
vaartis 2018-12-17 17:19:28 +03:00 committed by Ekaterina Vaartis
parent 6e2f64a0a6
commit 35522fef09
1 changed files with 19 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

27
lib/pleroma/captcha/kocaptcha.ex
View file

@ -19,7 +19,11 @@ defmodule Pleroma.Captcha.Kocaptcha do
token = json_resp["token"] token = json_resp["token"]
true = :ets.insert(@ets, {token, json_resp["md5"], DateTime.now_utc()}) true =
:ets.insert(
@ets,
{token, json_resp["md5"], DateTime.now_utc() |> DateTime.Format.unix()}
)
%{type: :kocaptcha, token: token, url: endpoint <> json_resp["url"]} %{type: :kocaptcha, token: token, url: endpoint <> json_resp["url"]}
end end
@ -42,14 +46,21 @@ defmodule Pleroma.Captcha.Kocaptcha do
@impl Service @impl Service
def cleanup() do def cleanup() do
seconds_retained = Pleroma.Config.get!([Pleroma.Captcha, :seconds_retained]) seconds_retained = Pleroma.Config.get!([Pleroma.Captcha, :seconds_retained])
# If the time in ETS is less than current_time - seconds_retained, then the time has
# already passed
delete_after =
DateTime.subtract!(DateTime.now_utc(), seconds_retained) |> DateTime.Format.unix()
# Go through captchas and remove expired ones :ets.select_delete(
:ets.tab2list(@ets) @ets,
|> Enum.each(fn {token, _, time_inserted} -> [
# time created + expiration time = time when the captcha should be removed {
remove_time = DateTime.add!(time_inserted, seconds_retained) {:_, :_, :"$1"},
if DateTime.after?(DateTime.now_utc(), remove_time), do: :ets.delete(@ets, token) [{:<, :"$1", {:const, delete_after}}],
end) [true]
}
]
)
:ok :ok
end end