1 defmodule Mix.Tasks.Pleroma.Benchmarks.Tags do
4 alias Pleroma.LoadTesting.Generator
8 Mix.Pleroma.start_pleroma()
9 activities_count = Repo.aggregate(from(a in Pleroma.Activity), :count, :id)
11 if activities_count == 0 do
12 IO.puts("Did not find any activities, cleaning and generating")
14 Generator.generate_users(users_max: 10)
15 Generator.generate_tagged_activities()
17 IO.puts("Found #{activities_count} activities, won't generate new ones")
20 tags = Enum.map(0..20, fn i -> {"For #tag_#{i}", "tag_#{i}"} end)
22 Enum.each(tags, fn {_, tag} ->
24 from(o in Pleroma.Object,
25 where: fragment("(?)->'tag' \\? (?)", o.data, ^tag)
28 count = Repo.aggregate(query, :count, :id)
29 IO.puts("Database contains #{count} posts tagged with #{tag}")
32 user = Repo.all(Pleroma.User) |> List.first()
36 "Hashtag fetching, any" => fn tags ->
37 Pleroma.Web.MastodonAPI.TimelineController.hashtag_fetching(
45 # Will always return zero results because no overlapping hashtags are generated.
46 "Hashtag fetching, all" => fn tags ->
47 Pleroma.Web.MastodonAPI.TimelineController.hashtag_fetching(
58 |> Enum.map(fn {_, v} -> v end)
59 |> Enum.chunk_every(2)
60 |> Enum.map(fn tags -> {"For #{inspect(tags)}", tags} end),
66 "Hashtag fetching" => fn tag ->
67 Pleroma.Web.MastodonAPI.TimelineController.hashtag_fetching(
82 IO.puts("Deleting old data...\n")
83 Ecto.Adapters.SQL.query!(Repo, "TRUNCATE users CASCADE;")
84 Ecto.Adapters.SQL.query!(Repo, "TRUNCATE activities CASCADE;")
85 Ecto.Adapters.SQL.query!(Repo, "TRUNCATE objects CASCADE;")