mirror of
https://github.com/yu-i-i/overleaf-cep.git
synced 2026-05-23 17:19:37 +02:00
ad79c85cea
* [metrics] mongo: fail when command monitoring is not available * [metrics] mongo: add optional client label to pool metrics * [web] collect mongo stats on native client * [metrics] mongo: record namespace of find commands * [metrics] mongo: add counter for all the commands with collection label * [web] add missing mock GitOrigin-RevId: 9f378d8aa8d7167f56cf512681d63ef115c6dd98
108 lines
3.3 KiB
JavaScript
108 lines
3.3 KiB
JavaScript
const Metrics = require('../../..')
|
|
|
|
const { expect } = require('chai')
|
|
const prom = require('prom-client')
|
|
|
|
describe('mongodb', function () {
|
|
beforeEach(function () {
|
|
prom.register.clear()
|
|
this.pool = {
|
|
totalConnectionCount: 8,
|
|
availableConnectionCount: 2,
|
|
waitQueueSize: 4,
|
|
options: { maxPoolSize: 10 },
|
|
}
|
|
this.servers = new Map([['server1', { s: { pool: this.pool } }]])
|
|
|
|
this.mongoClient = {
|
|
on() {},
|
|
topology: { s: { servers: this.servers } },
|
|
}
|
|
Metrics.mongodb.reset()
|
|
})
|
|
|
|
it('handles an unconnected client', async function () {
|
|
const mongoClient = { on() {} }
|
|
Metrics.mongodb.monitor(mongoClient)
|
|
const metrics = await getMetrics()
|
|
expect(metrics).to.deep.equal({})
|
|
})
|
|
|
|
it('collects Mongo metrics', async function () {
|
|
Metrics.mongodb.monitor(this.mongoClient)
|
|
const metrics = await getMetrics()
|
|
expect(metrics).to.deep.equal({
|
|
'mongo_connection_pool_max:server1': 10,
|
|
'mongo_connection_pool_size:server1': 8,
|
|
'mongo_connection_pool_available:server1': 2,
|
|
'mongo_connection_pool_waiting:server1': 4,
|
|
})
|
|
})
|
|
|
|
it('collects Mongo metrics with client', async function () {
|
|
Metrics.mongodb.monitor(this.mongoClient, 'native')
|
|
const metrics = await getMetrics()
|
|
expect(metrics).to.deep.equal({
|
|
'mongo_connection_pool_max:server1:native': 10,
|
|
'mongo_connection_pool_size:server1:native': 8,
|
|
'mongo_connection_pool_available:server1:native': 2,
|
|
'mongo_connection_pool_waiting:server1:native': 4,
|
|
})
|
|
})
|
|
|
|
it('handles topology changes', async function () {
|
|
Metrics.mongodb.monitor(this.mongoClient)
|
|
let metrics = await getMetrics()
|
|
expect(metrics).to.deep.equal({
|
|
'mongo_connection_pool_max:server1': 10,
|
|
'mongo_connection_pool_size:server1': 8,
|
|
'mongo_connection_pool_available:server1': 2,
|
|
'mongo_connection_pool_waiting:server1': 4,
|
|
})
|
|
|
|
// Add a server
|
|
this.servers.set('server2', this.servers.get('server1'))
|
|
metrics = await getMetrics()
|
|
expect(metrics).to.deep.equal({
|
|
'mongo_connection_pool_max:server1': 10,
|
|
'mongo_connection_pool_size:server1': 8,
|
|
'mongo_connection_pool_available:server1': 2,
|
|
'mongo_connection_pool_waiting:server1': 4,
|
|
'mongo_connection_pool_max:server2': 10,
|
|
'mongo_connection_pool_size:server2': 8,
|
|
'mongo_connection_pool_available:server2': 2,
|
|
'mongo_connection_pool_waiting:server2': 4,
|
|
})
|
|
|
|
// Delete a server
|
|
this.servers.delete('server1')
|
|
metrics = await getMetrics()
|
|
expect(metrics).to.deep.equal({
|
|
'mongo_connection_pool_max:server2': 10,
|
|
'mongo_connection_pool_size:server2': 8,
|
|
'mongo_connection_pool_available:server2': 2,
|
|
'mongo_connection_pool_waiting:server2': 4,
|
|
})
|
|
|
|
// Delete another server
|
|
this.servers.delete('server2')
|
|
metrics = await getMetrics()
|
|
expect(metrics).to.deep.equal({})
|
|
})
|
|
})
|
|
|
|
async function getMetrics() {
|
|
const metrics = await prom.register.getMetricsAsJSON()
|
|
const result = {}
|
|
for (const metric of metrics) {
|
|
for (const value of metric.values) {
|
|
let key = `${metric.name}:${value.labels.mongo_server}`
|
|
if (value.labels.client) {
|
|
key = `${key}:${value.labels.client}`
|
|
}
|
|
result[key] = value.value
|
|
}
|
|
}
|
|
return result
|
|
}
|