1
0
mirror of https://github.com/actix/examples synced 2024-12-18 00:13:57 +01:00
examples/server-sent-events/benchmark.js
2024-08-07 02:04:57 +01:00

74 lines
1.5 KiB
JavaScript

const http = require("http")
const n = 120
let connected = 0
let messages = 0
let start = Date.now()
let phase = "connecting"
let connection_time
let broadcast_time
let message = process.argv[2] || "msg"
let expected_data = "data: " + message
for (let i = 0; i < n; i++) {
http
.get(
{
host: "127.0.0.1",
port: 8080,
path: "/events",
},
response => {
response.on("data", data => {
if (data.includes(expected_data)) {
messages += 1
} else if (data.includes("data: connected\n")) {
connected += 1
}
})
},
)
.on("error", _ => {})
}
setInterval(() => {
if (phase === "connecting" && connected === n) {
// done connecting
phase = "messaging"
connection_time = Date.now() - start
}
if (phase === "messaging") {
phase = "waiting"
start = Date.now()
http
.request(
{
method: "POST",
host: "127.0.0.1",
port: 8080,
path: "/broadcast/" + message,
},
response => {
response.on("data", _ => {})
},
)
.end()
}
if (phase === "waiting" && messages >= n) {
// all messages received
broadcast_time = Date.now() - start
phase = "paused"
messages = 0
phase = "messaging"
}
process.stdout.write("\r\x1b[K")
process.stdout.write(
`Connected: ${connected}, connection time: ${connection_time} ms, total broadcast time: ${broadcast_time} ms`,
)
}, 20)