1 # Pleroma: A lightweight social networking server
2 # Copyright © 2017-2020 Pleroma Authors <https://pleroma.social/>
3 # SPDX-License-Identifier: AGPL-3.0-only
5 defmodule Pleroma.Repo do
8 adapter: Ecto.Adapters.Postgres,
9 migration_timestamps: [type: :naive_datetime_usec]
14 defmodule Instrumenter do
15 use Prometheus.EctoInstrumenter
19 Dynamically loads the repository url from the
20 DATABASE_URL environment variable.
23 {:ok, Keyword.put(opts, :url, System.get_env("DATABASE_URL"))}
26 @doc "find resource based on prepared query"
27 @spec find_resource(Ecto.Query.t()) :: {:ok, struct()} | {:error, :not_found}
28 def find_resource(%Ecto.Query{} = query) do
29 case __MODULE__.one(query) do
30 nil -> {:error, :not_found}
31 resource -> {:ok, resource}
35 def find_resource(_query), do: {:error, :not_found}
38 Gets association from cache or loads if need
42 iex> Repo.get_assoc(token, :user)
46 @spec get_assoc(struct(), atom()) :: {:ok, struct()} | {:error, :not_found}
47 def get_assoc(resource, association) do
48 case __MODULE__.preload(resource, association) do
49 %{^association => assoc} when not is_nil(assoc) -> {:ok, assoc}
50 _ -> {:error, :not_found}
54 def check_migrations_applied!() do
55 unless Pleroma.Config.get(
56 [:i_am_aware_this_may_cause_data_loss, :disable_migration_check],
59 Ecto.Migrator.with_repo(__MODULE__, fn repo ->
61 Ecto.Migrator.migrations(repo)
64 {:down, _, _} -> false
67 if length(down_migrations) > 0 do
68 down_migrations_text =
69 Enum.map(down_migrations, fn {:down, id, name} -> "- #{name} (#{id})\n" end)
72 "The following migrations were not applied:\n#{down_migrations_text}If you want to start Pleroma anyway, set\nconfig :pleroma, :i_am_aware_this_may_cause_data_loss, disable_migration_check: true"
75 raise Pleroma.Repo.UnappliedMigrationsError
83 def chunk_stream(query, chunk_size) do
84 # We don't actually need start and end funcitons of resource streaming,
85 # but it seems to be the only way to not fetch records one-by-one and
86 # have individual records be the elements of the stream, instead of
94 |> where([r], r.id > ^last_id)
102 last_id = List.last(records).id
111 defmodule Pleroma.Repo.UnappliedMigrationsError do
112 defexception message: "Unapplied Migrations detected"