Это может быть больше вопрос SQL, чем вопрос Elixir / Etco.
У меня много ко многим отношениям Пользователь-Транзакции-Торговец, где у пользователя есть много Продавцов через Транзакции, а у Продавца много клиентов через Транзакции. Это довольно типично. Я могу привлечь всех клиентов продавца через Ecto, сделав что-то вроде этого:
def find_merchant_customers(%{"merchant_id" => id}) do
merchant = Repo.get(Merchant, id)
Repo.all assoc(merchant, :customers)
end
Если я хочу найти баланс для пользователя с конкретным продавцом, у меня есть SQL-запрос, подобный этому, который суммирует транзакции и создает баланс для этого продавца.
def customer_balance(user_id: user_id, merchant_id: merchant_id) do
q = from t in Transaction,
select: fragment("SUM(CASE WHEN ? = 'credit' THEN (?) ELSE - (?) END)", t.type, t.amount, t.amount),
where: t.user_id == ^user_id and t.merchant_id == ^merchant_id
balance = Repo.one(q) || 0
do_balance(balance, "asset")
|> Money.new(:USD)
end
Вопрос
Как объединить две операции в один запрос, чтобы функция Join извлекала список пользователей и заполняла пользователя виртуальным атрибутом Balance. Я знаю, что могу просто запустить первый запрос и получить список пользователей, а затем преобразовать эти данные, извлекая каждый из балансов для каждого пользователя, хотя это кажется очень неэффективным. Альтернативой может быть понимание того, как назначить select fragement(
атрибуту в запросе в качестве подзапроса. Любое руководство было бы полезно.
Моя модель пользователя
defmodule MyApp.User do
@moduledoc """
User struct for user related data
"""
import MyApp.Validation
use MyApp.Model
use Coherence.Schema
schema "my_app_users" do
field :email, :string
field :first_name, :string
field :last_name, :string
field :role, :integer
field :birthdate, Ecto.Date
field :address1, :string
field :address2, :string
field :city, :string
field :state, :string
field :zip, :string
field :status, :boolean, default: true
field :verified_email, :boolean, default: false
field :verified_phone, :boolean, default: false
field :mobile, :string
field :card, :string
field :sms_code, :string
field :balance, Money.Ecto, virtual: true
field :points, :integer, virtual: true
coherence_schema
has_many :transactions, MyApp.Transaction
has_many :merchants, through: [:transactions, :merchant]
timestamps()
end
@doc """
Builds a changeset based on the `struct` and `params`.
"""
def changeset(struct, params \\ %{}) do
struct
|> cast(params, [:email, :first_name, :last_name, :password_hash, :role, :birthdate, :address1, :address2, :city, :state, :zip, :status, :mobile, :card, :sms_code, :status, :merchant_id, :verified_email, :verified_phone])
|> validate_required_inclusion([:email, :mobile])
|> validate_format(:email, ~r/(\w+)@([\w.]+)/)
end
defp put_password_hash(changeset) do
case changeset do
%Ecto.Changeset{valid?: true, changes: %{password: password}} ->
put_change(changeset, :password_hash, Comeonin.Bcrypt.hashpwsalt(password))
_ ->
changeset
end
end
Моя модель продавца
defmodule MyApp.Merchant do
@moduledoc """
Merchant Struct
Merchant has an owner of a User - Which must exist
"""
use MyApp.Model
use Arc.Ecto.Schema
schema "my_app_merchants" do
field :name, :string
field :email, :string
field :address1, :string
field :address2, :string
field :city, :string
field :zip, :string
field :state, :string
field :status, :boolean, default: true
field :description, :string
field :image, MyRewards.Avatar.Type
field :phone, :string
field :website, :string
has_many :transactions, MyApp.Transaction
has_many :customers, through: [:transactions, :user]
timestamps()
end
@doc """
Builds a changeset based on the `struct` and `params`.
"""
def changeset(struct, params \\ %{}) do
struct
|> cast(params, [:name, :email, :address1, :address2, :city, :zip, :state, :status, :description, :phone, :website, :status, :category_id, :user_id])
|> cast_attachments(params, [:image])
|> validate_required([:name])
|> validate_format(:email, ~r/(\w+)@([\w.]+)/)
end
end
Функции запроса
def find_merchant_customers(%{"merchant_id" => id}) do
merchant = Repo.get(Merchant, id)
Repo.all assoc(merchant, :customers)
end
def customer_balance(user_id: user_id, merchant_id: merchant_id) do
q = from t in Transaction,
select: fragment("SUM(CASE WHEN ? = 'credit' THEN (?) ELSE - (?) END)", t.type, t.amount, t.amount),
where: t.user_id == ^user_id and t.merchant_id == ^merchant_id
balance = Repo.one(q) || 0
do_balance(balance, "asset")
|> Money.new(:USD)
end
1 ответ
Переместите фрагмент в макрос, чтобы код оставался ясным:
defmacro balance_amount(transaction) do
quote do
fragment("CASE WHEN ? = 'credit' THEN (?) ELSE - (?) END",
unquote(transaction).type, unquote(transaction).amount, unquote(transaction).amount)
end
end
Создайте подзапрос с% {user_id, merchant_id, balance}
def user_merchant_balance do
from t in Transaction,
select: %{user_id: t.user_id, merchant_id: t.merchant_id, balance: sum(balance_amount(t))},
group_by: [t.user_id, t.merchant_id]
end
Присоединитесь к подзапросу из основного запроса, используйте синтаксис обновления карты% {|} для заполнения виртуального поля:
def merchant_customers(merchant_id) do
from u in User,
join: b in subquery(user_merchant_balance()), on: u.id == b.user_id,
where: b.merchant_id == ^merchant_id,
select: %{u | balance: b.balance}
end
Изменить: в Ecto 2.2 поле balance
можно преобразовать в Money.Ecto.Type
def merchant_customers(merchant_id) do
from u in User,
join: b in subquery(user_merchant_balance()), on: u.id == b.user_id,
where: b.merchant_id == ^merchant_id,
select: %{u | balance: type(b.balance, Money.Ecto.Type)}
end
Похожие вопросы
Новые вопросы
sql
Язык структурированных запросов (SQL) - это язык запросов к базам данных. Вопросы должны включать примеры кода, структуру таблицы, примеры данных и тег для используемой реализации СУБД (например, MySQL, PostgreSQL, Oracle, MS SQL Server, IBM DB2 и т. Д.). Если ваш вопрос относится исключительно к конкретной СУБД (использует определенные расширения / функции), используйте вместо этого тег этой СУБД. Ответы на вопросы, помеченные SQL, должны использовать стандарт ISO / IEC SQL.