|
1 | 1 | local metrics = require('metrics')
|
2 | 2 |
|
3 | 3 | local default_metrics = {
|
4 |
| - require('metrics.default_metrics.tarantool.network'), |
5 |
| - require('metrics.default_metrics.tarantool.operations'), |
6 |
| - require('metrics.default_metrics.tarantool.system'), |
7 |
| - require('metrics.default_metrics.tarantool.replicas'), |
8 |
| - require('metrics.default_metrics.tarantool.info'), |
9 |
| - require('metrics.default_metrics.tarantool.slab'), |
10 |
| - require('metrics.default_metrics.tarantool.runtime'), |
11 |
| - require('metrics.default_metrics.tarantool.memory'), |
12 |
| - require('metrics.default_metrics.tarantool.spaces'), |
13 |
| - require('metrics.default_metrics.tarantool.fibers'), |
14 |
| - require('metrics.default_metrics.tarantool.cpu'), |
15 |
| - require('metrics.tarantool.vinyl') |
| 4 | + network = require('metrics.default_metrics.tarantool.network'), |
| 5 | + operations = require('metrics.default_metrics.tarantool.operations'), |
| 6 | + system = require('metrics.default_metrics.tarantool.system'), |
| 7 | + replicas = require('metrics.default_metrics.tarantool.replicas'), |
| 8 | + info = require('metrics.default_metrics.tarantool.info'), |
| 9 | + slab = require('metrics.default_metrics.tarantool.slab'), |
| 10 | + runtime = require('metrics.default_metrics.tarantool.runtime'), |
| 11 | + memory = require('metrics.default_metrics.tarantool.memory'), |
| 12 | + spaces = require('metrics.default_metrics.tarantool.spaces'), |
| 13 | + fibers = require('metrics.default_metrics.tarantool.fibers'), |
| 14 | + cpu = require('metrics.default_metrics.tarantool.cpu'), |
| 15 | + vinyl = require('metrics.tarantool.vinyl'), |
| 16 | + luajit = require('metrics.tarantool.luajit'), |
| 17 | + cartridge_issues = require('metrics.cartridge.issues'), |
| 18 | + clock = require('metrics.cartridge.clock'), |
16 | 19 | }
|
17 | 20 |
|
18 |
| -local function enable() |
19 |
| - for _, metric in ipairs(default_metrics) do |
20 |
| - metrics.register_callback(metric.update) |
| 21 | +local function delete_collectors(list) |
| 22 | + if list == nil then |
| 23 | + return |
| 24 | + end |
| 25 | + for _, collector in pairs(list) do |
| 26 | + metrics.registry:unregister(collector) |
| 27 | + end |
| 28 | + table.clear(list) |
| 29 | +end |
| 30 | + |
| 31 | +local function enable(include, exclude) |
| 32 | + include = include or {} |
| 33 | + exclude = exclude or {} |
| 34 | + if next(include) ~= nil and next(exclude) ~= nil then |
| 35 | + error('Only one of "exclude" or "include" should present') |
| 36 | + end |
| 37 | + |
| 38 | + local exclude_map = {} |
| 39 | + for _, name in ipairs(exclude) do |
| 40 | + exclude_map[name] = true |
| 41 | + end |
| 42 | + local include_map = {} |
| 43 | + for _, name in ipairs(include) do |
| 44 | + include_map[name] = true |
| 45 | + end |
| 46 | + |
| 47 | + for name, value in pairs(default_metrics) do |
| 48 | + if next(include) ~= nil then |
| 49 | + if include_map[name] ~= nil then |
| 50 | + metrics.register_callback(value.update) |
| 51 | + else |
| 52 | + metrics.unregister_callback(value.update) |
| 53 | + delete_collectors(value.list) |
| 54 | + end |
| 55 | + elseif next(exclude) ~= nil then |
| 56 | + if exclude_map[name] ~= nil then |
| 57 | + metrics.unregister_callback(value.update) |
| 58 | + delete_collectors(value.list) |
| 59 | + else |
| 60 | + metrics.register_callback(value.update) |
| 61 | + end |
| 62 | + else |
| 63 | + metrics.register_callback(value.update) |
| 64 | + end |
21 | 65 | end
|
22 | 66 | end
|
23 | 67 |
|
|
0 commit comments