node-red source code

liudong

2023-05-29 340f156319b863525e50e900c58e59b86ecb3d5e

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
'use strict';
 
var util = require('util');
var stream = require('stream');
var constants = require('./constants');
 
var WritableStreamBuffer = module.exports = function(opts) {
  opts = opts || {};
  opts.decodeStrings = true;
 
  stream.Writable.call(this, opts);
 
  var initialSize = opts.initialSize || constants.DEFAULT_INITIAL_SIZE;
  var incrementAmount = opts.incrementAmount || constants.DEFAULT_INCREMENT_AMOUNT;
 
  var buffer = new Buffer(initialSize);
  var size = 0;
 
  this.size = function() {
    return size;
  };
 
  this.maxSize = function() {
    return buffer.length;
  };
 
  this.getContents = function(length) {
    if(!size) return false;
 
    var data = new Buffer(Math.min(length || size, size));
    buffer.copy(data, 0, 0, data.length);
 
    if(data.length < size)
      buffer.copy(buffer, 0, data.length);
 
    size -= data.length;
 
    return data;
  };
 
  this.getContentsAsString = function(encoding, length) {
    if(!size) return false;
 
    var data = buffer.toString(encoding || 'utf8', 0, Math.min(length || size, size));
    var dataLength = Buffer.byteLength(data);
 
    if(dataLength < size)
      buffer.copy(buffer, 0, dataLength);
 
    size -= dataLength;
    return data;
  };
 
  var increaseBufferIfNecessary = function(incomingDataSize) {
    if((buffer.length - size) < incomingDataSize) {
      var factor = Math.ceil((incomingDataSize - (buffer.length - size)) / incrementAmount);
 
      var newBuffer = new Buffer(buffer.length + (incrementAmount * factor));
      buffer.copy(newBuffer, 0, 0, size);
      buffer = newBuffer;
    }
  };
 
  this._write = function(chunk, encoding, callback) {
    increaseBufferIfNecessary(chunk.length);
    chunk.copy(buffer, size, 0);
    size += chunk.length;
    callback();
  };
};
 
util.inherits(WritableStreamBuffer, stream.Writable);