VUE_GabenParadise/node_modules/websocket-driver/lib/websocket/driver/stream_reader.js

81 lines
1.8 KiB
JavaScript
Raw Normal View History

2020-08-10 12:35:19 +00:00
'use strict';
var StreamReader = function() {
this._queue = [];
this._queueSize = 0;
this._offset = 0;
};
StreamReader.prototype.put = function(buffer) {
if (!buffer || buffer.length === 0) return;
2020-08-13 08:31:15 +00:00
if (!buffer.copy) buffer = new Buffer(buffer);
2020-08-10 12:35:19 +00:00
this._queue.push(buffer);
this._queueSize += buffer.length;
};
StreamReader.prototype.read = function(length) {
if (length > this._queueSize) return null;
2020-08-13 08:31:15 +00:00
if (length === 0) return new Buffer(0);
2020-08-10 12:35:19 +00:00
this._queueSize -= length;
var queue = this._queue,
remain = length,
first = queue[0],
buffers, buffer;
if (first.length >= length) {
if (first.length === length) {
return queue.shift();
} else {
buffer = first.slice(0, length);
queue[0] = first.slice(length);
return buffer;
}
}
for (var i = 0, n = queue.length; i < n; i++) {
if (remain < queue[i].length) break;
remain -= queue[i].length;
}
buffers = queue.splice(0, i);
if (remain > 0 && queue.length > 0) {
buffers.push(queue[0].slice(0, remain));
queue[0] = queue[0].slice(remain);
}
2020-08-13 08:31:15 +00:00
return this._concat(buffers, length);
2020-08-10 12:35:19 +00:00
};
StreamReader.prototype.eachByte = function(callback, context) {
var buffer, n, index;
while (this._queue.length > 0) {
buffer = this._queue[0];
n = buffer.length;
while (this._offset < n) {
index = this._offset;
this._offset += 1;
callback.call(context, buffer[index]);
}
this._offset = 0;
this._queue.shift();
}
};
2020-08-13 08:31:15 +00:00
StreamReader.prototype._concat = function(buffers, length) {
if (Buffer.concat) return Buffer.concat(buffers, length);
var buffer = new Buffer(length),
offset = 0;
for (var i = 0, n = buffers.length; i < n; i++) {
buffers[i].copy(buffer, offset);
offset += buffers[i].length;
}
return buffer;
};
2020-08-10 12:35:19 +00:00
module.exports = StreamReader;