blob: 0c13168e58919dcc500e4df2e851190ec2ef9cd2 (
plain)
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
|
var Stream = require('stream');
function prop(propName) {
return function (data) {
return data[propName];
};
}
module.exports = unique;
function unique(propName) {
var keyfn = JSON.stringify;
if (typeof propName === 'string') {
keyfn = prop(propName);
} else if (typeof propName === 'function') {
keyfn = propName;
}
var seen = {};
var s = new Stream();
s.readable = true;
s.writable = true;
var pipes = 0;
s.write = function (data) {
var key = keyfn(data);
if (seen[key] === undefined) {
seen[key] = true;
s.emit('data', data);
}
};
var ended = 0;
s.end = function (data) {
if (arguments.length) s.write(data);
ended++;
if (ended === pipes || pipes === 0) {
s.writable = false;
s.emit('end');
}
};
s.destroy = function (data) {
s.writable = false;
};
s.on('pipe', function () {
pipes++;
});
s.on('unpipe', function () {
pipes--;
});
return s;
}
|