X-Git-Url: http://git.squeep.com/?a=blobdiff_plain;f=lib%2Fpleroma%2Fweb%2Foauth%2Ftoken%2Fclean_worker.ex;h=f639f9c6fd202c38ea753c9336192700df93ddc2;hb=527afb813af6c64337d02ddf1a2f159fe557acbc;hp=dca85244931d5cd2c664ebb41f3bd9c305b113f9;hpb=620908a2db86942a00bc0ba9c71c037061e26967;p=akkoma

diff --git a/lib/pleroma/web/oauth/token/clean_worker.ex b/lib/pleroma/web/oauth/token/clean_worker.ex
index dca852449..e3aa4eb7e 100644
--- a/lib/pleroma/web/oauth/token/clean_worker.ex
+++ b/lib/pleroma/web/oauth/token/clean_worker.ex
@@ -1,41 +1,38 @@
 # Pleroma: A lightweight social networking server
-# Copyright © 2017-2018 Pleroma Authors <https://pleroma.social/>
+# Copyright © 2017-2020 Pleroma Authors <https://pleroma.social/>
 # SPDX-License-Identifier: AGPL-3.0-only
 
 defmodule Pleroma.Web.OAuth.Token.CleanWorker do
   @moduledoc """
-  The module represents functions to clean an expired oauth tokens.
+  The module represents functions to clean an expired OAuth and MFA tokens.
   """
+  use GenServer
 
-  # 10 seconds
-  @start_interval 10_000
-  @interval Pleroma.Config.get(
-              # 24 hours
-              [:oauth2, :clean_expired_tokens_interval],
-              86_400_000
-            )
-  @queue :background
+  @ten_seconds 10_000
+  @one_day 86_400_000
 
-  alias Pleroma.Web.OAuth.Token
+  alias Pleroma.MFA
+  alias Pleroma.Web.OAuth
+  alias Pleroma.Workers.BackgroundWorker
 
-  def start_link, do: GenServer.start_link(__MODULE__, nil)
+  def start_link(_), do: GenServer.start_link(__MODULE__, %{})
 
   def init(_) do
-    if Pleroma.Config.get([:oauth2, :clean_expired_tokens], false) do
-      Process.send_after(self(), :perform, @start_interval)
-      {:ok, nil}
-    else
-      :ignore
-    end
+    Process.send_after(self(), :perform, @ten_seconds)
+    {:ok, nil}
   end
 
   @doc false
   def handle_info(:perform, state) do
-    Process.send_after(self(), :perform, @interval)
-    PleromaJobQueue.enqueue(@queue, __MODULE__, [:clean])
+    BackgroundWorker.enqueue("clean_expired_tokens", %{})
+    interval = Pleroma.Config.get([:oauth2, :clean_expired_tokens_interval], @one_day)
+
+    Process.send_after(self(), :perform, interval)
     {:noreply, state}
   end
 
-  # Job Worker Callbacks
-  def perform(:clean), do: Token.delete_expired_tokens()
+  def perform(:clean) do
+    OAuth.Token.delete_expired_tokens()
+    MFA.Token.delete_expired_tokens()
+  end
 end