git.squeep.com Git - akkoma/blob - lib/mix/tasks/pleroma/benchmark.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 Mix.Tasks.Pleroma.Benchmark do
   6   import Mix.Pleroma
   7   use Mix.Task
   8
   9   def run(["search"]) do
  10     start_pleroma()
  11
  12     Benchee.run(%{
  13       "search" => fn ->
  14         Pleroma.Activity.search(nil, "cofe")
  15       end
  16     })
  17   end
  18
  19   def run(["tag"]) do
  20     start_pleroma()
  21
  22     Benchee.run(%{
  23       "tag" => fn ->
  24         %{"type" => "Create", "tag" => "cofe"}
  25         |> Pleroma.Web.ActivityPub.ActivityPub.fetch_public_activities()
  26       end
  27     })
  28   end
  29
  30   def run(["render_timeline", nickname]) do
  31     start_pleroma()
  32     user = Pleroma.User.get_by_nickname(nickname)
  33
  34     activities =
  35       %{}
  36       |> Map.put("type", ["Create", "Announce"])
  37       |> Map.put("blocking_user", user)
  38       |> Map.put("muting_user", user)
  39       |> Map.put("user", user)
  40       |> Map.put("limit", 80)
  41       |> Pleroma.Web.ActivityPub.ActivityPub.fetch_public_activities()
  42       |> Enum.reverse()
  43
  44     inputs = %{
  45       "One activity" => Enum.take_random(activities, 1),
  46       "Ten activities" => Enum.take_random(activities, 10),
  47       "Twenty activities" => Enum.take_random(activities, 20),
  48       "Forty activities" => Enum.take_random(activities, 40),
  49       "Eighty activities" => Enum.take_random(activities, 80)
  50     }
  51
  52     Benchee.run(
  53       %{
  54         "Parallel rendering" => fn activities ->
  55           Pleroma.Web.MastodonAPI.StatusView.render("index.json", %{
  56             activities: activities,
  57             for: user,
  58             as: :activity
  59           })
  60         end,
  61         "Standart rendering" => fn activities ->
  62           Pleroma.Web.MastodonAPI.StatusView.render("index.json", %{
  63             activities: activities,
  64             for: user,
  65             as: :activity,
  66             parallel: false
  67           })
  68         end
  69       },
  70       inputs: inputs
  71     )
  72   end
  73 end