初始化项目

15815213711

2024-08-26 67b8b6731811983447e053d4396b3708c14dfe3c

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
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
'use strict';
const {Transform} = require('stream');
 
module.exports = {
    download(response, emitter, downloadBodySize) {
        let downloaded = 0;
 
        return new Transform({
            transform(chunk, encoding, callback) {
                downloaded += chunk.length;
 
                const percent = downloadBodySize ? downloaded / downloadBodySize : 0;
 
                // Let `flush()` be responsible for emitting the last event
                if (percent < 1) {
                    emitter.emit('downloadProgress', {
                        percent,
                        transferred: downloaded,
                        total: downloadBodySize
                    });
                }
 
                callback(null, chunk);
            },
 
            flush(callback) {
                emitter.emit('downloadProgress', {
                    percent: 1,
                    transferred: downloaded,
                    total: downloadBodySize
                });
 
                callback();
            }
        });
    },
 
    upload(request, emitter, uploadBodySize) {
        const uploadEventFrequency = 150;
        let uploaded = 0;
        let progressInterval;
 
        emitter.emit('uploadProgress', {
            percent: 0,
            transferred: 0,
            total: uploadBodySize
        });
 
        request.once('error', () => {
            clearInterval(progressInterval);
        });
 
        request.once('response', () => {
            clearInterval(progressInterval);
 
            emitter.emit('uploadProgress', {
                percent: 1,
                transferred: uploaded,
                total: uploadBodySize
            });
        });
 
        request.once('socket', socket => {
            const onSocketConnect = () => {
                progressInterval = setInterval(() => {
                    const lastUploaded = uploaded;
                    /* istanbul ignore next: see #490 (occurs randomly!) */
                    const headersSize = request._header ? Buffer.byteLength(request._header) : 0;
                    uploaded = socket.bytesWritten - headersSize;
 
                    // Don't emit events with unchanged progress and
                    // prevent last event from being emitted, because
                    // it's emitted when `response` is emitted
                    if (uploaded === lastUploaded || uploaded === uploadBodySize) {
                        return;
                    }
 
                    emitter.emit('uploadProgress', {
                        percent: uploadBodySize ? uploaded / uploadBodySize : 0,
                        transferred: uploaded,
                        total: uploadBodySize
                    });
                }, uploadEventFrequency);
            };
 
            /* istanbul ignore next: hard to test */
            if (socket.connecting) {
                socket.once('connect', onSocketConnect);
            } else if (socket.writable) {
                // The socket is being reused from pool,
                // so the connect event will not be emitted
                onSocketConnect();
            }
        });
    }
};