Skip to content

Commit

Permalink
feat: implement ipfs ping flags ipfs#928
Browse files Browse the repository at this point in the history
  • Loading branch information
melvin0008 authored and Melvin Philips committed Feb 2, 2018
1 parent 3bca165 commit 0dba866
Show file tree
Hide file tree
Showing 7 changed files with 112 additions and 3 deletions.
44 changes: 44 additions & 0 deletions src/cli/commands/ping.js
Original file line number Diff line number Diff line change
@@ -0,0 +1,44 @@
'use strict'

const print = require('../utils').print

module.exports = {
command: 'ping <peerId>',

describe: 'Measure the latency of a connection',

builder: {
count: {
alias: 'n',
type: 'integer',
default: 10
}
},

handler (argv) {
const peerId = argv.peerId
const count = argv.count || 10

print('PING ' + peerId)

let noOfTimes = 0
let totalTime = 0

const pingCb = (err, p) => {
if (err) {
throw err
}
let time = p.Time
totalTime = totalTime + time
noOfTimes = noOfTimes + 1
print('Pong received: time=' + time + ' ms')
if (noOfTimes === count) {
print('Average latency: ' + totalTime / count + 'ms')
}
}

for (let i = 0; i < count; i++) {
argv.ipfs.ping(peerId, pingCb)
}
}
}
30 changes: 28 additions & 2 deletions src/core/components/ping.js
Original file line number Diff line number Diff line change
@@ -1,9 +1,35 @@
'use strict'

const promisify = require('promisify-es6')
const OFFLINE_ERROR = require('../utils').OFFLINE_ERROR
const PeerId = require('peer-id')
const PeerInfo = require('peer-info')
var Readable = require('stream').Readable

module.exports = function ping (self) {
return promisify((callback) => {
callback(new Error('Not implemented'))
return promisify((peerId, cb) => {
if (!self.isOnline()) {
return cb(new Error(OFFLINE_ERROR))
}

var outputStream = new Readable()
outputStream._read = function (size) {
}

let peer
try {
peer = self._libp2pNode.peerBook.get(peerId)
} catch (err) {
peer = new PeerInfo(PeerId.createFromB58String(peerId))
}

self._libp2pNode.ping(peer, (err, p) => {
p.once('ping', (time) => {
outputStream.push(JSON.stringify([{}, { Success: true, Time: time }, { Text: 'Average latency: ' + time + ' ms' }]))
outputStream.push(null)
p.stop()
cb(err, outputStream)
})
})
})
}
1 change: 1 addition & 0 deletions src/http/api/resources/index.js
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,7 @@

exports.version = require('./version')
exports.id = require('./id')
exports.ping = require('./ping')
exports.bootstrap = require('./bootstrap')
exports.repo = require('./repo')
exports.object = require('./object')
Expand Down
18 changes: 18 additions & 0 deletions src/http/api/resources/ping.js
Original file line number Diff line number Diff line change
@@ -0,0 +1,18 @@
'use strict'

const boom = require('boom')

exports = module.exports

exports.get = (request, reply) => {
const ipfs = request.server.app.ipfs
const peerId = request.query.arg

ipfs.ping(peerId, (err, outputStream) => {
if (err) {
return reply(boom.badRequest(err))
}

return reply(outputStream).type('application/json').header('x-chunked-output', '1')
})
}
1 change: 1 addition & 0 deletions src/http/api/routes/index.js
Original file line number Diff line number Diff line change
Expand Up @@ -8,6 +8,7 @@ module.exports = (server) => {
require('./object')(server)
// require('./repo')(server)
require('./config')(server)
require('./ping')(server)
require('./swarm')(server)
require('./bitswap')(server)
require('./file')(server)
Expand Down
19 changes: 19 additions & 0 deletions src/http/api/routes/ping.js
Original file line number Diff line number Diff line change
@@ -0,0 +1,19 @@
'use strict'

const resources = require('./../resources')

module.exports = (server) => {
const api = server.select('API')

api.route({
method: '*',
path: '/api/v0/ping',
config: {
payload: {
parse: false,
output: 'stream'
},
handler: resources.ping.get
}
})
}
2 changes: 1 addition & 1 deletion test/cli/commands.js
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
const expect = require('chai').expect
const runOnAndOff = require('../utils/on-and-off')

const commandCount = 60
const commandCount = 61

describe('commands', () => runOnAndOff((thing) => {
let ipfs
Expand Down

0 comments on commit 0dba866

Please sign in to comment.