forked from mirrors/akkoma
42 lines
996 B
Elixir
42 lines
996 B
Elixir
|
# Pleroma: A lightweight social networking server
|
||
|
|
# Copyright © 2017-2019 Pleroma Authors <https://pleroma.social/>
|
||
|
|
# SPDX-License-Identifier: AGPL-3.0-only
|
||
|
|
|
||
|
|
defmodule Pleroma.CounterCache do
|
||
|
|
alias Pleroma.CounterCache
|
||
|
|
alias Pleroma.Repo
|
||
|
|
use Ecto.Schema
|
||
|
|
import Ecto.Changeset
|
||
|
|
import Ecto.Query
|
||
|
|
|
||
|
|
schema "counter_cache" do
|
||
|
|
field(:name, :string)
|
||
|
|
field(:count, :integer)
|
||
|
|
end
|
||
|
|
|
||
|
|
def changeset(struct, params) do
|
||
|
|
struct
|
||
|
|
|> cast(params, [:name, :count])
|
||
|
|
|> validate_required([:name])
|
||
|
|
|> unique_constraint(:name)
|
||
|
|
end
|
||
|
|
|
||
|
|
def get_as_map(names) when is_list(names) do
|
||
|
|
CounterCache
|
||
|
|
|> where([cc], cc.name in ^names)
|
||
|
|
|> Repo.all()
|
||
|
|
|> Enum.group_by(& &1.name, & &1.count)
|
||
|
|
|> Map.new(fn {k, v} -> {k, hd(v)} end)
|
||
|
|
end
|
||
|
|
|
||
|
|
def set(name, count) do
|
||
|
|
%CounterCache{}
|
||
|
|
|> changeset(%{"name" => name, "count" => count})
|
||
|
|
|> Repo.insert(
|
||
|
|
on_conflict: [set: [count: count]],
|
||
|
|
returning: true,
|
||
|
|
conflict_target: :name
|
||
|
|
)
|
||
|
|
end
|
||
|
|
end
|