mirror of
https://github.com/DeNNiiInc/dbgate.git
synced 2026-04-19 18:56:00 +00:00
using SSE instead of socket IO
This commit is contained in:
@@ -4,7 +4,7 @@ const bodyParser = require('body-parser');
|
||||
const fileUpload = require('express-fileupload');
|
||||
const http = require('http');
|
||||
const cors = require('cors');
|
||||
const io = require('socket.io');
|
||||
// const io = require('socket.io');
|
||||
const fs = require('fs');
|
||||
const getPort = require('get-port');
|
||||
const childProcessChecker = require('./utility/childProcessChecker');
|
||||
@@ -43,7 +43,11 @@ function start() {
|
||||
const app = express();
|
||||
|
||||
const server = http.createServer(app);
|
||||
socket.set(io(server));
|
||||
|
||||
// const sse = new SSE();
|
||||
// app.get('/stream', sse.init);
|
||||
// socket.set(sse);
|
||||
// socket.set(io(server));
|
||||
|
||||
if (process.env.LOGIN && process.env.PASSWORD) {
|
||||
app.use(
|
||||
@@ -81,6 +85,30 @@ function start() {
|
||||
});
|
||||
|
||||
app.use(cors());
|
||||
|
||||
app.get('/stream', async function (req, res) {
|
||||
res.set({
|
||||
'Cache-Control': 'no-cache',
|
||||
'Content-Type': 'text/event-stream',
|
||||
Connection: 'keep-alive',
|
||||
});
|
||||
res.flushHeaders();
|
||||
|
||||
// Tell the client to retry every 10 seconds if connectivity is lost
|
||||
res.write('retry: 10000\n\n');
|
||||
socket.set(res);
|
||||
|
||||
// let count = 0;
|
||||
|
||||
// while (true) {
|
||||
// await new Promise((resolve) => setTimeout(resolve, 1000));
|
||||
|
||||
// console.log("Emit", ++count);
|
||||
// // Emit an SSE that contains the current 'count' as a string
|
||||
// res.write(`event: ping\ndata: ${JSON.stringify({ count })}\n\n`);
|
||||
// }
|
||||
});
|
||||
|
||||
app.use(bodyParser.json({ limit: '50mb' }));
|
||||
|
||||
app.use(
|
||||
|
||||
@@ -1,19 +1,35 @@
|
||||
let socket = null;
|
||||
let res = null;
|
||||
let init = '';
|
||||
|
||||
module.exports = {
|
||||
set(value) {
|
||||
socket = value;
|
||||
},
|
||||
get() {
|
||||
return socket;
|
||||
res = value;
|
||||
},
|
||||
// get() {
|
||||
// return socket;
|
||||
// },
|
||||
emit(message, data) {
|
||||
if (res) {
|
||||
if (init) {
|
||||
res.write(init);
|
||||
init = '';
|
||||
}
|
||||
res.write(`event: ${message}\ndata: ${JSON.stringify(data == null ? null : data)}\n\n`);
|
||||
} else {
|
||||
init += res;
|
||||
}
|
||||
|
||||
// console.log('EMIT:', message, data);
|
||||
socket.emit(message, data);
|
||||
// socket.emit(message, data);
|
||||
},
|
||||
emitChanged(key) {
|
||||
this.emit('clean-cache', key);
|
||||
this.emit(key);
|
||||
// console.log('EMIT_CHANGED:', key);
|
||||
socket.emit('clean-cache', key);
|
||||
socket.emit(key);
|
||||
// socket.emit('clean-cache', key);
|
||||
// socket.emit(key);
|
||||
|
||||
// socket.send(key, 'clean-cache');
|
||||
// socket.send(null, key);
|
||||
},
|
||||
};
|
||||
|
||||
Reference in New Issue
Block a user