forked from open-telemetry/opentelemetry-js-contrib
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathserver.js
54 lines (49 loc) · 1.42 KB
/
server.js
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
'use strict';
// eslint-disable-next-line import/order
const tracer = require('./tracer')('postgres-server-service');
const { SpanKind, CanonicalCode } = require('@opentelemetry/api');
const express = require('express');
const setupPg = require('./setupPsql');
const pool = setupPg.startPsql();
const app = express();
app.get('/:cmd', (req, res) => {
const cmd = req.path.slice(1);
if (!req.query.id) {
res.status(400).send('No id provided');
return;
}
let queryText = `SELECT id, text FROM test WHERE id = ${req.query.id}`;
if (cmd === 'insert') {
if (!req.query.text) {
res.status(400).send('No text provided');
return;
}
queryText = {
text: 'INSERT INTO test (id, text) VALUES($1, $2) ON CONFLICT(id) DO UPDATE SET text=$2',
values: [req.query.id, req.query.text],
};
}
const currentSpan = tracer.getCurrentSpan();
console.log(`traceid: ${currentSpan.context().traceId}`);
const span = tracer.startSpan(cmd, {
parent: currentSpan,
kind: SpanKind.SERVER,
});
tracer.withSpan(span, () => {
try {
pool.query(queryText, (err, ret) => {
if (err) throw err;
res.send(ret.rows);
});
} catch (e) {
res.status(400).send({ message: e.message });
span.setStatus(CanonicalCode.UNKNOWN);
}
span.end();
});
});
// start server
const port = 3000;
app.listen(port, () => {
console.log(`Node HTTP listening on ${port}`);
});