git.squeep.com Git - akkoma/blob - lib/mix/tasks/pleroma/database.ex

   1 # Pleroma: A lightweight social networking server
   2 # Copyright © 2017-2018 Pleroma Authors <https://pleroma.social/>
   3 # SPDX-License-Identifier: AGPL-3.0-only
   4
   5 defmodule Mix.Tasks.Pleroma.Database do
   6   alias Mix.Tasks.Pleroma.Common
   7   alias Pleroma.Conversation
   8   alias Pleroma.Repo
   9   alias Pleroma.User
  10   require Logger
  11   use Mix.Task
  12
  13   @shortdoc "A collection of database related tasks"
  14   @moduledoc """
  15    A collection of database related tasks
  16
  17    ## Replace embedded objects with their references
  18
  19    Replaces embedded objects with references to them in the `objects` table. Only needs to be ran once. The reason why this is not a migration is because it could significantly increase the database size after being ran, however after this `VACUUM FULL` will be able to reclaim about 20% (really depends on what is in the database, your mileage may vary) of the db size before the migration.
  20
  21        mix pleroma.database remove_embedded_objects
  22
  23     Options:
  24     - `--vacuum` - run `VACUUM FULL` after the embedded objects are replaced with their references
  25
  26   ## Create a conversation for all existing DMs. Can be safely re-run.
  27
  28       mix pleroma.database bump_all_conversations
  29
  30   ## Remove duplicated items from following and update followers count for all users
  31
  32       mix pleroma.database update_users_following_followers_counts
  33   """
  34   def run(["remove_embedded_objects" | args]) do
  35     {options, [], []} =
  36       OptionParser.parse(
  37         args,
  38         strict: [
  39           vacuum: :boolean
  40         ]
  41       )
  42
  43     Common.start_pleroma()
  44     Logger.info("Removing embedded objects")
  45
  46     Repo.query!(
  47       "update activities set data = jsonb_set(data, '{object}'::text[], data->'object'->'id') where data->'object'->>'id' is not null;",
  48       [],
  49       timeout: :infinity
  50     )
  51
  52     if Keyword.get(options, :vacuum) do
  53       Logger.info("Runnning VACUUM FULL")
  54
  55       Repo.query!(
  56         "vacuum full;",
  57         [],
  58         timeout: :infinity
  59       )
  60     end
  61   end
  62
  63   def run(["bump_all_conversations"]) do
  64     Common.start_pleroma()
  65     Conversation.bump_for_all_activities()
  66   end
  67
  68   def run(["update_users_following_followers_counts"]) do
  69     Common.start_pleroma()
  70
  71     users = Repo.all(User)
  72     Enum.each(users, &User.remove_duplicated_following/1)
  73     Enum.each(users, &User.update_follower_count/1)
  74   end
  75 end