nuc-node-event-test/server/server.ts

Event-driven Message Broker

#!/usr/bin/env node

import { Server } from 'socket.io';
import { Socket } from 'socket.io';
import http from 'http';

const server = http.createServer();
const io = new Server(server, {
  cors: { origin: '*' }
});

type Subscriptions = Record<string, Set<string>>;
const subscriptions: Subscriptions = {};

io.on('connection', (socket: Socket) => {
  console.log('Client connected:', socket.id);

  socket.on('subscribe', (type: string) => {
    if (!subscriptions[type]) subscriptions[type] = new Set();
    subscriptions[type].add(socket.id);
    console.log(`Socket ${socket.id} subscribed to ${type}`);
  });

  socket.on('unsubscribe', (type: string) => {
    if (subscriptions[type]) {
      subscriptions[type].delete(socket.id);
      if (subscriptions[type].size === 0) delete subscriptions[type];
      console.log(`Socket ${socket.id} unsubscribed from ${type}`);
    }
  });

  socket.on('publish', ({ type, payload }: { type: string; payload: any }) => {
    console.log(`Publish: ${type}`, payload);
    if (subscriptions[type]) {
      subscriptions[type].forEach((sid) => {
        if (sid !== socket.id) {
          io.to(sid).emit('event', { type, payload });
        }
      });
    }
  });

  socket.on('disconnect', () => {
    Object.keys(subscriptions).forEach((type) => {
      subscriptions[type].delete(socket.id);
      if (subscriptions[type].size === 0) delete subscriptions[type];
    });
    console.log('Client disconnected:', socket.id);
  });
});

const PORT = process.env.PORT || 8080;
server.listen(PORT, () => {
  console.log(`Event server listening on port ${PORT}`);
});