OverviewFilesCommitsIssuesDocsReleasesPeers

mirall-cli / node_modules / dir-compare / build / src / FileCompareHandler / default / defaultFileCompare.js

112 lines · javascript ref:
"use strict";
var __importDefault = (this && this.__importDefault) || function (mod) {
    return (mod && mod.__esModule) ? mod : { "default": mod };
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.defaultFileCompare = void 0;
const fs_1 = __importDefault(require("fs"));
const FileDescriptorQueue_1 = require("../../FileSystem/FileDescriptorQueue");
const BufferPool_1 = require("../../FileSystem/BufferPool");
const FileCloser_1 = require("../../FileSystem/FileCloser");
const FsPromise_1 = require("../../FileSystem/FsPromise");
const MAX_CONCURRENT_FILE_COMPARE = 8;
const BUF_SIZE = 1000000;
const fdQueue = new FileDescriptorQueue_1.FileDescriptorQueue(MAX_CONCURRENT_FILE_COMPARE * 2);
const asyncBufferPool = new BufferPool_1.BufferPool(BUF_SIZE, MAX_CONCURRENT_FILE_COMPARE); // fdQueue guarantees there will be no more than MAX_CONCURRENT_FILE_COMPARE async processes accessing the buffers concurrently
const syncBufferPool = new BufferPool_1.BufferPool(BUF_SIZE, 2);
exports.defaultFileCompare = {
    compareSync, compareAsync
};
/**
 * Compares two files by content.
 */
function compareSync(path1, stat1, path2, stat2, options) {
    let fd1;
    let fd2;
    if (stat1.size !== stat2.size) {
        return false;
    }
    const bufferPair = syncBufferPool.allocateBuffers();
    try {
        fd1 = fs_1.default.openSync(path1, 'r');
        fd2 = fs_1.default.openSync(path2, 'r');
        const buf1 = bufferPair.buf1;
        const buf2 = bufferPair.buf2;
        for (;;) {
            const size1 = fs_1.default.readSync(fd1, buf1, 0, BUF_SIZE, null);
            const size2 = fs_1.default.readSync(fd2, buf2, 0, BUF_SIZE, null);
            if (size1 !== size2) {
                return false;
            }
            else if (size1 === 0) {
                // End of file reached
                return true;
            }
            else if (!compareBuffers(buf1, buf2, size1)) {
                return false;
            }
        }
    }
    finally {
        FileCloser_1.FileCloser.closeFilesSync(fd1, fd2);
        syncBufferPool.freeBuffers(bufferPair);
    }
}
/**
 * Compares two files by content
 */
function compareAsync(path1, stat1, path2, stat2, options) {
    let fd1;
    let fd2;
    let bufferPair;
    if (stat1.size !== stat2.size) {
        return Promise.resolve(false);
    }
    if (stat1.size < BUF_SIZE && !options.forceAsyncContentCompare) {
        return Promise.resolve(compareSync(path1, stat1, path2, stat2, options));
    }
    return Promise.all([fdQueue.openPromise(path1, 'r'), fdQueue.openPromise(path2, 'r')])
        .then(fds => {
        bufferPair = asyncBufferPool.allocateBuffers();
        fd1 = fds[0];
        fd2 = fds[1];
        const buf1 = bufferPair.buf1;
        const buf2 = bufferPair.buf2;
        const compareAsyncInternal = () => {
            return Promise.all([
                FsPromise_1.FsPromise.read(fd1, buf1, 0, BUF_SIZE, null),
                FsPromise_1.FsPromise.read(fd2, buf2, 0, BUF_SIZE, null)
            ]).then((bufferSizes) => {
                const size1 = bufferSizes[0];
                const size2 = bufferSizes[1];
                if (size1 !== size2) {
                    return false;
                }
                else if (size1 === 0) {
                    // End of file reached
                    return true;
                }
                else if (!compareBuffers(buf1, buf2, size1)) {
                    return false;
                }
                else {
                    return compareAsyncInternal();
                }
            });
        };
        return compareAsyncInternal();
    })
        .then(
    // 'finally' polyfill for node 8 and below
    res => finalizeAsync(fd1, fd2, bufferPair).then(() => res), err => finalizeAsync(fd1, fd2, bufferPair).then(() => { throw err; }));
}
function compareBuffers(buf1, buf2, contentSize) {
    return buf1.slice(0, contentSize).equals(buf2.slice(0, contentSize));
}
function finalizeAsync(fd1, fd2, bufferPair) {
    if (bufferPair) {
        asyncBufferPool.freeBuffers(bufferPair);
    }
    return FileCloser_1.FileCloser.closeFilesAsync(fd1, fd2, fdQueue);
}
//# sourceMappingURL=defaultFileCompare.js.map