-
Notifications
You must be signed in to change notification settings - Fork 7
/
server.js
92 lines (74 loc) · 2.11 KB
/
server.js
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
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
var WebSocketServer = require('ws').Server
var debug = require('debug')('ldnode:ws')
var InMemory = require('./in-memory')
var parallel = require('run-parallel')
var url = require('url')
module.exports = WsServer
function defaultToChannel(iri) {
return url.parse(iri).path
}
function WsServer (server, opts) {
var self = this
opts = opts || {}
this.suffix = opts.suffix || '.changes'
this.store = opts.store || new InMemory(opts)
var toChannel = opts.toChannel || defaultToChannel
// Starting WSS server
var wss = new WebSocketServer({
server: server,
clientTracking: false,
path: opts.path
})
// Handling a single connection
wss.on('connection', function (client) {
debug('New connection')
// var location = url.parse(client.upgradeReq.url, true)
// Handling messages
client.on('message', function (message) {
debug('New message: ' + message)
if (!message || typeof message !== 'string') {
return
}
var tuple = message.split(' ')
var command = tuple[0]
var iri = tuple[1]
// Only accept 'sub http://example.tld/hello'
if (tuple.length < 2 || command !== 'sub') {
return
}
var channel = toChannel ? toChannel(iri) : iri
self.store.subscribe(channel, iri, client, function (err, uuid) {
if (err) {
// TODO Should return an error
return
}
client.send('ack ' + tuple[1])
})
})
// Respond to ping
client.on('ping', function () {
client.pong()
})
})
}
WsServer.prototype.publish = function (iri, callback) {
this.store.get(iri, function (err, subscribers) {
if (err) {
if (callback) return callback(err)
else return
}
if (!subscribers) {
subscribers = {}
}
var tasks = Object.keys(subscribers)
.map(function (uuid) {
return function (cb) {
var client = subscribers[uuid][0]
var channel = subscribers[uuid][1]
debug('pub ' + channel + ' to ' + client.uuid)
client.send('pub ' + channel)
}
})
parallel(tasks, callback)
})
}