defmodule Pleroma.Elasticsearch do
alias Pleroma.Activity
alias Pleroma.User
+ alias Pleroma.Object
alias Pleroma.Elasticsearch.DocumentMappings
alias Pleroma.Config
require Logger
Config.get([:search, :provider]) == Pleroma.Search.Elasticsearch
end
+ def delete_by_id(:activity, id) do
+ if enabled?() do
+ Elastix.Document.delete(url(), "activities", "activity", id)
+ end
+ end
+
def put_by_id(:activity, id) do
id
|> Activity.get_by_id_with_object()
{:ok, :skipped}
end
+ def maybe_bulk_post(data, type) do
+ if enabled?() do
+ bulk_post(data, type)
+ end
+ end
+
def put(%Activity{} = activity) do
- {:ok, _} =
- Elastix.Document.index(
- url(),
- "activities",
- "activity",
- DocumentMappings.Activity.id(activity),
- DocumentMappings.Activity.encode(activity)
- )
+ with {:ok, _} <-
+ Elastix.Document.index(
+ url(),
+ "activities",
+ "activity",
+ DocumentMappings.Activity.id(activity),
+ DocumentMappings.Activity.encode(activity)
+ ) do
+ activity
+ |> Map.get(:object)
+ |> Object.hashtags()
+ |> Enum.map(fn x ->
+ %{id: x, name: x, timestamp: DateTime.to_iso8601(DateTime.utc_now())}
+ end)
+ |> bulk_post(:hashtags)
+ else
+ {:error, %{reason: err}} ->
+ Logger.error("Could not put activity: #{err}")
+ :skipped
+ end
end
def put(%User{} = user) do
- {:ok, _} =
- Elastix.Document.index(
- url(),
- "users",
- "user",
- DocumentMappings.User.id(user),
- DocumentMappings.User.encode(user)
- )
+ with {:ok, _} <-
+ Elastix.Document.index(
+ url(),
+ "users",
+ "user",
+ DocumentMappings.User.id(user),
+ DocumentMappings.User.encode(user)
+ ) do
+ :ok
+ else
+ {:error, %{reason: err}} ->
+ Logger.error("Could not put user: #{err}")
+ :skipped
+ end
end
def bulk_post(data, :activities) do
end)
|> List.flatten()
- {:ok, %{body: %{"errors" => false}}} =
- Elastix.Bulk.post(
- url(),
- d,
- index: "activities",
- type: "activity"
- )
- end
+ with {:ok, %{body: %{"errors" => false}}} <-
+ Elastix.Bulk.post(
+ url(),
+ d,
+ index: "activities",
+ type: "activity"
+ ) do
+ :ok
+ else
+ {:error, %{reason: err}} ->
+ Logger.error("Could not bulk put activity: #{err}")
+ :skipped
- def maybe_bulk_post(data, type) do
- if enabled?() do
- bulk_post(data, type)
+ {:ok, %{body: body}} ->
+ IO.inspect(body)
+ :skipped
end
end
end)
|> List.flatten()
- Elastix.Bulk.post(
- url(),
- d,
- index: "users",
- type: "user"
- )
+ with {:ok, %{body: %{"errors" => false}}} <-
+ Elastix.Bulk.post(
+ url(),
+ d,
+ index: "users",
+ type: "user"
+ ) do
+ :ok
+ else
+ {:error, %{reason: err}} ->
+ Logger.error("Could not bulk put users: #{err}")
+ :skipped
+
+ {:ok, %{body: body}} ->
+ IO.inspect(body)
+ :skipped
+ end
end
- def bulk_post([] = data, :hashtags) do
+ def bulk_post(data, :hashtags) when is_list(data) do
d =
data
|> Enum.map(fn d ->
end)
|> List.flatten()
- Elastix.Bulk.post(
- url(),
- d,
- index: "hashtags",
- type: "hashtag"
- )
+ with {:ok, %{body: %{"errors" => false}}} <-
+ Elastix.Bulk.post(
+ url(),
+ d,
+ index: "hashtags",
+ type: "hashtag"
+ ) do
+ :ok
+ else
+ {:error, %{reason: err}} ->
+ Logger.error("Could not bulk put hashtags: #{err}")
+ :skipped
+
+ {:ok, %{body: body}} ->
+ IO.inspect(body)
+ :skipped
+ end
end
def bulk_post(_, :hashtags), do: {:ok, nil}