Connecting NextJS with NestJS via SocketIO

Introduction

SocketIO is a powerful library that enables bidirectional, real-time, event-based communication between the server and the browser.

The key advantages include:

  • Low Latency: Instant data transmission instead of having to send continuous requests.
  • Reliability: Automatic reconnection on connection loss and fallback support to HTTP long-polling if WebSocket is unavailable.
  • Broadcasting: Easily send data to one or multiple clients simultaneously.

In this article, I will guide you on using NextJS and NestJS to communicate with each other via SocketIO, simulating a chatbot app and the AI Agent's workflow.

Detail

On NestJS project you need to install these packages:

yarn add @nestjs/websockets @nestjs/platform-socket.io socket.io


Create the controller/conversations.controller.ts file returning the conversation list as follows:

import { Controller, Get } from "@nestjs/common";

@Controller("conversations")
export class ConversationsController {
@Get()
getConversations() {
return [
{
id: "1",
name: "NestJS System",
lastMsg: "Socket.io is ready",
time: "10:00",
online: true,
},
{
id: "2",
name: "React Squad",
lastMsg: "New components added",
time: "Yesterday",
online: false,
},
];
}
}


Create the socket/websocket.gateway.ts file:

import { MessageBody, SubscribeMessage, WebSocketGateway, WebSocketServer } from "@nestjs/websockets";
import { Server } from "socket.io";

@WebSocketGateway({
cors: { origin: "*" },
path: "/socket-io-ai",
})
export class ChatGateway {
@WebSocketServer()
server!: Server;

@SubscribeMessage("sendMessage")
async handleMessage(@MessageBody() data: { sender: string; text: string }) {
const steps = ["🔍 Analyzing user intent...", "📚 Querying internal knowledge base...", "🤖 Running inference with Llama-3...", "✨ Synthesizing final response..."];

for (const step of steps) {
await this.delay(600);
this.server.emit("ai_processing", { step });
}

const responseTemplates = [
`I've analyzed your message: "${data.text}". Based on my processing, everything looks optimal.`,
`Regarding "${data.text}", I've cross-referenced our database and found the relevant context for you.`,
`Processing "${data.text}" complete. I've utilized specialized logic to give you this insight.`,
`Interesting point about "${data.text}". My neural network suggests focusing on the core parameters mentioned.`,
`Task finished! I've successfully routed "${data.text}" through the NestJS Gateway and synthesized this result.`,
];

const randomIndex = Math.floor(Math.random() * responseTemplates.length);
const finalResponse = responseTemplates[randomIndex];

this.server.emit("ai_response", {
id: `ai-${Date.now()}`,
role: "ai",
content: finalResponse,
});
}

private delay(ms: number) {
return new Promise(resolve => setTimeout(resolve, ms));
}
}

The code above initializes a WebSocket Gateway in NestJS, serving as an intermediary server. When receiving the "sendMessage" event, the server simulates the AI's Chain of Thought steps by sending consecutive "ai_processing" events with a delay, then returns the final result via the "ai_response" event.


Update the app.module.ts file:

import { Module } from "@nestjs/common";
import { ConversationsController } from "./controller/conversations.controller";
import { ChatGateway } from "./socket/websocket.gateway";

@Module({
controllers: [ConversationsController],
providers: [ChatGateway],
})
export class AppModule {}


On NextJS project you need to install the package:

yarn add socket.io-client


Create the app/chatbot/page.tsx file:

"use client";

import { LoadingOutlined, RobotOutlined, SendOutlined, UserOutlined } from "@ant-design/icons";
import { Avatar, Button, Input, Spin, Tag } from "antd";
import { useEffect, useRef, useState } from "react";
import { io, Socket } from "socket.io-client";

const NEST_BACKEND_URL = process.env.NEXT_PUBLIC_SOCKET_URL;

interface Message {
id: string;
role: "user" | "ai";
content: string;
steps?: string[];
}

