Class: Rpush::Daemon::Store::ActiveRecord

Inherits:
Object
  • Object
show all
Includes:
Reconnectable
Defined in:
lib/rpush/daemon/store/active_record.rb,
lib/rpush/daemon/store/active_record/reconnectable.rb

Defined Under Namespace

Modules: Reconnectable

Constant Summary collapse

DEFAULT_MARK_OPTIONS = 
{ persist: true }

Constants included from Reconnectable

Reconnectable::ADAPTER_ERRORS

Instance Method Summary collapse

Methods included from Reconnectable

#check_database_is_connected, #database_connection_lost, #reconnect_database, #sleep_to_avoid_thrashing, #with_database_reconnect_and_retry

Constructor Details

#initialize ⇒ ActiveRecord

Returns a new instance of ActiveRecord.



13
14
15
 
# File 'lib/rpush/daemon/store/active_record.rb', line 13

def initialize
  reopen_log unless Rpush.config.embedded
end

Instance Method Details

#adapter_name ⇒ Object 



177
178
179
180
181
182
183
184
185
186
 
# File 'lib/rpush/daemon/store/active_record.rb', line 177

def adapter_name
  env = (defined?(Rails) && Rails.env) ? Rails.env : 'development'
  if ::ActiveRecord::VERSION::MAJOR > 6
    ::ActiveRecord::Base.configurations.configs_for(env_name: env).first.configuration_hash[:adapter]
  else
    config = ::ActiveRecord::Base.configurations[env]
    return '' unless config
    Hash[config.map { |k, v| [k.to_sym, v] }][:adapter]
  end
end

#all_apps ⇒ Object 



25
26
27
 
# File 'lib/rpush/daemon/store/active_record.rb', line 25

def all_apps
  Rpush::Client::ActiveRecord::App.all
end

#app(id) ⇒ Object 



21
22
23
 
# File 'lib/rpush/daemon/store/active_record.rb', line 21

def app(id)
  Rpush::Client::ActiveRecord::App.find(id)
end

#create_adm_notification(attrs, data, registration_ids, deliver_after, app) ⇒ Object 



146
147
148
149
 
# File 'lib/rpush/daemon/store/active_record.rb', line 146

def create_adm_notification(attrs, data, registration_ids, deliver_after, app)
  notification = Rpush::Client::ActiveRecord::Adm::Notification.new
  create_adm_like_notification(notification, attrs, data, registration_ids, deliver_after, app)
end

#create_fcm_notification(attrs, data, app) ⇒ Object 



141
142
143
144
 
# File 'lib/rpush/daemon/store/active_record.rb', line 141

def create_fcm_notification(attrs, data, app)
  notification = Rpush::Client::ActiveRecord::Fcm::Notification.new
  create_fcm_like_notification(notification, attrs, data, app)
end

#deliverable_notifications(limit) ⇒ Object 



29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
 
# File 'lib/rpush/daemon/store/active_record.rb', line 29

def deliverable_notifications(limit)
  with_database_reconnect_and_retry do
    notifications = Rpush::Client::ActiveRecord::Notification.transaction do
      relation = ready_for_delivery
      relation = relation.limit(limit)
      ids = relation.lock(true).ids
      unless ids.empty?
        relation = Rpush::Client::ActiveRecord::Notification.where(id: ids)
        # mark processing
        relation.update_all(processing: true, updated_at: Time.now)
        relation
      else
        []
      end
    end

    notifications.to_a
  end
end

#mark_batch_delivered(notifications) ⇒ Object 



92
93
94
95
96
97
98
99
100
101
102
103
104
 
# File 'lib/rpush/daemon/store/active_record.rb', line 92

def mark_batch_delivered(notifications)
  return if notifications.empty?

  now = Time.now
  ids = []
  notifications.each do |n|
    mark_delivered(n, now, persist: false)
    ids << n.id
  end
  with_database_reconnect_and_retry do
    Rpush::Client::ActiveRecord::Notification.where(id: ids).update_all(['processing = ?, delivered = ?, delivered_at = ?', false, true, now])
  end
end

#mark_batch_failed(notifications, code, description) ⇒ Object 



123
124
125
126
127
128
129
130
131
 
# File 'lib/rpush/daemon/store/active_record.rb', line 123

def mark_batch_failed(notifications, code, description)
  now = Time.now
  ids = []
  notifications.each do |n|
    mark_failed(n, code, description, now, persist: false)
    ids << n.id
  end
  mark_ids_failed(ids, code, description, now)
end

#mark_batch_retryable(notifications, deliver_after) ⇒ Object 



