-
Notifications
You must be signed in to change notification settings - Fork 1
/
server.js
99 lines (85 loc) · 2.94 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
93
94
95
96
97
98
99
const WebSocket = require('websocket').server
const http = require('http')
const https = require('https')
const fs = require('fs')
const { MongoClient } = require('mongodb')
require('dotenv').config()
const MONGODB_URL = process.env.DB_URL
const DB_NAME = process.env.DB_NAME
const COLLECTIONS = process.env.DB_COLLECTIONS.split(',')
const PORT = process.env.PORT
const ENABLE_WSS = process.env.ENABLE_WSS === 'true'
const SSL_KEY_PATH = process.env.SSL_KEY_PATH
const SSL_CERT_PATH = process.env.SSL_CERT_PATH
let connections = []
// Create a new MongoDB client and connect to the database
const client = new MongoClient(MONGODB_URL, {
useUnifiedTopology: true,
})
async function main() {
try {
await client.connect()
console.log('Connected to MongoDB successfully!')
} catch (err) {
console.log(`Error connecting to MongoDB: ${err}`)
}
let server
if (ENABLE_WSS) {
// Read SSL certificate and private key files
const sslOptions = {
key: fs.readFileSync(SSL_KEY_PATH),
cert: fs.readFileSync(SSL_CERT_PATH),
}
// Create an HTTPS server
server = https.createServer(sslOptions, (req, res) => {
res.writeHead(404)
res.end()
})
} else {
// Create an HTTP server
server = http.createServer((req, res) => {
res.writeHead(404)
res.end()
})
}
// Create a new WebSocket server and attach it to the HTTP/HTTPS server
const wsServer = new WebSocket({
httpServer: server,
autoAcceptConnections: false,
})
COLLECTIONS.forEach((coll) => {
// Listen for MongoDB change stream events
const collection = client.db(DB_NAME).collection(coll)
const changeStream = collection.watch({ fullDocument: 'updateLookup' })
changeStream.on('change', (change) => {
const { fullDocument, documentKey, operationType, ns } = change
let message = {
fullDocument: operationType === 'delete' ? documentKey : fullDocument,
operationType,
collection: ns.coll,
}
// Broadcast the new message to all connected clients
connections.forEach((conn) => conn.send(JSON.stringify(message)))
})
})
// Listen for WebSocket connection requests
wsServer.on('request', (request) => {
// Accept the connection
const connection = request.accept(null, request.origin)
// When a new connection is made, add it to the list of connections
console.log('New connection accepted.')
connections.push(connection)
// When the connection is closed, remove it from the list of connections
connection.on('close', (reasonCode, description) => {
console.log(`Connection closed: ${reasonCode} - ${description}`)
const index = connections.indexOf(connection)
if (index !== -1) connections.splice(index, 1)
})
})
// Start the server
server.listen(PORT, () => {
console.log(`Server is listening on port ${PORT}`)
console.log(`WebSocket server is ${ENABLE_WSS ? 'secure (WSS)' : 'unsecure (WS)'}`)
})
}
main()