export default function ChatPage() {
const [conversations, setConversations] = useState([]);
const [messages, setMessages] = useState<Message[]>([]);
const [inputValue, setInputValue] = useState("");
const [isTyping, setIsTyping] = useState(false);
const [currentStep, setCurrentStep] = useState("");

const tempSteps = useRef<string[]>([]);
const socketRef = useRef<Socket | null>(null);
const scrollRef = useRef<HTMLDivElement>(null);

useEffect(() => {
fetch(`${NEST_BACKEND_URL}/conversations`)
.then(res => res.json())
.then(setConversations)
.catch(err => console.error("Fetch error:", err));

socketRef.current = io(NEST_BACKEND_URL, {
path: "/socket-io-ai",
transports: ["websocket"],
});

socketRef.current.on("ai_processing", (data: { step: string }) => {
setIsTyping(true);
setCurrentStep(data.step);
tempSteps.current.push(data.step);
});

socketRef.current.on("ai_response", (data: Message) => {
const finalMsg: Message = {
...data,
steps: [...tempSteps.current],
};
setMessages(prev => [...prev, finalMsg]);
setIsTyping(false);
setCurrentStep("");
tempSteps.current = [];
});

return () => {
socketRef.current?.disconnect();
};
}, []);

useEffect(() => {
if (scrollRef.current) {
scrollRef.current.scrollTo({
top: scrollRef.current.scrollHeight,
behavior: "smooth",
});
}
}, [messages, isTyping, currentStep]);

const handleSend = () => {
if (!inputValue.trim()) return;
const userMsg: Message = {
id: Date.now().toString(),
role: "user",
content: inputValue,
};
setMessages(prev => [...prev, userMsg]);
socketRef.current?.emit("sendMessage", { sender: "User", text: inputValue });
setInputValue("");
tempSteps.current = [];
};

return (
<div className="flex h-screen bg-slate-50 p-4 gap-4 overflow-hidden font-sans text-slate-900">
<div className="w-80 bg-white rounded-2xl shadow-sm border border-slate-200 flex flex-col overflow-hidden">
<div className="p-5 font-bold text-lg border-b border-slate-100 text-blue-600 bg-white uppercase tracking-wider">Conversations</div>
<div className="flex-1 overflow-y-auto custom-scrollbar">
{conversations.map((item: any) => (
<div key={item.id} className="p-4 hover:bg-blue-50 cursor-pointer border-b border-slate-50 flex gap-3 items-center group transition-colors">
<div className="flex-shrink-0">
<Avatar src={`https://api.dicebear.com/7.x/avataaars/svg?seed=${item.id}`} size={45} className="border border-slate-100" />
</div>
<div className="flex-1 overflow-hidden">
<div className="flex justify-between items-center mb-1">
<span className="font-semibold text-slate-700 truncate">{item.name}</span>
<span className="text-[10px] text-slate-400 font-medium">{item.time}</span>
</div>
<div className="text-xs text-slate-500 truncate group-hover:text-blue-500 transition-colors">{item.lastMsg}</div>
</div>
</div>
))}
</div>
</div>

<div className="flex-1 bg-white rounded-2xl shadow-md border border-slate-200 flex flex-col overflow-hidden">
<div className="p-4 border-b border-slate-100 flex justify-between items-center bg-white/80 backdrop-blur-sm z-10">
<div className="flex items-center gap-3">
<div className="flex-shrink-0">
<Avatar icon={<RobotOutlined />} className="bg-blue-600 shadow-sm" size={40} />
</div>
<div>
<div className="font-bold text-slate-800 leading-tight italic">AI Agentic Node</div>
<div className="text-[10px] text-blue-500 font-bold uppercase tracking-widest">Processing Online</div>
</div>
</div>
<Tag color="blue" className="rounded-full px-3 border-none bg-blue-50 text-blue-600 font-medium uppercase text-[10px]">
Socket.io Stream
</Tag>
</div>

<div ref={scrollRef} className="flex-1 overflow-y-auto p-6 space-y-8 bg-slate-50/50">
{messages.map(msg => (
<div key={msg.id} className={`flex ${msg.role === "user" ? "justify-end" : "justify-start"} animate-in fade-in duration-300`}>
<div className={`flex max-w-[85%] gap-3 ${msg.role === "user" ? "flex-row-reverse" : "flex-row"}`}>
<div className="flex-shrink-0 self-end mb-1">
<Avatar icon={msg.role === "user" ? <UserOutlined /> : <RobotOutlined />} className={`${msg.role === "user" ? "bg-slate-800" : "bg-blue-600"} shadow-sm`} size={32} />
</div>
<div className="flex flex-col gap-4">
{msg.role === "ai" && msg.steps && msg.steps.length > 0 && (
<div className="flex flex-col gap-2 ml-1">
<div className="flex items-center gap-2 opacity-60">
<div className="h-[1px] w-4 bg-slate-300"></div>
<span className="text-[10px] font-bold text-slate-400 uppercase tracking-widest">Chain of Thought</span>
</div>
<div className="flex flex-col gap-1.5 border-l-2 border-slate-100 ml-1 pl-4">
{msg.steps.map((step, idx) => (
<div key={idx} className="flex items-center gap-3 py-0.5 group">
<div className="w-1.5 h-1.5 rounded-full bg-blue-400/50 group-last:bg-blue-500 shadow-[0_0_8px_rgba(96,165,250,0.5)]"></div>
<span className="text-[12px] text-slate-500 italic font-medium leading-none tracking-tight">{step}</span>
</div>
))}
</div>
</div>
)}

<div
className={`p-4 rounded-2xl shadow-sm text-[14.5px] leading-relaxed tracking-tight ${
msg.role === "user" ? "bg-blue-600 text-white rounded-br-none shadow-blue-100" : "bg-white text-slate-700 border border-slate-100 rounded-tl-none font-medium"
}`}
>
{msg.content}
</div>
</div>
</div>
</div>
))}

{isTyping && (
<div className="flex justify-start gap-3 pl-11">
<div className="bg-white/80 backdrop-blur-sm px-4 py-2 rounded-2xl border border-blue-100 flex items-center gap-3 shadow-sm ring-4 ring-blue-50/50">
<span className="text-[12px] text-blue-600 font-bold italic tracking-wide animate-pulse">{currentStep || "Analyzing..."}</span>
<Spin indicator={<LoadingOutlined className="text-blue-600 text-sm" spin />} />
</div>
</div>
)}
</div>

<div className="p-5 bg-white border-t border-slate-100">
<div className="flex gap-2 p-1.5 border border-slate-200 rounded-2xl bg-slate-50 focus-within:bg-white focus-within:border-blue-500 focus-within:ring-4 focus-within:ring-blue-100/50 transition-all shadow-inner">
<Input
variant="borderless"
placeholder="Ask AI anything..."
value={inputValue}
onChange={e => setInputValue(e.target.value)}
onPressEnter={handleSend}
disabled={isTyping}
className="py-2 px-4 text-slate-700 placeholder:text-slate-400"
/>
<Button
type="primary"
size="large"
shape="circle"
icon={<SendOutlined />}
onClick={handleSend}
disabled={!inputValue.trim() || isTyping}
className="flex-shrink-0 shadow-lg shadow-blue-200 flex items-center justify-center bg-blue-600 hover:scale-105 active:scale-95 transition-transform"
/>
</div>
</div>
</div>
</div>
);
}

