server.js
2.13 KB
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
/* Gun Multi-WS Monster */
/* Spawn multiple Gun WebSockets from the same HTTP/HTTPS server
* Each Gun is scoped to its ws.path and intended for ephemeral usage
* MIT Licensed (C) QXIP 2020
*/
const url = require("url");
var rimraf = require("rimraf");
const Gun = require("gun/gun"); // do not load storage adaptors by default
require("./gun-ws.js"); // required to allow external websockets into gun constructor
require("./mem.js"); // disable to allow file writing for debug
const http = require("http");
const WebSocket = require("ws");
var server = http.createServer();
// LRU with last used sockets
const QuickLRU = require("quick-lru");
var evict = function(key, value) {
console.log("Garbage Collect", key);
if (key)
rimraf("tmp/" + key, function() {
console.log("Cleaned up ID", key);
});
};
const lru = new QuickLRU({ maxSize: 10, onEviction: evict });
server.on("upgrade", async function(request, socket, head) {
var pathname = url.parse(request.url).pathname || "/gun";
console.log("Got WS request", pathname);
var gun = { gun: false, server: false };
if (pathname) {
if (lru.has(pathname)) {
// Existing Node
console.log("Recycle id", pathname);
gun = await lru.get(pathname);
} else {
// Create Node
console.log("Create id", pathname);
// NOTE: Only works with lib/ws.js shim allowing a predefined WS as ws.web parameter in Gun constructor
gun.server = new WebSocket.Server({ noServer: true, path: pathname });
console.log("set peer", request.headers.host + pathname);
gun.gun = new Gun({
peers: [], // should we use self as peer?
localStorage: false,
file: "tmp/" + pathname,
multicast: false,
ws: { noServer: true, path: pathname, web: gun.server },
web: gun.server
});
lru.set(pathname, gun);
}
}
if (gun.server) {
// Handle Request
gun.server.handleUpgrade(request, socket, head, function(ws) {
console.log("connecting to gun instance", gun.gun.opt()._.opt.ws.path);
gun.server.emit("connection", ws, request);
});
} else {
socket.destroy();
}
});
//
server.listen(3000);