Warm tip: This article is reproduced from serverfault.com, please click

其他-Node.js:有没有办法模拟Prometheus Pushgateway服务器?

(其他 - Node.js: Is there a way to mock the prometheus pushgateway server?)

发布于 2020-11-25 21:15:17

我正在使用node.js程序包prom-client尝试将不同的自定义指标推送到Prometheus Pushgateway服务器。

该代码正在运行,现在我正在为该功能编写测试代码。我想知道是否有一种方法可以模拟Prometheus Pushgateway服务器吗?

我曾尝试使用普通的Express API服务器(如下所示)

const express = require('express');
const bodyParser = require('body-parser');

const app = express();

let data = null;

// ROUTES FOR OUR API
// =============================================================================
const router = express.Router();

app.use(bodyParser.urlencoded({ extended: false }));
app.use(bodyParser.json());

router.get('/', (req, res) => {
  console.log("pushgateway server: get to '/' received");
  res.json(data);
});

router.post('/', (req, res) => {
  console.log("pushgateway server: post to '/' received");
  console.log(req.body);
  data = req.body;
  res.send('POST request to the homepage');
})

app.use(router);

// API JSON BASED ERRORS
// =============================================================================
app.use((err, req, res, next) => {
  if (req.xhr) {
    console.error(err);
    res.status(500).send({ err: 'Something failed!' });
  } else {
    next(err);
  }
});
app.use((err, req, res, next) => { // eslint-disable-line
  console.error(err);
  res.status(500).send({ err: err.message });
});

// START THE SERVER
// =============================================================================
const port = 9876;
const server = app.listen(port, '0.0.0.0');
console.log(`Prometheus Pushgateway Mock Server is listening on port ${port}`);

// SHUTDOWN HANDLING
// =============================================================================

// Nodemon Signal Handling
process.once('SIGUSR2', () => {
  console.log('Received kill signal, attempting gracefully close');
  server.close(() => {
    console.log('Closed out remaining connections');
    process.kill(process.pid, 'SIGUSR2');
  });
  setTimeout(() => {
    console.error('Timeout, forcefully shutting down');
    process.kill(process.pid, 'SIGUSR2');
  }, 3000);
});

module.exports = server;

但它不起作用-当我调用gateway.pushAdd()测试代码时,服务器未收到任何发布消息。

谁能给我一些有关如何执行此操作的提示(模拟prometheus pushgateway服务器)?

Questioner
Ken Tsoi
Viewed
11
Ken Tsoi 2020-12-18 22:25:18

因此,我已经通过创建一个简单的http服务器解决了该问题。

以下是服务器的代码:

const http = require('http');

const body = [];
let text = null;
let path = null;

function createServer(port) {
  return http.createServer()
    .on('request', (req, res) => {
      if (req.method === 'POST' || req.method === 'PUT') {
        path = req.url;
        req.on('data', (chunk) => {
          body.push(chunk);
        }).on('end', () => {
          text = Buffer.concat(body).toString();
          res.end(`${req.method} success`);
        }).on('error', (err) => {
          console.error(err);
        });
      } else if (req.method === 'GET') {
        res.end(JSON.stringify({ path, text }));
      } else if (req.method === 'DELETE') {
        path = null;
        text = null;
        res.end('DELETE success');
      }
    }).on('error', (err) => {
      console.log(`Server error: ${err}`);
    })
    .listen(port, '0.0.0.0');
}

module.exports = (createServer);

服务器接受POST / PUT / DELETE / GET请求,以处理网关的pushAdd()/ push()/ delete()函数来处理度量标准数据,并检查模拟的pushgateway上的push数据。

同样,保存到服务器的request.url和文本数据将被保存并传递到测试程序以进行验证。

以下是测试案例之一(使用Mocha + Chai):

describe('Check adding custom push count metric', () => {
  it('Check connection: should return - network status: 200, result contains custom count metric string', async () => {
    metricInstance = promMetric.createCustomPushMetric({
      name: 'test_counter',
      help: 'Used to test the pushgateway for custom counter metrics',
      type: 'counter',
      jobName: 'custom-metric-pushgateway-counter',
      groupings: { key: 'test', type: 'customMetric' },
    });
    await promMetric.addPushMetric(metricInstance, 0.879);
    await sleep(500);
    const result = await chai.request(mockPgwServer).get('/');
    expect(result).to.have.status(200);
    const json = JSON.parse(result.text);
    expect(json.path).to.equal('/metrics/job/custom-metric-pushgateway-counter/key/test/type/customMetric');
    expect(json.text).to.match(/# HELP Push_gateway_test_test_counter Used to test the pushgateway for custom counter metrics/);
    expect(json.text).to.match(/# TYPE Push_gateway_test_test_counter counter/);
    expect(json.text).to.match(/Push_gateway_test_test_counter 0.879/);
  });
});

从客户端发布度量标准时,存储在服务器中的“路径”和“文本”在查询和验证后会发送回客户端。