105 lines
2.2 KiB
JavaScript
105 lines
2.2 KiB
JavaScript
'use strict'
|
|
|
|
const { WebsocketFrameSend } = require('./frame')
|
|
const { opcodes, sendHints } = require('./constants')
|
|
const FixedQueue = require('../../dispatcher/fixed-queue')
|
|
|
|
/** @type {typeof Uint8Array} */
|
|
const FastBuffer = Buffer[Symbol.species]
|
|
|
|
/**
|
|
* @typedef {object} SendQueueNode
|
|
* @property {Promise<void> | null} promise
|
|
* @property {((...args: any[]) => any)} callback
|
|
* @property {Buffer | null} frame
|
|
*/
|
|
|
|
class SendQueue {
|
|
/**
|
|
* @type {FixedQueue}
|
|
*/
|
|
#queue = new FixedQueue()
|
|
|
|
/**
|
|
* @type {boolean}
|
|
*/
|
|
#running = false
|
|
|
|
/** @type {import('node:net').Socket} */
|
|
#socket
|
|
|
|
constructor (socket) {
|
|
this.#socket = socket
|
|
}
|
|
|
|
add (item, cb, hint) {
|
|
if (hint !== sendHints.blob) {
|
|
const frame = createFrame(item, hint)
|
|
if (!this.#running) {
|
|
// fast-path
|
|
this.#socket.write(frame, cb)
|
|
} else {
|
|
/** @type {SendQueueNode} */
|
|
const node = {
|
|
promise: null,
|
|
callback: cb,
|
|
frame
|
|
}
|
|
this.#queue.push(node)
|
|
}
|
|
return
|
|
}
|
|
|
|
/** @type {SendQueueNode} */
|
|
const node = {
|
|
promise: item.arrayBuffer().then((ab) => {
|
|
node.promise = null
|
|
node.frame = createFrame(ab, hint)
|
|
}),
|
|
callback: cb,
|
|
frame: null
|
|
}
|
|
|
|
this.#queue.push(node)
|
|
|
|
if (!this.#running) {
|
|
this.#run()
|
|
}
|
|
}
|
|
|
|
async #run () {
|
|
this.#running = true
|
|
const queue = this.#queue
|
|
while (!queue.isEmpty()) {
|
|
const node = queue.shift()
|
|
// wait pending promise
|
|
if (node.promise !== null) {
|
|
await node.promise
|
|
}
|
|
// write
|
|
this.#socket.write(node.frame, node.callback)
|
|
// cleanup
|
|
node.callback = node.frame = null
|
|
}
|
|
this.#running = false
|
|
}
|
|
}
|
|
|
|
function createFrame (data, hint) {
|
|
return new WebsocketFrameSend(toBuffer(data, hint)).createFrame(hint === sendHints.string ? opcodes.TEXT : opcodes.BINARY)
|
|
}
|
|
|
|
function toBuffer (data, hint) {
|
|
switch (hint) {
|
|
case sendHints.string:
|
|
return Buffer.from(data)
|
|
case sendHints.arrayBuffer:
|
|
case sendHints.blob:
|
|
return new FastBuffer(data)
|
|
case sendHints.typedArray:
|
|
return new FastBuffer(data.buffer, data.byteOffset, data.byteLength)
|
|
}
|
|
}
|
|
|
|
module.exports = { SendQueue }
|