This repository has been archived on 2023-11-26. You can view files and clone it, but cannot push or open issues or pull requests.
bdfr-browser/lib/bdfr_browser/comment.ex
Daniel Kempkens d40db4ca4c
All checks were successful
Build / build (push) Successful in 3m46s
feat(search): use postgres native fulltext search
2023-10-09 21:51:53 +02:00

94 lines
2.6 KiB
Elixir

defmodule BdfrBrowser.Comment do
use Ecto.Schema
import Ecto.Query, only: [from: 2]
alias BdfrBrowser.{Post, Repo}
@primary_key {:id, :string, autogenerate: false}
@foreign_key_type :string
schema "comments" do
field :author, :string
field :body, :string
field :score, :integer
field :posted_at, :utc_datetime
belongs_to :post, Post
belongs_to :parent, __MODULE__
has_many :children, __MODULE__, foreign_key: :parent_id
end
def fetch_children!(comment) do
Repo.preload(comment, children: from(c in __MODULE__, order_by: [asc: c.posted_at])).children
end
def by_author(author) do
from(c in __MODULE__,
join: p in assoc(c, :post),
join: s in assoc(p, :subreddit),
select: %{
id: c.id,
body: c.body,
children: [],
posted_at: c.posted_at,
subreddit: s.name,
post_id: p.id,
post_title: p.title,
post_date: fragment("to_char(?, 'YYYY-MM')", p.posted_at)
},
where: c.author == ^author,
order_by: [desc: c.posted_at],
group_by: [c.id, p.id, s.name]
)
end
def search(str), do: search(str, nil)
def search(str, subreddits) when is_nil(subreddits) do
from(c in __MODULE__,
join: p in assoc(c, :post),
join: s in assoc(p, :subreddit),
select: %{
id: c.id,
author: c.author,
body: c.body,
children: [],
posted_at: c.posted_at,
subreddit: s.name,
post_id: p.id,
post_title: p.title,
post_date: fragment("to_char(?, 'YYYY-MM')", p.posted_at)
},
where:
fragment("? @@ websearch_to_tsquery('english', ?)", c.searchable_content, ^str) or
fragment("? @@ websearch_to_tsquery('german', ?)", c.searchable_content, ^str),
order_by: [desc: c.posted_at],
group_by: [c.id, p.id, s.name]
)
end
def search(str, subreddits) when is_list(subreddits) do
from(c in __MODULE__,
join: p in assoc(c, :post),
join: s in assoc(p, :subreddit),
select: %{
id: c.id,
author: c.author,
body: c.body,
children: [],
posted_at: c.posted_at,
subreddit: s.name,
post_id: p.id,
post_title: p.title,
post_date: fragment("to_char(?, 'YYYY-MM')", p.posted_at)
},
where:
s.name in ^subreddits and
(fragment("? @@ websearch_to_tsquery('english', ?)", c.searchable_content, ^str) or
fragment("? @@ websearch_to_tsquery('german', ?)", c.searchable_content, ^str)),
order_by: [desc: c.posted_at],
group_by: [c.id, p.id, s.name]
)
end
end