akkoma/lib/pleroma/elasticsearch/store.ex

46 lines
934 B
Elixir
Raw Normal View History

2021-12-11 17:36:49 +00:00
defmodule Pleroma.Elasticsearch do
alias Pleroma.Activity
alias Pleroma.Elasticsearch.DocumentMappings
defp url do
Pleroma.Config.get([:elasticsearch, :url])
end
def put(%Activity{} = activity) do
Elastix.Document.index(
url(),
"activities",
"activity",
DocumentMappings.Activity.id(activity),
DocumentMappings.Activity.encode(activity)
)
end
def bulk_post(data, :activities) do
d = data
|> Enum.map(fn d ->
[
%{index: %{_id: DocumentMappings.Activity.id(d)}},
DocumentMappings.Activity.encode(d)
]
end)
|> List.flatten()
2021-12-11 18:48:46 +00:00
Elastix.Bulk.post(
2021-12-11 17:36:49 +00:00
url(),
d,
index: "activities",
type: "activity"
)
end
2021-12-11 18:48:46 +00:00
def search(query) do
Elastix.Search.search(
url(),
"activities",
["activity"],
%{query: %{term: %{content: query}}}
)
end
2021-12-11 17:36:49 +00:00
end