This is the user interface (Client) using socket.io-client to listen for events from NestJS. When "ai_processing" is received, it displays the processing status and stores the steps in a temporary array. When "ai_response" is received, it updates the full content and the list of thought steps into the chat window for the user to follow.


Update the .env file; you can change the value if your server is at a different address:

NEXT_PUBLIC_SOCKET_URL = http://localhost:4000


The result is as follows:


You can see that the socket connection has been successfully created.

Happy coding!

See more articles here.

Comments

Post a Comment

Popular posts from this blog

All Practice Series

Image
Introduction This is a comprehensive page about the technologies I have shared in series format. You can view brief introductions and links to directly access each series you are interested in. In the field of software development, to deploy a product from the initial idea to its release, the standard process typically involves several stages as follows: Database : Designing and implementing the database according to business requirements, storing data during the system's operation. Backend : Handling the main logic of the system, communicating with the database and services. Frontend : Building the interface for users to interact with the system, which could be a desktop, mobile, or web application. This usually includes implementing UI/UX and integrating APIs from the backend. DevOps : Deploying the system for use, which can be done on a server or in the cloud. Testing : Applying testing methods to ensure the product meets the standards for release. Of course, these are just stan...
Read more

Kubernetes Deployment for Zero Downtime

Image
Introduction In Kubernetes (K8s) , a Pod is the smallest resource unit used to run one or more containers during deployment. There are several ways to create a Pod : you can create it directly, use a ReplicationController , or a ReplicaSet . However, the most commonly used resource for managing Pods is the Deployment . When you use a Deployment , it actually creates a ReplicaSet to manage the Pods but comes with many additional benefits that support the deployment process. Some Advantages of Deployment : Ensures Pod Availability : It guarantees that the specified number of Pods are always running according to the configuration, automatically deploying additional Pods if any failures occur. Supports Restart and Undo Deployment : Allows you to easily restart or roll back to previous versions of your Deployment . Zero Downtime Deployment : When updating configurations or scaling Deployments , zero downtime is crucial. This means that new Pods are created while the old Pods are stil...
Read more

