При создании WebSocket-сервера обычно необходимо хранить некоторую идентифицирующую информацию или контекст, связанный с каждым подключённым клиентом.
С Bun.serve() эти «контекстные данные» устанавливаются при первоначальном обновлении соединения путём передачи параметра data в вызове server.upgrade().
ts
Bun.serve({
fetch(req, server) {
const success = server.upgrade(req, {
data: {
socketId: Math.random(),
},
});
if (success) return undefined;
// обрабатывать HTTP-запрос как обычно
// ...
},
websocket: {
// TypeScript: указать тип ws.data вот так
data: {} as { socketId: number },
// определить обработчики websocket
async message(ws, message) {
// контекстные данные доступны как свойство `data`
// на экземпляре WebSocket
console.log(`Получено ${message} от ${ws.data.socketId}`);
},
},
});Обычно читают cookies/headers из входящего запроса для идентификации подключающегося клиента.
ts
type WebSocketData = {
createdAt: number;
token: string;
userId: string;
};
Bun.serve({
async fetch(req, server) {
// использовать библиотеку для разбора cookies
const cookies = parseCookies(req.headers.get("Cookie"));
const token = cookies["X-Token"];
const user = await getUserFromToken(token);
const upgraded = server.upgrade(req, {
data: {
createdAt: Date.now(),
token: cookies["X-Token"],
userId: user.id,
},
});
if (upgraded) return undefined;
},
websocket: {
// TypeScript: указать тип ws.data вот так
data: {} as WebSocketData,
async message(ws, message) {
// сохранить сообщение в базу данных
await saveMessageToDatabase({
message: String(message),
userId: ws.data.userId,
});
},
},
});