refresh_counter_cache.ex
No OneTemporary
Actions

Size

1 KB

Referenced Files

None

Subscribers

None

refresh_counter_cache.ex
View Options

	# Pleroma: A lightweight social networking server
	# Copyright © 2017-2022 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} ->
	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

	defp instance_counters(instance) do
	counters = %{"public" => 0, "unlisted" => 0, "private" => 0, "direct" => 0}

	Activity
	\|> where([a], fragment("(? ->> 'type'::text) = 'Create'", a.data))
	\|> where([a], fragment("split_part(?, '/', 3) = ?", a.actor, ^instance))
	\|> select(
	[a],
	{fragment(
	"activity_visibility(?, ?, ?)",
	a.actor,
	a.recipients,
	a.data
	), count(a.id)}
	)
	\|> group_by(
	[a],
	fragment(
	"activity_visibility(?, ?, ?)",
	a.actor,
	a.recipients,
	a.data
	)
	)
	\|> Repo.all(timeout: :timer.minutes(30))
	\|> Enum.reduce(counters, fn {visibility, count}, acc ->
	Map.put(acc, visibility, count)
	end)
	end
	end