# Pleroma: A lightweight social networking server # Copyright © 2017-2020 Pleroma Authors <https://pleroma.social/> # SPDX-License-Identifier: AGPL-3.0-only defmodule Pleroma.Stats do import Ecto.Query alias Pleroma.CounterCache alias Pleroma.Repo alias Pleroma.User use GenServer def start_link(_) do GenServer.start_link( __MODULE__, nil, name: __MODULE__ ) end @doc "Performs update stats" def force_update do GenServer.call(__MODULE__, :force_update) end @doc "Performs collect stats" def do_collect do GenServer.cast(__MODULE__, :run_update) end @doc "Returns stats data" @spec get_stats() :: %{domain_count: integer(), status_count: integer(), user_count: integer()} def get_stats do %{stats: stats} = GenServer.call(__MODULE__, :get_state) stats end @doc "Returns list peers" @spec get_peers() :: list(String.t()) def get_peers do %{peers: peers} = GenServer.call(__MODULE__, :get_state) peers end def init(_args) do {:ok, get_stat_data()} end def handle_call(:force_update, _from, _state) do new_stats = get_stat_data() {:reply, new_stats, new_stats} end def handle_call(:get_state, _from, state) do {:reply, state, state} end def handle_cast(:run_update, _state) do new_stats = get_stat_data() {:noreply, new_stats} end defp get_stat_data do peers = from( u in User, select: fragment("distinct split_part(?, '@', 2)", u.nickname), where: u.local != ^true ) |> Repo.all() |> Enum.filter(& &1) domain_count = Enum.count(peers) status_count = Repo.aggregate(User.Query.build(%{local: true}), :sum, :note_count) user_count = Repo.aggregate(User.Query.build(%{local: true, active: true}), :count, :id) %{ peers: peers, stats: %{ domain_count: domain_count, status_count: status_count, user_count: user_count } } end def get_status_visibility_count do counter_cache = CounterCache.get_as_map([ "status_visibility_public", "status_visibility_private", "status_visibility_unlisted", "status_visibility_direct" ]) %{ public: counter_cache["status_visibility_public"] || 0, unlisted: counter_cache["status_visibility_unlisted"] || 0, private: counter_cache["status_visibility_private"] || 0, direct: counter_cache["status_visibility_direct"] || 0 } end end