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   require Logger
   9   use Mix.Task
  10
  11   @shortdoc "A collection of database related tasks"
  12   @moduledoc """
  13    A collection of database related tasks
  14
  15    ## Replace embedded objects with their references
  16
  17    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.
  18
  19        mix pleroma.database remove_embedded_objects
  20
  21     Options:
  22     - `--vacuum` - run `VACUUM FULL` after the embedded objects are replaced with their references
  23
  24   ## Create a conversation for all existing DMs. Can be safely re-run.
  25
  26       mix pleroma.database bump_all_conversations
  27
  28   """
  29   def run(["remove_embedded_objects" | args]) do
  30     {options, [], []} =
  31       OptionParser.parse(
  32         args,
  33         strict: [
  34           vacuum: :boolean
  35         ]
  36       )
  37
  38     Common.start_pleroma()
  39     Logger.info("Removing embedded objects")
  40
  41     Pleroma.Repo.query!(
  42       "update activities set data = jsonb_set(data, '{object}'::text[], data->'object'->'id') where data->'object'->>'id' is not null;",
  43       [],
  44       timeout: :infinity
  45     )
  46
  47     if Keyword.get(options, :vacuum) do
  48       Logger.info("Runnning VACUUM FULL")
  49
  50       Pleroma.Repo.query!(
  51         "vacuum full;",
  52         [],
  53         timeout: :infinity
  54       )
  55     end
  56   end
  57
  58   def run(["bump_all_conversations"]) do
  59     Common.start_pleroma()
  60     Conversation.bump_for_all_activities()
  61   end
  62 end