postgrejs
Version:
Professional PostgreSQL client NodeJS
179 lines (178 loc) • 5.83 kB
JavaScript
"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
exports.SmartBuffer = void 0;
const tslib_1 = require("tslib");
const os = tslib_1.__importStar(require("os"));
const bigint_methods_js_1 = require("../util/bigint-methods.js");
const buffer_reader_js_1 = require("./buffer-reader.js");
class SmartBuffer extends buffer_reader_js_1.BufferReader {
constructor(cfg) {
super(Buffer.allocUnsafe(cfg?.pageSize || SmartBuffer.DEFAULT_PAGE_SIZE));
this._lastHouseKeep = 0;
this._stMaxPages = 1;
this._length = 0;
this._houseKeepInterval = cfg?.houseKeepInterval || 5000;
this.pageSize = this.buffer.length;
this.maxSize = cfg?.maxLength || SmartBuffer.DEFAULT_MAX_SIZE;
this._length = 0;
}
get capacity() {
return this.buffer.length;
}
get length() {
return this._length;
}
start() {
this.offset = 0;
this._length = 0;
if (this._houseKeepTimer) {
clearTimeout(this._houseKeepTimer);
this._houseKeepTimer = undefined;
}
return this;
}
flush() {
if (this._houseKeepTimer)
clearTimeout(this._houseKeepTimer);
const length = this.length;
this._length = 0;
const out = this.buffer.slice(0, length);
const pages = length ? Math.ceil(length / this.pageSize) : 1;
this._stMaxPages = Math.max(this._stMaxPages, pages);
if (this._lastHouseKeep < Date.now() + this._houseKeepInterval)
this._houseKeep();
this._houseKeepTimer = setTimeout(() => {
this._houseKeepTimer = undefined;
this._houseKeep();
}, this._houseKeepInterval).unref();
return out;
}
growSize(len) {
const endOffset = this.offset + len;
if (this.capacity < endOffset) {
if (endOffset > this.maxSize)
throw new Error('Buffer limit exceeded.');
const newSize = Math.ceil(endOffset / this.pageSize) * this.pageSize;
const newBuffer = Buffer.allocUnsafe(newSize);
this.buffer.copy(newBuffer);
this.buffer = newBuffer;
}
this._length = Math.max(this.length, endOffset);
return this;
}
fill(value = 0, len = 1) {
this.ensureSize(len);
this.buffer.fill(value, this.offset, this.offset + len);
this.offset += len;
return this;
}
writeCString(str, encoding) {
const len = str ? Buffer.byteLength(str, encoding) : 0;
this.ensureSize(len + 1);
if (str) {
this.buffer.write(str, this.offset, encoding);
this.offset += len;
}
this.writeUInt8(0);
return this;
}
writeLString(str, encoding) {
const len = str ? Buffer.byteLength(str, encoding) : 0;
this.ensureSize(len + 4);
this.writeInt32BE(str == null ? -1 : len);
if (str) {
if (encoding)
this.offset += this.buffer.write(str, this.offset, encoding);
else
this.offset += this.buffer.write(str, this.offset);
}
return this;
}
writeString(str, encoding) {
if (str) {
const len = Buffer.byteLength(str, encoding);
this.ensureSize(len);
this.offset += this.buffer.write(str, this.offset, encoding);
}
return this;
}
writeInt8(n) {
this.ensureSize(1);
this.buffer.writeInt8(n, this.offset);
this.offset++;
return this;
}
writeUInt8(n) {
this.ensureSize(1);
this.buffer.writeUInt8(n, this.offset);
this.offset++;
return this;
}
writeUInt16BE(n) {
this.ensureSize(2);
this.buffer.writeUInt16BE(n, this.offset);
this.offset += 2;
return this;
}
writeUInt32BE(n) {
this.ensureSize(4);
this.buffer.writeUInt32BE(n, this.offset);
this.offset += 4;
return this;
}
writeInt16BE(n) {
this.ensureSize(2);
this.buffer.writeInt16BE(n, this.offset);
this.offset += 2;
return this;
}
writeInt32BE(n) {
this.ensureSize(4);
this.buffer.writeInt32BE(n, this.offset);
this.offset += 4;
return this;
}
writeBigInt64BE(n) {
n = typeof n === 'bigint' ? n : BigInt(n);
this.ensureSize(8);
if (typeof this.buffer.writeBigInt64BE === 'function')
this.buffer.writeBigInt64BE(n, this.offset);
else
(0, bigint_methods_js_1.writeBigUInt64BE)(this.buffer, n, this.offset);
this.offset += 8;
return this;
}
writeFloatBE(n) {
this.ensureSize(4);
this.buffer.writeFloatBE(n, this.offset);
this.offset += 4;
return this;
}
writeDoubleBE(n) {
this.ensureSize(8);
this.buffer.writeDoubleBE(n, this.offset);
this.offset += 8;
return this;
}
writeBuffer(buffer) {
this.ensureSize(buffer.length);
buffer.copy(this.buffer, this.offset, 0, buffer.length);
this.offset += buffer.length;
return this;
}
ensureSize(len) {
const n = this.offset + len - this.length;
if (n > 0)
this.growSize(n);
return this;
}
_houseKeep() {
const needSize = this._stMaxPages * this.pageSize;
if (this.buffer.length > needSize)
this.buffer = Buffer.allocUnsafe(needSize);
this._stMaxPages = this.length ? Math.ceil(this.length / this.pageSize) : 1;
}
}
exports.SmartBuffer = SmartBuffer;
SmartBuffer.DEFAULT_PAGE_SIZE = 4096;
SmartBuffer.DEFAULT_MAX_SIZE = Math.min(Math.floor(os.totalmem() / 2), 1024 * 1024 * 1024 * 2);