git.squeep.com Git - akkoma/blob - lib/pleroma/stats.ex

   1 # Pleroma: A lightweight social networking server
   2 # Copyright © 2017-2019 Pleroma Authors <https://pleroma.social/>
   3 # SPDX-License-Identifier: AGPL-3.0-only
   4
   5 defmodule Pleroma.Stats do
   6   import Ecto.Query
   7   alias Pleroma.{User, Repo}
   8
   9   def start_link do
  10     agent = Agent.start_link(fn -> {[], %{}} end, name: __MODULE__)
  11     spawn(fn -> schedule_update() end)
  12     agent
  13   end
  14
  15   def get_stats do
  16     Agent.get(__MODULE__, fn {_, stats} -> stats end)
  17   end
  18
  19   def get_peers do
  20     Agent.get(__MODULE__, fn {peers, _} -> peers end)
  21   end
  22
  23   def schedule_update do
  24     spawn(fn ->
  25       # 1 hour
  26       Process.sleep(1000 * 60 * 60 * 1)
  27       schedule_update()
  28     end)
  29
  30     update_stats()
  31   end
  32
  33   def update_stats do
  34     peers =
  35       from(
  36         u in Pleroma.User,
  37         select: fragment("distinct ?->'host'", u.info),
  38         where: u.local != ^true
  39       )
  40       |> Repo.all()
  41
  42     domain_count = Enum.count(peers)
  43
  44     status_query =
  45       from(u in User.local_user_query(), select: fragment("sum((?->>'note_count')::int)", u.info))
  46
  47     status_count = Repo.one(status_query)
  48     user_count = Repo.aggregate(User.local_user_query(), :count, :id)
  49
  50     Agent.update(__MODULE__, fn _ ->
  51       {peers, %{domain_count: domain_count, status_count: status_count, user_count: user_count}}
  52     end)
  53   end
  54 end