跳到主要內容
版本:4.x

伺服器傳遞

有兩種常見方式可在重新連線時同步用戶端的狀態

  • 伺服器傳送整個狀態
  • 或用戶端追蹤已處理的最後一個事件,而伺服器傳送遺漏的部分

這兩種都是完全有效的解決方案,選擇哪一種將取決於你的使用案例。在本教學中,我們將採用後者。

首先,讓我們保留聊天應用程式的訊息。今天有很多很棒的選項,我們將在此使用 SQLite

提示

如果你不熟悉 SQLite,網路上有很多教學可供使用,例如 這個

讓我們安裝必要的套件

npm install sqlite sqlite3

我們會將每則訊息儲存在 SQL 表格中

index.js
const express = require('express');
const { createServer } = require('node:http');
const { join } = require('node:path');
const { Server } = require('socket.io');
const sqlite3 = require('sqlite3');
const { open } = require('sqlite');

async function main() {
// open the database file
const db = await open({
filename: 'chat.db',
driver: sqlite3.Database
});

// create our 'messages' table (you can ignore the 'client_offset' column for now)
await db.exec(`
CREATE TABLE IF NOT EXISTS messages (
id INTEGER PRIMARY KEY AUTOINCREMENT,
client_offset TEXT UNIQUE,
content TEXT
);
`);

const app = express();
const server = createServer(app);
const io = new Server(server, {
connectionStateRecovery: {}
});

app.get('/', (req, res) => {
res.sendFile(join(__dirname, 'index.html'));
});

io.on('connection', (socket) => {
socket.on('chat message', async (msg) => {
let result;
try {
// store the message in the database
result = await db.run('INSERT INTO messages (content) VALUES (?)', msg);
} catch (e) {
// TODO handle the failure
return;
}
// include the offset with the message
io.emit('chat message', msg, result.lastID);
});
});

server.listen(3000, () => {
console.log('server running at https://127.0.0.1:3000');
});
}

main();

然後,客戶端會追蹤偏移量

index.html
<script>
const socket = io({
auth: {
serverOffset: 0
}
});

const form = document.getElementById('form');
const input = document.getElementById('input');
const messages = document.getElementById('messages');

form.addEventListener('submit', (e) => {
e.preventDefault();
if (input.value) {
socket.emit('chat message', input.value);
input.value = '';
}
});

socket.on('chat message', (msg, serverOffset) => {
const item = document.createElement('li');
item.textContent = msg;
messages.appendChild(item);
window.scrollTo(0, document.body.scrollHeight);
socket.auth.serverOffset = serverOffset;
});
</script>

最後,伺服器會在(重新)連線時傳送遺失的訊息

index.js
// [...]

io.on('connection', async (socket) => {
socket.on('chat message', async (msg) => {
let result;
try {
result = await db.run('INSERT INTO messages (content) VALUES (?)', msg);
} catch (e) {
// TODO handle the failure
return;
}
io.emit('chat message', msg, result.lastID);
});

if (!socket.recovered) {
// if the connection state recovery was not successful
try {
await db.each('SELECT id, content FROM messages WHERE id > ?',
[socket.handshake.auth.serverOffset || 0],
(_err, row) => {
socket.emit('chat message', row.content, row.id);
}
)
} catch (e) {
// something went wrong
}
}
});

// [...]

讓我們看看實際運作

如上方的影片所示,它在暫時斷線和重新整理整頁後都能運作。

提示

與「連線狀態復原」功能的差異在於,成功的復原可能不需要連線到你的主資料庫(例如,它可能會從 Redis 串流中擷取訊息)。

好的,現在讓我們來談談客戶端傳遞。