Deploying a NodeJS Server on Google Kubernetes Engine

Image
Introduction to GKE Google Kubernetes Engine (GKE) is a managed Kubernetes service provided by Google Cloud Platform , facilitating simple and efficient deployment of Docker images. We only need to provide some configuration for the number of nodes, machine types, and replicas to use. Some Concepts Cluster A Cluster is a collection of Nodes where Kubernetes can deploy applications. A cluster includes at least one Master Node and multiple Worker Nodes . The Master Node is used to manage the Worker Nodes . Node A Node is a server in the Kubernetes Cluster. Nodes can be physical servers or virtual machines. Each Node runs  Kubernetes , which is responsible for communication between the Master Node and Worker Node , as well as managing Pods and containers running on it. Pod A Pod is the smallest deployable unit in Kubernetes . Each Pod contains one or more containers, typically Docker containers. Containers in the same Pod share a network namespace, meaning they have the ...
Read more

Setting up Kubernetes Dashboard with Kind

Image
Introduction In a previous article, I guided you through using Helm to deploy on Google Kubernetes Engine . However, if you want to cut down costs by using Kubernetes in your local environment instead of relying on a cloud provider during development, then Kind is your go-to. There are several tools to help set up Kubernetes locally, such as MiniKube , Kind , K3S , KubeAdm , and more. Each tool has its own pros and cons. In this article, I'll walk you through using Kind to quickly set up a Kubernetes cluster on Docker . Kind stands out for its compactness, making Kubernetes start up quickly, being user-friendly, and supporting the latest Kubernetes versions. Working with Kind Firstly, follow the instructions here to install Kind according to your operating system. If you're using Ubuntu , execute the command: [ $( uname -m ) = x86_64 ] && curl -Lo ./kind https://kind.sigs.k8s.io/dl/v0.23.0/kind-linux-amd64 chmod +x ./kind sudo mv ./kind /usr/local/bin/ki...
Read more

Practicing with Google Cloud Platform - Google Kubernetes Engine to deploy nginx

