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
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
|
# Copyright (C) 2015 MongoDB, Inc.
#
# Licensed under the Apache License, Version 2.0 (the 'License');
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an 'AS IS' BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
module Mongo
class Monitoring
# Defines behaviour for an object that can publish monitoring events.
#
# @since 2.1.0
module Publishable
# @return [ Monitoring ] monitoring The monitoring.
attr_reader :monitoring
# Publish a command event to the global monitoring.
#
# @example Publish a command event.
# publish_command do |messages|
# # ...
# end
#
# @param [ Array<Message> ] messages The messages.
#
# @return [ Object ] The result of the yield.
#
# @since 2.1.0
def publish_command(messages, operation_id = Monitoring.next_operation_id)
start = Time.now
message = messages.first
message.set_request_id
payload = message.payload
send_duration = duration(start)
command_started(address, operation_id, payload)
receive_start = Time.now
begin
result = yield(messages)
total_duration = duration(receive_start) + send_duration
command_completed(result, address, operation_id, payload, total_duration)
result
rescue Exception => e
total_duration = duration(receive_start) + send_duration
command_failed(address, operation_id, payload, e.message, total_duration)
raise e
end
end
def publish_event(topic, event)
monitoring.succeeded(topic, event)
end
def publish_sdam_event(topic, event)
monitoring.succeeded(topic, event) if monitoring?
end
private
def command_started(address, operation_id, payload)
monitoring.started(
Monitoring::COMMAND,
Event::CommandStarted.generate(address, operation_id, payload)
)
end
def command_completed(result, address, operation_id, payload, duration)
document = result ? (result.documents || []).first : nil
if error?(document)
parser = Error::Parser.new(document)
command_failed(address, operation_id, payload, parser.message, duration)
else
command_succeeded(result, address, operation_id, payload, duration)
end
end
def command_succeeded(result, address, operation_id, payload, duration)
monitoring.succeeded(
Monitoring::COMMAND,
Event::CommandSucceeded.generate(
address,
operation_id,
payload,
result ? result.payload : nil,
duration
)
)
end
def command_failed(address, operation_id, payload, message, duration)
monitoring.failed(
Monitoring::COMMAND,
Event::CommandFailed.generate(address, operation_id, payload, message, duration)
)
end
def duration(start)
Time.now - start
end
def error?(document)
document && (document['ok'] == 0 || document.key?('$err'))
end
def monitoring?
options[:monitoring] != false
end
end
end
end
|
