mirror of https://gitee.com/openkylin/nodejs.git
50 lines
1.2 KiB
JavaScript
50 lines
1.2 KiB
JavaScript
/* eslint-disable node-core/require-common-first, node-core/required-modules */
|
|
'use strict';
|
|
const { Duplex } = require('stream');
|
|
const assert = require('assert');
|
|
|
|
const kCallback = Symbol('Callback');
|
|
const kOtherSide = Symbol('Other');
|
|
|
|
class DuplexSocket extends Duplex {
|
|
constructor() {
|
|
super();
|
|
this[kCallback] = null;
|
|
this[kOtherSide] = null;
|
|
}
|
|
|
|
_read() {
|
|
const callback = this[kCallback];
|
|
if (callback) {
|
|
this[kCallback] = null;
|
|
callback();
|
|
}
|
|
}
|
|
|
|
_write(chunk, encoding, callback) {
|
|
assert.notStrictEqual(this[kOtherSide], null);
|
|
assert.strictEqual(this[kOtherSide][kCallback], null);
|
|
if (chunk.length === 0) {
|
|
process.nextTick(callback);
|
|
} else {
|
|
this[kOtherSide].push(chunk);
|
|
this[kOtherSide][kCallback] = callback;
|
|
}
|
|
}
|
|
|
|
_final(callback) {
|
|
this[kOtherSide].on('end', callback);
|
|
this[kOtherSide].push(null);
|
|
}
|
|
}
|
|
|
|
function makeDuplexPair() {
|
|
const clientSide = new DuplexSocket();
|
|
const serverSide = new DuplexSocket();
|
|
clientSide[kOtherSide] = serverSide;
|
|
serverSide[kOtherSide] = clientSide;
|
|
return { clientSide, serverSide };
|
|
}
|
|
|
|
module.exports = makeDuplexPair;
|