Image
Introduction This article provides simple step-by-step instructions for those who are new to Google Cloud Platform (GCP) and Google Kubernetes Engine (GKE) . I'll guide you through using  GKE  to create clusters and deploy nginx . The instructions below will primarily use gcloud  and kubectl  to initialize the cluster, which is more convenient than manual management on the Google Cloud interface. Prerequisites First, you need to prepare the following: Have a GCP account with permission to use Cloud services. If you're new, you'll get a $300 free trial to use for 90 days . Create a new GCP project. Enable Compute Engine and Kubernetes Engine services. Install Google Cloud SDK and kubectl For this installation step, refer to the GCP documentation for instructions tailored to your operating system. Once installed, execute the following commands to check if gcloud  and kubectl  are installed: gcloud version kubectl version If you see the version resul...
Read more

Using Kafka with Docker and NodeJS

Image
Introduction to Kafka Kafka is an open-source , distributed messaging system that functions on a publish/subscribe model. It is widely used by numerous large companies for high-performance , real-time data streaming. Developed by LinkedIn since 2011, Kafka has grown into the most popular distributed streaming platform. It can handle vast amounts of records with high efficiency. Advantages of Kafka Open-source : Freely available and continuously improved by a large community. High-throughput, high-frequency : Capable of processing large volumes of data across topics continuously. Automatic message storage : Allows for easy message retrieval and verification. Large user community : Offers extensive support and shared resources. Basic Concepts If you're new to Kafka and Message Queues, here are some key concepts to understand: Producer : Creates and sends data to the Kafka server, where data is sent as messages in byte array format. Consumer : One or more consumers subscribe to...
Read more

Monitoring with cAdvisor, Prometheus and Grafana on Docker

Image
Introduction Monitoring a system is crucial after deploying a product to a production environment. Keeping an eye on system metrics like logs , CPU , RAM , disks , etc, helps identify the system's status, performance issues, and provides timely solutions to ensure stable operations. While cloud providers like Google , Amazon , or Azure offer built-in monitoring systems, if your company needs to manage multiple applications/systems/containers and desires a centralized monitoring system for easier management, using cAdvisor , Prometheus , and Grafana is a sensible choice. These three popular open-source tools are widely used by DevOps teams, especially for monitoring container applications. cAdvisor Developed by Google , cAdvisor is an open-source project used to analyze resource usage, performance, and other metrics from container applications, providing an overview of all running containers. Find more details here Prometheus Prometheus is a toolkit for system monitoring and a...
Read more

Kubernetes Practice Series

Image
Introduction Kubernetes (often abbreviated as K8s ) is an open-source system designed to automate the deployment, scaling, and management of containerized applications. This page serves as a collection of articles related to Kubernetes (K8s) , including theoretical concepts and practical guides on using Kubernetes to set up essential tools for the software development process. I will continue to update this series with new articles as ideas come to mind, to ensure the series becomes more comprehensive. The articles are arranged in increasing order of difficulty to make it easier for you to follow. If you have time, it's recommended that you start from the beginning of the series to acquire the necessary knowledge and information that will prepare you for the subsequent articles. Key Topics Covered in This Series Basic Knowledge : Fundamental concepts, common commands, etc. Resources : Pod, Deployment, Service, StatefulSet, Ingress, etc. Pod Autoscaler : Horizontal and Vertical sc...
Read more

Sitemap

Image
A page aggregator designed for easy reference and search, compiling all blog posts for convenient access. Loading all posts...
Read more

NodeJS Practice Series

Image
Introduction NodeJS is an open-source and cross-platform JavaScript runtime environment . Here are some key points about NodeJS : V8 Engine : NodeJS runs on the V8 JavaScript engine , which is also the core of Google Chrome . This allows NodeJS to be highly performant. Asynchronous and Non-Blocking : NodeJS  uses an event-driven, non-blocking I/O model. It’s lightweight and efficient, making it ideal for data-intensive real-time applications. Single-Threaded : NodeJS  runs in a single process, handling multiple requests without creating new threads. It eliminates waiting and continues with the next request. Common Language : Frontend developers who write JavaScript for browsers can use the same language for server-side code in NodeJS . You can even use the latest ECMAScript standards without waiting for browser updates. This page is designed to compile articles related to NodeJS , including how to integrate it with various libraries and relevant tech stacks. I will cont...
Read more