protoobject
Version:
A universal class for creating any JSON objects and simple manipulations with them.
152 lines (151 loc) • 6.35 kB
JavaScript
;
/**
* ProtoObjectStream - Streaming utilities for ProtoObject using @sergdudko/objectstream
* @description Provides methods for efficient streaming serialization and deserialization
* @author Siarhei Dudko <siarhei@dudko.dev>
*/
var __importDefault = (this && this.__importDefault) || function (mod) {
return (mod && mod.__esModule) ? mod : { "default": mod };
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.ProtoObjectStream = void 0;
/* eslint-disable no-unused-vars */
const node_stream_1 = require("node:stream");
const objectstream_1 = __importDefault(require("@sergdudko/objectstream"));
const { Stringifer, Parser } = objectstream_1.default;
/**
* Utility class for streaming ProtoObject serialization using @sergdudko/objectstream
*/
class ProtoObjectStream {
/**
* Create a transform stream that converts ProtoObject instances to Buffer
* @returns Transform stream for object-to-buffer conversion
*/
static createObjectToBufferStream() {
return new node_stream_1.Transform({
objectMode: true,
transform(chunk, _encoding, callback) {
try {
// Convert ProtoObject to plain object if it has toJSON method
const plainObject = chunk && typeof chunk.toJSON === "function"
? chunk.toJSON()
: chunk;
const stringifer = new Stringifer();
let result = "";
stringifer.on("data", (data) => {
result += data.toString();
});
stringifer.on("end", () => {
callback(null, Buffer.from(result));
});
stringifer.on("error", (error) => {
callback(error);
});
stringifer.write(plainObject);
stringifer.end();
}
catch (error) {
callback(error instanceof Error ? error : new Error(String(error)));
}
},
});
}
/**
* Create a transform stream that converts Buffer to ProtoObject instances
* @param objectClass - The ProtoObject class constructor
* @returns Transform stream for buffer-to-object conversion
*/
static createBufferToObjectStream(objectClass) {
return new node_stream_1.Transform({
objectMode: true,
transform(chunk, encoding, callback) {
try {
const parser = new Parser();
parser.on("data", (obj) => {
// Convert plain object back to ProtoObject instance if fromJSON exists
const result = objectClass && typeof objectClass.fromJSON === "function"
? objectClass.fromJSON(obj)
: new objectClass(obj);
callback(null, result);
});
parser.on("error", (error) => {
callback(error);
});
parser.write(chunk);
parser.end();
}
catch (error) {
callback(error instanceof Error ? error : new Error(String(error)));
}
},
});
}
/**
* Create a complete serialization pipeline
* @param objectClass - The ProtoObject class constructor
* @returns Object with serialize and deserialize transforms
*/
static createSerializationPipeline(objectClass) {
return {
serialize: this.createObjectToBufferStream(),
deserialize: this.createBufferToObjectStream(objectClass),
};
}
/**
* Create a bidirectional stream for two-way communication
* @param objectClass - The ProtoObject class constructor
* @returns Transform stream for bidirectional communication
*/
static createBidirectionalStream(objectClass) {
return new node_stream_1.Transform({
objectMode: true,
transform(chunk, encoding, callback) {
try {
if (Buffer.isBuffer(chunk)) {
// Buffer to object - deserialize
const parser = new Parser();
parser.on("data", (obj) => {
const result = objectClass &&
typeof objectClass.fromJSON === "function"
? objectClass.fromJSON(obj)
: new objectClass(obj);
callback(null, result);
});
parser.on("error", (error) => {
callback(error);
});
parser.write(chunk);
parser.end();
}
else if (chunk &&
typeof chunk.toJSON === "function" &&
!Buffer.isBuffer(chunk)) {
// Object to buffer - serialize (but not if it's already a Buffer)
const plainObject = chunk.toJSON();
const stringifer = new Stringifer();
let result = "";
stringifer.on("data", (data) => {
result += data.toString();
});
stringifer.on("end", () => {
callback(null, Buffer.from(result));
});
stringifer.on("error", (error) => {
callback(error);
});
stringifer.write(plainObject);
stringifer.end();
}
else {
// Pass through
callback(null, chunk);
}
}
catch (error) {
callback(error instanceof Error ? error : new Error(String(error)));
}
},
});
}
}
exports.ProtoObjectStream = ProtoObjectStream;