-
Notifications
You must be signed in to change notification settings - Fork 8.2k
/
telemetry_ui_metric_collector.ts
82 lines (70 loc) · 2.25 KB
/
telemetry_ui_metric_collector.ts
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
/*
* Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
* or more contributor license agreements. Licensed under the Elastic License
* 2.0 and the Server Side Public License, v 1; you may not use this file except
* in compliance with, at your election, the Elastic License 2.0 or the Server
* Side Public License, v 1.
*/
import {
ISavedObjectsRepository,
SavedObjectAttributes,
SavedObjectsServiceSetup,
} from 'kibana/server';
import { UsageCollectionSetup } from 'src/plugins/usage_collection/server';
import { uiMetricSchema } from './schema';
interface UIMetricsSavedObjects extends SavedObjectAttributes {
count: number;
}
interface UIMetricElement {
key: string;
value: number;
}
export type UIMetricUsage = Record<string, UIMetricElement[]>;
export function registerUiMetricUsageCollector(
usageCollection: UsageCollectionSetup,
registerType: SavedObjectsServiceSetup['registerType'],
getSavedObjectsClient: () => ISavedObjectsRepository | undefined
) {
registerType({
name: 'ui-metric',
hidden: false,
namespaceType: 'agnostic',
mappings: {
properties: {
count: {
type: 'integer',
},
},
},
});
const collector = usageCollection.makeUsageCollector<UIMetricUsage | undefined>({
type: 'ui_metric',
schema: uiMetricSchema,
fetch: async () => {
const savedObjectsClient = getSavedObjectsClient();
if (typeof savedObjectsClient === 'undefined') {
return;
}
const { saved_objects: rawUiMetrics } = await savedObjectsClient.find<UIMetricsSavedObjects>({
type: 'ui-metric',
fields: ['count'],
perPage: 10000,
});
const uiMetricsByAppName = rawUiMetrics.reduce((accum, rawUiMetric) => {
const {
id,
attributes: { count },
} = rawUiMetric;
const [appName, ...metricType] = id.split(':');
const pair = { key: metricType.join(':'), value: count };
return {
...accum,
[appName]: [...(accum[appName] || []), pair],
};
}, {} as UIMetricUsage);
return uiMetricsByAppName;
},
isReady: () => typeof getSavedObjectsClient() !== 'undefined',
});
usageCollection.registerCollector(collector);
}