2018-11-19 08:43:52 +09:00
# frozen_string_literal: true
2023-12-01 20:00:41 +09:00
module Account::Counters
2018-11-19 08:43:52 +09:00
extend ActiveSupport :: Concern
2021-03-19 21:14:57 +09:00
ALLOWED_COUNTER_KEYS = % i ( statuses_count following_count followers_count ) . freeze
2018-11-19 08:43:52 +09:00
included do
has_one :account_stat , inverse_of : :account
after_save :save_account_stat
end
delegate :statuses_count ,
:statuses_count = ,
:following_count ,
:following_count = ,
:followers_count ,
:followers_count = ,
2018-12-07 01:36:11 +09:00
:last_status_at ,
2018-11-19 08:43:52 +09:00
to : :account_stat
2021-03-19 21:14:57 +09:00
# @param [Symbol] key
def increment_count! ( key )
update_count! ( key , 1 )
end
# @param [Symbol] key
def decrement_count! ( key )
update_count! ( key , - 1 )
end
# @param [Symbol] key
# @param [Integer] value
def update_count! ( key , value )
raise ArgumentError , " Invalid key #{ key } " unless ALLOWED_COUNTER_KEYS . include? ( key )
raise ArgumentError , 'Do not call update_count! on dirty objects' if association ( :account_stat ) . loaded? && account_stat & . changed? && account_stat . changed_attribute_names_to_save == %w( id )
value = value . to_i
default_value = value . positive? ? value : 0
# We do an upsert using manually written SQL, as Rails' upsert method does
# not seem to support writing expressions in the UPDATE clause, but only
# re-insert the provided values instead.
# Even ARel seem to be missing proper handling of upserts.
sql = if value . positive? && key == :statuses_count
<<-SQL.squish
INSERT INTO account_stats ( account_id , #{key}, created_at, updated_at, last_status_at)
VALUES ( :account_id , :default_value , now ( ) , now ( ) , now ( ) )
ON CONFLICT ( account_id ) DO UPDATE
SET #{key} = account_stats.#{key} + :value,
last_status_at = now ( ) ,
updated_at = now ( )
RETURNING id ;
SQL
else
<<-SQL.squish
INSERT INTO account_stats ( account_id , #{key}, created_at, updated_at)
VALUES ( :account_id , :default_value , now ( ) , now ( ) )
ON CONFLICT ( account_id ) DO UPDATE
SET #{key} = account_stats.#{key} + :value,
updated_at = now ( )
RETURNING id ;
SQL
end
sql = AccountStat . sanitize_sql ( [ sql , account_id : id , default_value : default_value , value : value ] )
account_stat_id = AccountStat . connection . exec_query ( sql ) [ 0 ] [ 'id' ]
# Reload account_stat if it was loaded, taking into account newly-created unsaved records
if association ( :account_stat ) . loaded?
account_stat . id = account_stat_id if account_stat . new_record?
account_stat . reload
end
end
2018-11-19 08:43:52 +09:00
def account_stat
super || build_account_stat
end
private
def save_account_stat
2019-08-18 21:55:03 +09:00
return unless association ( :account_stat ) . loaded? && account_stat & . changed?
2018-11-19 08:43:52 +09:00
account_stat . save
end
end