优化

‘liusuyi’

2023-10-21 ae92b6823499cfab77201b4033922fa373fb37a3

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
'use strict';
const fs = require('fs');
const stream = require('stream');
const zlib = require('zlib');
const duplexer = require('duplexer');
const pify = require('pify');
 
const getOptions = options => Object.assign({level: 9}, options);
 
module.exports = (input, options) => {
    if (!input) {
        return Promise.resolve(0);
    }
 
    return pify(zlib.gzip)(input, getOptions(options)).then(data => data.length).catch(_ => 0);
};
 
module.exports.sync = (input, options) => zlib.gzipSync(input, getOptions(options)).length;
 
module.exports.stream = options => {
    const input = new stream.PassThrough();
    const output = new stream.PassThrough();
    const wrapper = duplexer(input, output);
 
    let gzipSize = 0;
    const gzip = zlib.createGzip(getOptions(options))
        .on('data', buf => {
            gzipSize += buf.length;
        })
        .on('error', () => {
            wrapper.gzipSize = 0;
        })
        .on('end', () => {
            wrapper.gzipSize = gzipSize;
            wrapper.emit('gzip-size', gzipSize);
            output.end();
        });
 
    input.pipe(gzip);
    input.pipe(output, {end: false});
 
    return wrapper;
};
 
module.exports.file = (path, options) => {
    return new Promise((resolve, reject) => {
        const stream = fs.createReadStream(path);
        stream.on('error', reject);
 
        const gzipStream = stream.pipe(module.exports.stream(options));
        gzipStream.on('error', reject);
        gzipStream.on('gzip-size', resolve);
    });
};
 
module.exports.fileSync = (path, options) => module.exports.sync(fs.readFileSync(path), options);