Class: Rpush::Daemon::Store::ActiveRecord
- Inherits:
-
Object
- Object
- Rpush::Daemon::Store::ActiveRecord
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 }
Reconnectable::ADAPTER_ERRORS
Instance Method Summary
collapse
-
#all_apps ⇒ Object
-
#app(id) ⇒ Object
-
#create_adm_notification(attrs, data, registration_ids, deliver_after, app) ⇒ Object
-
#create_apns_feedback(failed_at, device_token, app) ⇒ Object
-
#create_gcm_notification(attrs, data, registration_ids, deliver_after, app) ⇒ Object
-
#deliverable_notifications(limit) ⇒ Object
-
#initialize ⇒ ActiveRecord
constructor
A new instance of ActiveRecord.
-
#mark_batch_delivered(notifications) ⇒ Object
-
#mark_batch_failed(notifications, code, description) ⇒ Object
-
#mark_batch_retryable(notifications, deliver_after) ⇒ Object
-
#mark_delivered(notification, time, opts = {}) ⇒ Object
-
#mark_failed(notification, code, description, time, opts = {}) ⇒ Object
-
#mark_ids_failed(ids, code, description, time) ⇒ Object
-
#mark_ids_retryable(ids, deliver_after) ⇒ Object
-
#mark_retryable(notification, deliver_after, opts = {}) ⇒ Object
-
#pending_delivery_count ⇒ Object
-
#release_connection ⇒ Object
-
#reopen_log ⇒ Object
-
#translate_integer_notification_id(id) ⇒ Object
-
#update_app(app) ⇒ Object
-
#update_notification(notification) ⇒ Object
#check_database_is_connected, #database_connection_lost, #reconnect_database, #sleep_to_avoid_thrashing, #with_database_reconnect_and_retry
Constructor Details
Returns a new instance of ActiveRecord.
13
14
15
16
|
# File 'lib/rpush/daemon/store/active_record.rb', line 13
def initialize
@using_oracle = adapter_name =~ /oracle/
reopen_log unless Rpush.config.embedded
end
|
Instance Method Details
#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_gcm_like_notification(notification, attrs, data, registration_ids, deliver_after, app)
end
|
#create_apns_feedback(failed_at, device_token, app) ⇒ Object
134
135
136
137
138
139
|
# File 'lib/rpush/daemon/store/active_record.rb', line 134
def create_apns_feedback(failed_at, device_token, app)
with_database_reconnect_and_retry do
Rpush::Client::ActiveRecord::Apns::Feedback.create!(failed_at: failed_at,
device_token: device_token, app_id: app.id)
end
end
|
#create_gcm_notification(attrs, data, registration_ids, deliver_after, app) ⇒ Object
141
142
143
144
|
# File 'lib/rpush/daemon/store/active_record.rb', line 141
def create_gcm_notification(attrs, data, registration_ids, deliver_after, app)
notification = Rpush::Client::ActiveRecord::Gcm::Notification.new
create_gcm_like_notification(notification, attrs, data, registration_ids, deliver_after, app)
end
|
#deliverable_notifications(limit) ⇒ Object
30
31
32
33
34
35
36
37
38
39
40
|
# File 'lib/rpush/daemon/store/active_record.rb', line 30
def deliverable_notifications(limit)
with_database_reconnect_and_retry do
Rpush::Client::ActiveRecord::Notification.transaction do
relation = ready_for_delivery
relation = relation.limit(limit)
notifications = claim(relation)
mark_processing(notifications)
notifications
end
end
end
|
#mark_batch_delivered(notifications) ⇒ Object
85
86
87
88
89
90
91
92
93
94
95
96
97
|
# File 'lib/rpush/daemon/store/active_record.rb', line 85
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
116
117
118
119
120
121
122
123
124
|
# File 'lib/rpush/daemon/store/active_record.rb', line 116
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
55
56
57
58
59
60
61
62
|
# File 'lib/rpush/daemon/store/active_record.rb', line 55
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
72
73
74
75
76
77
78
79
80
81
82
83
|
# File 'lib/rpush/daemon/store/active_record.rb', line 72
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
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
|
# File 'lib/rpush/daemon/store/active_record.rb', line 99
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
126
127
128
129
130
131
132
|
# File 'lib/rpush/daemon/store/active_record.rb', line 126
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
64
65
66
67
68
69
70
|
# File 'lib/rpush/daemon/store/active_record.rb', line 64
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
42
43
44
45
46
47
48
49
50
51
52
53
|
# File 'lib/rpush/daemon/store/active_record.rb', line 42
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
18
19
20
|
# File 'lib/rpush/daemon/store/active_record.rb', line 18
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
|