make bulk user creation from admin works as a transaction
[akkoma] / 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 require Logger
8 use Mix.Task
9
10 @shortdoc "A collection of database related tasks"
11 @moduledoc """
12 A collection of database related tasks
13
14 ## Replace embedded objects with their references
15
16 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.
17
18 mix pleroma.database remove_embedded_objects
19
20 Options:
21 - `--vacuum` - run `VACUUM FULL` after the embedded objects are replaced with their references
22 """
23 def run(["remove_embedded_objects" | args]) do
24 {options, [], []} =
25 OptionParser.parse(
26 args,
27 strict: [
28 vacuum: :boolean
29 ]
30 )
31
32 Common.start_pleroma()
33 Logger.info("Removing embedded objects")
34
35 Pleroma.Repo.query!(
36 "update activities set data = jsonb_set(data, '{object}'::text[], data->'object'->'id') where data->'object'->>'id' is not null;",
37 [],
38 timeout: :infinity
39 )
40
41 if Keyword.get(options, :vacuum) do
42 Logger.info("Runnning VACUUM FULL")
43
44 Pleroma.Repo.query!(
45 "vacuum full;",
46 [],
47 timeout: :infinity
48 )
49 end
50 end
51 end