24: socket.io chat

src/chat/chat.gateway.ts

@@ -0,0 +1,44 @@
+import {
+    ConnectedSocket,
+    MessageBody, OnGatewayConnection,
+    SubscribeMessage,
+    WebSocketGateway,
+    WebSocketServer,
+} from '@nestjs/websockets';
+import {Server, Socket} from 'socket.io';
+import {ChatService} from "./chat.service";
+
+@WebSocketGateway()
+export class ChatGateway implements OnGatewayConnection {
+    @WebSocketServer()
+    server: Server;
+
+    constructor(
+        private readonly chatService: ChatService
+    ) {
+    }
+
+    async handleConnection(socket: Socket) {
+        await this.chatService.getUserFromSocket(socket);
+    }
+
+    @SubscribeMessage('send_message')
+    async listenForMessages(@MessageBody() content: string, @ConnectedSocket() socket: Socket,) {
+        const author = await this.chatService.getUserFromSocket(socket);
+        const message = await this.chatService.saveMessage(content, author);
+        this.server.sockets.emit('receive_message', message);
+        return message;
+    }
+
+    @SubscribeMessage('request_all_messages')
+    async requestAllMessages(
+        @ConnectedSocket() socket: Socket,
+    ) {
+        await this.chatService.getUserFromSocket(socket);
+        const messages = await this.chatService.getAllMessages();
+
+        socket.emit('send_all_messages', messages);
+    }
+
+
+}