rdf-dataset-fragmenter/lib/io/QuadSinkCsv.js

Fragments an RDF dataset into multiple parts

"use strict";
var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
    if (k2 === undefined) k2 = k;
    var desc = Object.getOwnPropertyDescriptor(m, k);
    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
      desc = { enumerable: true, get: function() { return m[k]; } };
    }
    Object.defineProperty(o, k2, desc);
}) : (function(o, m, k, k2) {
    if (k2 === undefined) k2 = k;
    o[k2] = m[k];
}));
var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
    Object.defineProperty(o, "default", { enumerable: true, value: v });
}) : function(o, v) {
    o["default"] = v;
});
var __importStar = (this && this.__importStar) || (function () {
    var ownKeys = function(o) {
        ownKeys = Object.getOwnPropertyNames || function (o) {
            var ar = [];
            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
            return ar;
        };
        return ownKeys(o);
    };
    return function (mod) {
        if (mod && mod.__esModule) return mod;
        var result = {};
        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
        __setModuleDefault(result, mod);
        return result;
    };
})();
Object.defineProperty(exports, "__esModule", { value: true });
exports.QuadSinkCsv = void 0;
const fs = __importStar(require("node:fs"));
const node_path_1 = require("node:path");
const mkdirp_1 = require("mkdirp");
const rdf_string_1 = require("rdf-string");
/**
 * A quad sink that writes quads to a CSV file.
 */
class QuadSinkCsv {
    file;
    columns;
    fileStream;
    /**
     * @param file The file to write to.
     * @param columns The quad term names that will be serialized as columns.
     */
    constructor(file, columns) {
        this.file = file;
        this.columns = columns;
    }
    close() {
        return new Promise((resolve, reject) => {
            if (this.fileStream) {
                this.fileStream.close((error) => {
                    if (error) {
                        return reject(error);
                    }
                    return resolve();
                });
            }
            else {
                resolve();
            }
        });
    }
    async push(iri, quad) {
        if (!this.fileStream) {
            // Ensure output directory exists
            const folder = (0, node_path_1.dirname)(this.file);
            await (0, mkdirp_1.mkdirp)(folder);
            // Open file stream
            this.fileStream = fs.createWriteStream(this.file);
            await new Promise((resolve) => {
                this.fileStream.on('open', () => {
                    // Write CSV header
                    this.fileStream.write(`${this.columns.join(',')}\n`);
                    resolve();
                });
            });
        }
        this.fileStream.write(`${this.columns.map(column => (0, rdf_string_1.termToString)(quad[column])).join(',')}\n`);
    }
}
exports.QuadSinkCsv = QuadSinkCsv;
//# sourceMappingURL=QuadSinkCsv.js.map