Newer
Older
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
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
import client, { register } from "prom-client";
import { prisma } from "./prisma";
import e from "express";
import { SocketServer } from "./SocketServer";
import Canvas from "./Canvas";
import { Redis } from "./redis";
client.collectDefaultMetrics({
labels: process.env.NODE_APP_INSTANCE
? {
NODE_APP_INSTANCE: process.env.NODE_APP_INSTANCE,
}
: {},
});
export const PixelCount = new client.Gauge({
name: "pixel_count",
help: "total pixel count",
async collect() {
this.set(await prisma.pixel.count());
},
});
export const UserCount = new client.Gauge({
name: "user_count",
help: "total user count",
async collect() {
this.set(await prisma.user.count());
},
});
export const OnlineUsers = new client.Gauge({
name: "connected_count",
help: "total connected sockets",
async collect() {
this.set((await SocketServer.instance.io.fetchSockets()).length);
},
});
/**
* Rough estimate of empty pixels
*/
export const EmptyPixels = new client.Gauge({
name: "empty_pixels",
help: "total number of empty pixels",
async collect() {
let queries: Promise<boolean>[] = [];
for (let x = 0; x < Canvas.getCanvasConfig().size[0]; x++) {
for (let y = 0; y < Canvas.getCanvasConfig().size[1]; y++) {
queries.push(Canvas.isPixelEmpty(x, y));
}
}
let count = 0;
const allSettled = await Promise.allSettled(queries);
for (const settle of allSettled) {
if (settle.status === "fulfilled") {
count += Number(settle.value);
} else {
count++;
}
}
this.set(count);
},
});
export const TotalPixels = new client.Gauge({
name: "total_pixels",
help: "total number of pixels the canvas allows",
async collect() {
const [width, height] = Canvas.getCanvasConfig().size;
this.set(width * height);
},
});
export const UniqueInstances = new client.Gauge({
name: "instance_count",
help: "total number of unique instances",
async collect() {
this.set(await prisma.instance.count());
},
});
export const handleMetricsEndpoint = async (
req: e.Request,
res: e.Response
) => {
if (!process.env.PROMETHEUS_TOKEN) {
res.status(500);
res.send("PROMETHEUS_TOKEN is not set.");
return;
}
if (req.headers.authorization !== "Bearer " + process.env.PROMETHEUS_TOKEN) {
res.status(401);
res.send("Invalid bearer token");
return;
}
res.setHeader("Content-Type", register.contentType);
res.send(await register.metrics());
res.end();
};