git.squeep.com Git - akkoma/blob - lib/pleroma/migrators/hashtags_table_migrator/state.ex

   1 # Pleroma: A lightweight social networking server
   2 # Copyright © 2017-2021 Pleroma Authors <https://pleroma.social/>
   3 # SPDX-License-Identifier: AGPL-3.0-only
   4
   5 defmodule Pleroma.Migrators.HashtagsTableMigrator.State do
   6   use Agent
   7
   8   @init_state %{}
   9   @reg_name {:global, __MODULE__}
  10
  11   def start_link(_) do
  12     Agent.start_link(fn -> @init_state end, name: @reg_name)
  13   end
  14
  15   def clear do
  16     Agent.update(@reg_name, fn _state -> @init_state end)
  17   end
  18
  19   def get do
  20     Agent.get(@reg_name, & &1)
  21   end
  22
  23   def put(key, value) do
  24     Agent.update(@reg_name, fn state ->
  25       Map.put(state, key, value)
  26     end)
  27   end
  28
  29   def increment(key, increment \\ 1) do
  30     Agent.update(@reg_name, fn state ->
  31       updated_value = (state[key] || 0) + increment
  32       Map.put(state, key, updated_value)
  33     end)
  34   end
  35 end