akkoma/lib/pleroma/elasticsearch/store.ex

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()

    Elastix.Bulk.post(
        url(),
        d,
        index: "activities",
        type: "activity"
    )
  end

  def search(query) do
    Elastix.Search.search(
        url(),
        "activities",
        ["activity"],
        %{query: %{term: %{content: query}}}
    )
  end
end