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
|
% 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(mango_execution_stats).
-export([
to_json/1,
to_map/1,
incr_keys_examined/1,
incr_docs_examined/1,
incr_docs_examined/2,
incr_quorum_docs_examined/1,
incr_results_returned/1,
log_start/1,
log_end/1,
log_stats/1,
maybe_add_stats/4
]).
-include("mango_cursor.hrl").
to_json(Stats) ->
{[
{total_keys_examined, Stats#execution_stats.totalKeysExamined},
{total_docs_examined, Stats#execution_stats.totalDocsExamined},
{total_quorum_docs_examined, Stats#execution_stats.totalQuorumDocsExamined},
{results_returned, Stats#execution_stats.resultsReturned},
{execution_time_ms, Stats#execution_stats.executionTimeMs}
]}.
to_map(Stats) ->
#{
total_keys_examined => Stats#execution_stats.totalKeysExamined,
total_docs_examined => Stats#execution_stats.totalDocsExamined,
total_quorum_docs_examined => Stats#execution_stats.totalQuorumDocsExamined,
results_returned => Stats#execution_stats.resultsReturned,
execution_time_ms => Stats#execution_stats.executionTimeMs
}.
incr_keys_examined(Stats) ->
Stats#execution_stats{
totalKeysExamined = Stats#execution_stats.totalKeysExamined + 1
}.
incr_docs_examined(Stats) ->
incr_docs_examined(Stats, 1).
incr_docs_examined(Stats, N) ->
Stats#execution_stats{
totalDocsExamined = Stats#execution_stats.totalDocsExamined + N
}.
incr_quorum_docs_examined(Stats) ->
Stats#execution_stats{
totalQuorumDocsExamined = Stats#execution_stats.totalQuorumDocsExamined + 1
}.
incr_results_returned(Stats) ->
couch_stats:increment_counter([mango, results_returned]),
Stats#execution_stats{
resultsReturned = Stats#execution_stats.resultsReturned + 1
}.
log_start(Stats) ->
Stats#execution_stats{
executionStartTime = os:timestamp()
}.
log_end(Stats) ->
End = os:timestamp(),
Diff = timer:now_diff(End, Stats#execution_stats.executionStartTime) / 1000,
Stats#execution_stats{
executionTimeMs = Diff
}.
maybe_add_stats(Opts, UserFun, Stats0, UserAcc) ->
Stats1 = log_end(Stats0),
couch_stats:update_histogram([mango, query_time], Stats1#execution_stats.executionTimeMs),
%% TODO: add rows read when we collect the stats
%% TODO: add docs vs quorum docs
chttpd_stats:incr_reads(Stats1#execution_stats.totalDocsExamined),
FinalAcc =
case couch_util:get_value(execution_stats, Opts) of
true ->
JSONValue = to_json(Stats1),
Arg = {add_key, execution_stats, JSONValue},
{_Go, FinalUserAcc} = UserFun(Arg, UserAcc),
FinalUserAcc;
_ ->
UserAcc
end,
{FinalAcc, Stats1}.
log_stats(Stats) ->
MStats0 = to_map(Stats),
Nonce = list_to_binary(couch_log_util:get_msg_id()),
MStats1 = MStats0#{nonce => Nonce},
couch_log:report("mango-stats", MStats1).
|
