优化光电巡检手动控制

‘liusuyi’

2023-06-01 eb0f13efdbd48b88411b4ff214bb92169202d157

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
'use strict';
const bufferStream = require('./buffer-stream');
 
function getStream(inputStream, opts) {
    if (!inputStream) {
        return Promise.reject(new Error('Expected a stream'));
    }
 
    opts = Object.assign({maxBuffer: Infinity}, opts);
 
    const maxBuffer = opts.maxBuffer;
    let stream;
    let clean;
 
    const p = new Promise((resolve, reject) => {
        const error = err => {
            if (err) { // null check
                err.bufferedData = stream.getBufferedValue();
            }
 
            reject(err);
        };
 
        stream = bufferStream(opts);
        inputStream.once('error', error);
        inputStream.pipe(stream);
 
        stream.on('data', () => {
            if (stream.getBufferedLength() > maxBuffer) {
                reject(new Error('maxBuffer exceeded'));
            }
        });
        stream.once('error', error);
        stream.on('end', resolve);
 
        clean = () => {
            // some streams doesn't implement the `stream.Readable` interface correctly
            if (inputStream.unpipe) {
                inputStream.unpipe(stream);
            }
        };
    });
 
    p.then(clean, clean);
 
    return p.then(() => stream.getBufferedValue());
}
 
module.exports = getStream;
module.exports.buffer = (stream, opts) => getStream(stream, Object.assign({}, opts, {encoding: 'buffer'}));
module.exports.array = (stream, opts) => getStream(stream, Object.assign({}, opts, {array: true}));