62
63
64
65
66
67
68
69
 
# File 'lib/rpush/daemon/store/active_record.rb', line 62

def mark_batch_retryable(notifications, deliver_after)
  ids = []
  notifications.each do |n|
    mark_retryable(n, deliver_after, persist: false)
    ids << n.id
  end
  mark_ids_retryable(ids, deliver_after)
end

#mark_delivered(notification, time, opts = {}) ⇒ Object 



79
80
81
82
83
84
85
86
87
88
89
90
 
# File 'lib/rpush/daemon/store/active_record.rb', line 79

def mark_delivered(notification, time, opts = {})
  opts = DEFAULT_MARK_OPTIONS.dup.merge(opts)
  notification.processing = false
  notification.delivered = true
  notification.delivered_at = time

  return unless opts[:persist]

  with_database_reconnect_and_retry do
    notification.save!(validate: false)
  end
end

#mark_failed(notification, code, description, time, opts = {}) ⇒ Object 



106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
 
# File 'lib/rpush/daemon/store/active_record.rb', line 106

def mark_failed(notification, code, description, time, opts = {})
  opts = DEFAULT_MARK_OPTIONS.dup.merge(opts)
  notification.processing = false
  notification.delivered = false
  notification.delivered_at = nil
  notification.failed = true
  notification.failed_at = time
  notification.error_code = code
  notification.error_description = description

  return unless opts[:persist]

  with_database_reconnect_and_retry do
    notification.save!(validate: false)
  end
end

#mark_ids_failed(ids, code, description, time) ⇒ Object 



133
134
135
136
137
138
139
 
# File 'lib/rpush/daemon/store/active_record.rb', line 133

def mark_ids_failed(ids, code, description, time)
  return if ids.empty?

  with_database_reconnect_and_retry do
    Rpush::Client::ActiveRecord::Notification.where(id: ids).update_all(['processing = ?, delivered = ?, delivered_at = NULL, failed = ?, failed_at = ?, error_code = ?, error_description = ?', false, false, true, time, code, description])
  end
end

#mark_ids_retryable(ids, deliver_after) ⇒ Object 



71
72
73
74
75
76
77
 
# File 'lib/rpush/daemon/store/active_record.rb', line 71

def mark_ids_retryable(ids, deliver_after)
  return if ids.empty?

  with_database_reconnect_and_retry do
    Rpush::Client::ActiveRecord::Notification.where(id: ids).update_all(['processing = ?, delivered = ?, delivered_at = ?, failed = ?, failed_at = ?, retries = retries + 1, deliver_after = ?', false, false, nil, false, nil, deliver_after])
  end
end

#mark_retryable(notification, deliver_after, opts = {}) ⇒ Object 



49
50
51
52
53
54
55
56
57
58
59
60
 
# File 'lib/rpush/daemon/store/active_record.rb', line 49

def mark_retryable(notification, deliver_after, opts = {})
  opts = DEFAULT_MARK_OPTIONS.dup.merge(opts)
  notification.processing = false
  notification.retries += 1
  notification.deliver_after = deliver_after

  return unless opts[:persist]

  with_database_reconnect_and_retry do
    notification.save!(validate: false)
  end
end

#pending_delivery_count ⇒ Object 



169
170
171
 
# File 'lib/rpush/daemon/store/active_record.rb', line 169

def pending_delivery_count
  ready_for_delivery.count
end

#release_connection ⇒ Object 



163
164
165
166
167
 
# File 'lib/rpush/daemon/store/active_record.rb', line 163

def release_connection
  ::ActiveRecord::Base.connection.close
rescue StandardError => e
  Rpush.logger.error(e)
end

#reopen_log ⇒ Object 



17
18
19
 
# File 'lib/rpush/daemon/store/active_record.rb', line 17

def reopen_log
  ::ActiveRecord::Base.logger = Rpush.logger.internal_logger
end

#translate_integer_notification_id(id) ⇒ Object 



173
174
175
 
# File 'lib/rpush/daemon/store/active_record.rb', line 173

def translate_integer_notification_id(id)
  id
end

#update_app(app) ⇒ Object 



151
152
153
154
155
 
# File 'lib/rpush/daemon/store/active_record.rb', line 151

def update_app(app)
  with_database_reconnect_and_retry do
    app.save!
  end
end

#update_notification(notification) ⇒ Object 



157
158
159
160
161
 
# File 'lib/rpush/daemon/store/active_record.rb', line 157

def update_notification(notification)
  with_database_reconnect_and_retry do
    notification.save!
  end
end