2020-05-31 17:51:41 +12:00
|
|
|
defmodule AshPostgres.Repo do
|
2020-06-04 17:27:26 +12:00
|
|
|
@moduledoc """
|
2022-08-24 11:56:46 +12:00
|
|
|
Resources that use `AshPostgres.DataLayer` use a `Repo` to access the database.
|
2020-06-04 17:27:26 +12:00
|
|
|
|
2022-08-24 11:56:46 +12:00
|
|
|
This repo is a thin wrapper around an `Ecto.Repo`.
|
2020-07-08 12:01:01 +12:00
|
|
|
|
|
|
|
You can use `Ecto.Repo`'s `init/2` to configure your repo like normal, but
|
|
|
|
instead of returning `{:ok, config}`, use `super(config)` to pass the
|
|
|
|
configuration to the `AshPostgres.Repo` implementation.
|
2020-09-03 20:18:11 +12:00
|
|
|
|
2020-11-17 18:37:04 +13:00
|
|
|
## Installed Extensions
|
2020-09-03 20:18:11 +12:00
|
|
|
|
2020-11-17 18:37:04 +13:00
|
|
|
To configure your list of installed extensions, define `installed_extensions/0`
|
|
|
|
|
2023-11-28 05:12:22 +13:00
|
|
|
Extensions can be a string, representing a standard postgres extension, or a module that implements `AshPostgres.CustomExtension`.
|
|
|
|
That custom extension will be called to generate migrations that serve a specific purpose.
|
|
|
|
|
2020-11-17 18:37:04 +13:00
|
|
|
Extensions that are relevant to ash_postgres:
|
|
|
|
|
2022-07-21 06:25:47 +12:00
|
|
|
* "ash-functions" - This isn't really an extension, but it expresses that certain functions
|
|
|
|
should be added when generating migrations, to support the `||` and `&&` operators in expressions.
|
2022-11-21 20:42:26 +13:00
|
|
|
* `"uuid-ossp"` - Sets UUID primary keys defaults in the migration generator
|
2023-09-12 14:34:51 +12:00
|
|
|
* `"pg_trgm"` - Makes the `AshPostgres.Functions.TrigramSimilarity` function available
|
2021-03-20 11:41:16 +13:00
|
|
|
* "citext" - Allows case insensitive fields to be used
|
2023-09-12 14:34:51 +12:00
|
|
|
* `"vector"` - Makes the `AshPostgres.Functions.VectorCosineDistance` function available. See `AshPostgres.Extensions.Vector` for more setup instructions.
|
2020-09-03 20:18:11 +12:00
|
|
|
|
|
|
|
```
|
|
|
|
def installed_extensions() do
|
2023-11-28 05:12:22 +13:00
|
|
|
["pg_trgm", "uuid-ossp", "vector", YourCustomExtension]
|
2020-09-03 20:18:11 +12:00
|
|
|
end
|
|
|
|
```
|
2022-12-01 14:52:36 +13:00
|
|
|
|
|
|
|
## Transaction Hooks
|
|
|
|
|
|
|
|
You can define `on_transaction_begin/1`, which will be invoked whenever a transaction is started for Ash.
|
|
|
|
|
|
|
|
This will be invoked with a map containing a `type` key and metadata.
|
|
|
|
|
|
|
|
```elixir
|
|
|
|
%{type: :create, %{resource: YourApp.YourResource, action: :action}}
|
|
|
|
```
|
2020-06-04 17:27:26 +12:00
|
|
|
"""
|
2020-07-08 12:01:01 +12:00
|
|
|
|
|
|
|
@doc "Use this to inform the data layer about what extensions are installed"
|
2024-02-06 10:52:09 +13:00
|
|
|
@callback installed_extensions() :: [String.t() | module()]
|
2022-11-21 18:31:14 +13:00
|
|
|
|
2024-03-28 09:52:28 +13:00
|
|
|
@doc "Configure the version of postgres that is being used."
|
|
|
|
@callback pg_version() :: Version.t()
|
|
|
|
|
2022-11-21 18:31:14 +13:00
|
|
|
@doc """
|
|
|
|
Use this to inform the data layer about the oldest potential postgres version it will be run on.
|
|
|
|
|
|
|
|
Must be an integer greater than or equal to 13.
|
2023-10-14 15:47:11 +13:00
|
|
|
|
|
|
|
## Combining with other tools
|
|
|
|
|
|
|
|
For things like `Fly.Repo`, where you might need to have more fine grained control over the repo module,
|
|
|
|
you can use the `define_ecto_repo?: false` option to `use AshPostgres.Repo`.
|
2022-11-21 18:31:14 +13:00
|
|
|
"""
|
|
|
|
|
2022-12-01 13:06:51 +13:00
|
|
|
@callback on_transaction_begin(reason :: Ash.DataLayer.transaction_reason()) :: term
|
|
|
|
|
2020-10-29 16:53:28 +13:00
|
|
|
@doc "Return a list of all schema names (only relevant for a multitenant implementation)"
|
|
|
|
@callback all_tenants() :: [String.t()]
|
|
|
|
@doc "The path where your tenant migrations are stored (only relevant for a multitenant implementation)"
|
2023-02-06 06:46:44 +13:00
|
|
|
@callback tenant_migrations_path() :: String.t() | nil
|
2021-01-27 13:16:29 +13:00
|
|
|
@doc "The path where your migrations are stored"
|
2023-02-06 06:46:44 +13:00
|
|
|
@callback migrations_path() :: String.t() | nil
|
2021-07-12 18:43:39 +12:00
|
|
|
@doc "The default prefix(postgres schema) to use when building queries"
|
|
|
|
@callback default_prefix() :: String.t()
|
2024-03-28 09:52:28 +13:00
|
|
|
|
2021-11-10 22:18:36 +13:00
|
|
|
@doc "Allows overriding a given migration type for *all* fields, for example if you wanted to always use :timestamptz for :utc_datetime fields"
|
|
|
|
@callback override_migration_type(atom) :: atom
|
2024-03-28 10:05:14 +13:00
|
|
|
@doc "Should the repo should be created by `mix ash_postgres.create`?"
|
|
|
|
@callback create?() :: boolean
|
|
|
|
@doc "Should the repo should be dropped by `mix ash_postgres.drop`?"
|
|
|
|
@callback drop?() :: boolean
|
2020-05-31 17:51:41 +12:00
|
|
|
|
|
|
|
defmacro __using__(opts) do
|
|
|
|
quote bind_quoted: [opts: opts] do
|
2023-10-14 15:47:11 +13:00
|
|
|
if Keyword.get(opts, :define_ecto_repo?, true) do
|
|
|
|
otp_app = opts[:otp_app] || raise("Must configure OTP app")
|
2020-05-31 17:51:41 +12:00
|
|
|
|
2023-10-14 15:47:11 +13:00
|
|
|
use Ecto.Repo,
|
|
|
|
adapter: Ecto.Adapters.Postgres,
|
|
|
|
otp_app: otp_app
|
|
|
|
end
|
2020-05-31 17:51:41 +12:00
|
|
|
|
2023-02-06 06:46:44 +13:00
|
|
|
@behaviour AshPostgres.Repo
|
|
|
|
|
2023-01-30 14:27:43 +13:00
|
|
|
defoverridable insert: 2, insert: 1, insert!: 2, insert!: 1
|
|
|
|
|
2020-10-29 16:53:28 +13:00
|
|
|
def installed_extensions, do: []
|
|
|
|
def tenant_migrations_path, do: nil
|
2021-01-27 13:16:29 +13:00
|
|
|
def migrations_path, do: nil
|
2021-07-12 18:43:39 +12:00
|
|
|
def default_prefix, do: "public"
|
2021-11-10 22:18:36 +13:00
|
|
|
def override_migration_type(type), do: type
|
2024-03-28 10:05:14 +13:00
|
|
|
def create?, do: true
|
|
|
|
def drop?, do: true
|
|
|
|
|
2021-04-03 03:24:42 +13:00
|
|
|
|
2023-12-30 15:49:34 +13:00
|
|
|
def transaction!(fun) do
|
|
|
|
case fun.() do
|
|
|
|
{:ok, value} -> value
|
|
|
|
{:error, error} -> raise Ash.Error.to_error_class(error)
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
2021-04-03 03:24:42 +13:00
|
|
|
def all_tenants do
|
|
|
|
raise """
|
|
|
|
`#{inspect(__MODULE__)}.all_tenants/0` was called, but was not defined. In order to migrate tenants, you must define this function.
|
|
|
|
For example, you might say:
|
|
|
|
|
|
|
|
def all_tenants do
|
|
|
|
for org <- MyApp.Accounts.all_organizations!() do
|
|
|
|
org.schema
|
|
|
|
end
|
|
|
|
end
|
|
|
|
"""
|
|
|
|
end
|
2020-05-31 17:51:41 +12:00
|
|
|
|
2020-06-28 07:11:28 +12:00
|
|
|
def init(_, config) do
|
2020-10-29 16:53:28 +13:00
|
|
|
new_config =
|
|
|
|
config
|
|
|
|
|> Keyword.put(:installed_extensions, installed_extensions())
|
|
|
|
|> Keyword.put(:tenant_migrations_path, tenant_migrations_path())
|
2021-01-27 13:16:29 +13:00
|
|
|
|> Keyword.put(:migrations_path, migrations_path())
|
2021-07-12 18:43:39 +12:00
|
|
|
|> Keyword.put(:default_prefix, default_prefix())
|
2020-05-31 17:51:41 +12:00
|
|
|
|
|
|
|
{:ok, new_config}
|
|
|
|
end
|
|
|
|
|
2022-12-01 13:06:51 +13:00
|
|
|
def on_transaction_begin(_reason), do: :ok
|
|
|
|
|
2023-01-30 14:27:43 +13:00
|
|
|
def insert(struct_or_changeset, opts \\ []) do
|
|
|
|
struct_or_changeset
|
|
|
|
|> to_ecto()
|
|
|
|
|> then(fn value ->
|
|
|
|
repo = get_dynamic_repo()
|
|
|
|
|
|
|
|
Ecto.Repo.Schema.insert(
|
|
|
|
__MODULE__,
|
|
|
|
repo,
|
|
|
|
value,
|
|
|
|
Ecto.Repo.Supervisor.tuplet(repo, prepare_opts(:insert, opts))
|
|
|
|
)
|
|
|
|
end)
|
|
|
|
|> from_ecto()
|
|
|
|
end
|
|
|
|
|
|
|
|
def insert!(struct_or_changeset, opts \\ []) do
|
|
|
|
struct_or_changeset
|
|
|
|
|> to_ecto()
|
|
|
|
|> then(fn value ->
|
|
|
|
repo = get_dynamic_repo()
|
|
|
|
|
|
|
|
Ecto.Repo.Schema.insert!(
|
|
|
|
__MODULE__,
|
|
|
|
repo,
|
|
|
|
value,
|
|
|
|
Ecto.Repo.Supervisor.tuplet(repo, prepare_opts(:insert, opts))
|
|
|
|
)
|
|
|
|
end)
|
|
|
|
|> from_ecto()
|
|
|
|
end
|
|
|
|
|
|
|
|
def from_ecto({:ok, result}), do: {:ok, from_ecto(result)}
|
|
|
|
def from_ecto({:error, _} = other), do: other
|
|
|
|
|
|
|
|
def from_ecto(nil), do: nil
|
|
|
|
|
|
|
|
def from_ecto(value) when is_list(value) do
|
|
|
|
Enum.map(value, &from_ecto/1)
|
|
|
|
end
|
|
|
|
|
|
|
|
def from_ecto(%resource{} = record) do
|
|
|
|
if Spark.Dsl.is?(resource, Ash.Resource) do
|
|
|
|
empty = struct(resource)
|
|
|
|
|
|
|
|
resource
|
|
|
|
|> Ash.Resource.Info.relationships()
|
|
|
|
|> Enum.reduce(record, fn relationship, record ->
|
|
|
|
case Map.get(record, relationship.name) do
|
|
|
|
%Ecto.Association.NotLoaded{} ->
|
|
|
|
Map.put(record, relationship.name, Map.get(empty, relationship.name))
|
|
|
|
|
|
|
|
value ->
|
|
|
|
Map.put(record, relationship.name, from_ecto(value))
|
|
|
|
end
|
|
|
|
end)
|
|
|
|
else
|
|
|
|
record
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
def from_ecto(other), do: other
|
|
|
|
|
|
|
|
def to_ecto(nil), do: nil
|
|
|
|
|
|
|
|
def to_ecto(value) when is_list(value) do
|
|
|
|
Enum.map(value, &to_ecto/1)
|
|
|
|
end
|
|
|
|
|
|
|
|
def to_ecto(%resource{} = record) do
|
|
|
|
if Spark.Dsl.is?(resource, Ash.Resource) do
|
|
|
|
resource
|
|
|
|
|> Ash.Resource.Info.relationships()
|
|
|
|
|> Enum.reduce(record, fn relationship, record ->
|
|
|
|
value =
|
|
|
|
case Map.get(record, relationship.name) do
|
|
|
|
%Ash.NotLoaded{} ->
|
|
|
|
%Ecto.Association.NotLoaded{
|
2023-03-21 09:11:05 +13:00
|
|
|
__field__: relationship.name,
|
2023-01-30 14:27:43 +13:00
|
|
|
__cardinality__: relationship.cardinality
|
|
|
|
}
|
|
|
|
|
|
|
|
value ->
|
|
|
|
to_ecto(value)
|
|
|
|
end
|
|
|
|
|
|
|
|
Map.put(record, relationship.name, value)
|
|
|
|
end)
|
|
|
|
else
|
|
|
|
record
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
def to_ecto(other), do: other
|
|
|
|
|
2021-07-12 18:43:39 +12:00
|
|
|
defoverridable init: 2,
|
2022-12-01 13:06:51 +13:00
|
|
|
on_transaction_begin: 1,
|
2021-07-12 18:43:39 +12:00
|
|
|
installed_extensions: 0,
|
|
|
|
all_tenants: 0,
|
|
|
|
tenant_migrations_path: 0,
|
2021-11-10 22:18:36 +13:00
|
|
|
default_prefix: 0,
|
2024-03-28 10:05:14 +13:00
|
|
|
override_migration_type: 1,
|
|
|
|
create?: 0,
|
|
|
|
drop?: 0
|
2020-05-31 17:51:41 +12:00
|
|
|
end
|
|
|
|
end
|
|
|
|
end
|