about summary refs log tree commit diff
path: root/app/models/account_stat.rb
blob: e70b54d79b93df00916053708e1ba0aabd9c790c (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
# frozen_string_literal: true
# == Schema Information
#
# Table name: account_stats
#
#  id              :bigint(8)        not null, primary key
#  account_id      :bigint(8)        not null
#  statuses_count  :bigint(8)        default(0), not null
#  following_count :bigint(8)        default(0), not null
#  followers_count :bigint(8)        default(0), not null
#  created_at      :datetime         not null
#  updated_at      :datetime         not null
#  last_status_at  :datetime
#  lock_version    :integer          default(0), not null
#

class AccountStat < ApplicationRecord
  belongs_to :account, inverse_of: :account_stat

  update_index('accounts#account', :account)

  def increment_count!(key)
    update(attributes_for_increment(key))
  rescue ActiveRecord::StaleObjectError, ActiveRecord::RecordNotUnique
    begin
      reload_with_id
    rescue ActiveRecord::RecordNotFound
      return
    end

    retry
  end

  def decrement_count!(key)
    update(attributes_for_decrement(key))
  rescue ActiveRecord::StaleObjectError, ActiveRecord::RecordNotUnique
    begin
      reload_with_id
    rescue ActiveRecord::RecordNotFound
      return
    end

    retry
  end

  private

  def attributes_for_increment(key)
    attrs = { key => public_send(key) + 1 }
    attrs[:last_status_at] = Time.now.utc if key == :statuses_count
    attrs
  end

  def attributes_for_decrement(key)
    attrs = { key => [public_send(key) - 1, 0].max }
    attrs
  end

  def reload_with_id
    self.id = self.class.find_by!(account: account).id if new_record?
    reload
  end
end