akkoma/lib/mix/tasks/pleroma/refresh_counter_cache.ex

70 lines
1.6 KiB
Elixir
Raw Normal View History

# Pleroma: A lightweight social networking server
# Copyright © 2017-2020 Pleroma Authors <https://pleroma.social/>
# SPDX-License-Identifier: AGPL-3.0-only
defmodule Mix.Tasks.Pleroma.RefreshCounterCache do
@shortdoc "Refreshes counter cache"
use Mix.Task
alias Pleroma.Activity
alias Pleroma.CounterCache
alias Pleroma.Repo
require Logger
import Ecto.Query
def run([]) do
Mix.Pleroma.start_pleroma()
instances =
Activity
|> distinct([a], true)
|> select([a], fragment("split_part(?, '/', 3)", a.actor))
|> Repo.all()
instances
|> Enum.with_index(1)
|> Enum.each(fn {instance, i} ->
2020-05-08 22:20:50 +00:00
counters = instance_counters(instance)
CounterCache.set(instance, counters)
Mix.Pleroma.shell_info(
"[#{i}/#{length(instances)}] Setting #{instance} counters: #{inspect(counters)}"
)
end)
Mix.Pleroma.shell_info("Done")
end
2020-05-08 22:20:50 +00:00
defp instance_counters(instance) do
counters = %{"public" => 0, "unlisted" => 0, "private" => 0, "direct" => 0}
Activity
2020-05-08 22:20:50 +00:00
|> where([a], fragment("(? ->> 'type'::text) = 'Create'", a.data))
2020-05-09 10:13:26 +00:00
|> where([a], fragment("split_part(?, '/', 3) = ?", a.actor, ^instance))
2020-05-08 22:20:50 +00:00
|> select(
[a],
{fragment(
"activity_visibility(?, ?, ?)",
a.actor,
a.recipients,
a.data
), count(a.id)}
)
|> group_by(
[a],
fragment(
2020-05-08 22:20:50 +00:00
"activity_visibility(?, ?, ?)",
a.actor,
a.recipients,
2020-05-08 22:20:50 +00:00
a.data
)
)
2020-05-08 22:20:50 +00:00
|> Repo.all(timeout: :timer.minutes(30))
|> Enum.reduce(counters, fn {visibility, count}, acc ->
Map.put(acc, visibility, count)
end)
end
end