mirror of
https://github.com/fluencelabs/js-libp2p-tcp
synced 2025-07-02 05:01:34 +00:00
Compare commits
5 Commits
Author | SHA1 | Date | |
---|---|---|---|
730a477bd8 | |||
19b29fe40b | |||
073e14553b | |||
2ed01e8f5b | |||
4a121696d1 |
@ -1,6 +1,6 @@
|
||||
{
|
||||
"name": "libp2p-tcp",
|
||||
"version": "0.6.1",
|
||||
"version": "0.6.2",
|
||||
"description": "Node.js implementation of the TCP module that libp2p uses, which implements the interface-connection and interface-transport interfaces",
|
||||
"main": "lib/index.js",
|
||||
"jsnext:main": "src/index.js",
|
||||
@ -48,9 +48,9 @@
|
||||
},
|
||||
"contributors": [
|
||||
"David Dias <daviddias.p@gmail.com>",
|
||||
"Friedel Ziegelmayer <dignifiedquire@gmail.com>",
|
||||
"João Antunes <j.goncalo.antunes@gmail.com>",
|
||||
"Richard Littauer <richard.littauer@gmail.com>",
|
||||
"Stephen Whitmore <stephen.whitmore@gmail.com>"
|
||||
"Stephen Whitmore <stephen.whitmore@gmail.com>",
|
||||
"dignifiedquire <dignifiedquire@gmail.com>"
|
||||
]
|
||||
}
|
24
src/index.js
24
src/index.js
@ -12,6 +12,7 @@ const contains = require('lodash.contains')
|
||||
exports = module.exports = TCP
|
||||
|
||||
const IPFS_CODE = 421
|
||||
const CLOSE_TIMEOUT = 300
|
||||
|
||||
function TCP () {
|
||||
if (!(this instanceof TCP)) {
|
||||
@ -56,6 +57,17 @@ function TCP () {
|
||||
}
|
||||
handler(conn)
|
||||
})
|
||||
|
||||
listener.__connections = {}
|
||||
listener.on('connection', (conn) => {
|
||||
const key = `${conn.remoteAddress}:${conn.remotePort}`
|
||||
listener.__connections[key] = conn
|
||||
|
||||
conn.on('close', () => {
|
||||
delete listener.__connections[key]
|
||||
})
|
||||
})
|
||||
|
||||
listener.listen(m.toOptions(), () => {
|
||||
// Node.js likes to convert addr to IPv6 (when 0.0.0.0 for e.g)
|
||||
const address = listener.address()
|
||||
@ -86,13 +98,21 @@ function TCP () {
|
||||
}
|
||||
|
||||
this.close = (callback) => {
|
||||
log('closing')
|
||||
if (listeners.length === 0) {
|
||||
log('Called close with no active listeners')
|
||||
return callback()
|
||||
}
|
||||
|
||||
parallel(listeners.map((listener) => {
|
||||
return (cb) => listener.close(cb)
|
||||
parallel(listeners.map((listener) => (cb) => {
|
||||
setTimeout(() => {
|
||||
Object.keys(listener.__connections).forEach((key) => {
|
||||
log('destroying %s', key)
|
||||
listener.__connections[key].destroy()
|
||||
})
|
||||
}, CLOSE_TIMEOUT)
|
||||
|
||||
listener.close(cb)
|
||||
}), callback)
|
||||
}
|
||||
|
||||
|
@ -148,4 +148,35 @@ describe('libp2p-tcp', function () {
|
||||
expect(valid[0]).to.deep.equal(mh1)
|
||||
done()
|
||||
})
|
||||
|
||||
it('destroys after timeout', (done) => {
|
||||
const server = new TCPlibp2p()
|
||||
const mh = multiaddr('/ip4/127.0.0.1/tcp/9090')
|
||||
server.createListener(mh, (conn) => {
|
||||
const i = setInterval(() => {
|
||||
conn.read()
|
||||
conn.write('hi\n')
|
||||
}, 100)
|
||||
i.unref()
|
||||
}, () => {
|
||||
let connected = 0
|
||||
const connectHandler = () => {
|
||||
connected++
|
||||
if (connected === 10) {
|
||||
setTimeout(() => {
|
||||
server.close(done)
|
||||
}, 1)
|
||||
}
|
||||
}
|
||||
const errorHandler = () => {}
|
||||
|
||||
for (let i = 0; i < 10; i++) {
|
||||
const client = net.connect(9090)
|
||||
client.on('connect', connectHandler)
|
||||
|
||||
// just ignore the resets
|
||||
client.on('error', errorHandler)
|
||||
}
|
||||
})
|
||||
})
|
||||
})
|
||||
|
Reference in New Issue
Block a user