Socket.IO - 聊天应用程序
现在我们已经熟悉了 Socket.IO,让我们编写一个聊天应用程序,我们可以用它在不同的聊天室中聊天。 我们将允许用户选择用户名并允许他们使用该用户名进行聊天。 首先,让我们设置 HTML 文件来请求用户名 −
<!DOCTYPE html> <html> <head><title>Hello world</title></head> <script src="/socket.io/socket.io.js"></script> <script> var socket = io(); </script> <body> <input type="text" name="name" value="" placeholder="Enter your name!"> <button type="button" name="button">Let me chat!</button> </body> </html>
现在我们已经设置了 HTML 来请求用户名,让我们创建服务器来接受来自客户端的连接。 我们将允许人们使用 setUsername 事件发送他们选择的用户名。 如果用户存在,我们将通过 userExists 事件进行响应,否则使用 userSet 事件。
var app = require('express')(); var http = require('http').Server(app); var io = require('socket.io')(http); app.get('/', function(req, res){ res.sendFile('E:/test/index.html');}); users = []; io.on('connection', function(socket){ console.log('A user connected'); socket.on('setUsername', function(data){ if(users.indexOf(data) > -1){ users.push(data); socket.emit('userSet', {username: data}); } else { socket.emit('userExists', data + ' username is taken! Try some other username.'); } }) }); http.listen(3000, function(){ console.log('listening on localhost:3000'); });
当人们点击按钮时,我们需要将用户名发送到服务器。 如果用户存在,我们会显示错误消息; 否则,我们会显示一个消息屏幕 −
<!DOCTYPE html> <html> <head><title>Hello world</title></head> <script src="/socket.io/socket.io.js"></script> <script> var socket = io(); function setUsername(){ socket.emit('setUsername', document.getElementById('name').value); }; var user; socket.on('userExists', function(data){ document.getElementById('error-container').innerHTML = data; }); socket.on('userSet', function(data){ user = data.username; document.body.innerHTML = '<input type="text" id="message">\ <button type="button" name="button" onclick="sendMessage()">Send</button>\ <div id="message-container"></div>'; }); function sendMessage(){ var msg = document.getElementById('message').value; if(msg){ socket.emit('msg', {message: msg, user: user}); } } socket.on('newmsg', function(data){ if(user){ document.getElementById('message-container').innerHTML +='<div><b>' + data.user + '</b>: ' + data.message + '</div>' } }) </script> <body> <div id="error-container"></div> <input id="name" type="text" name="name" value="" placeholder="Enter your name!"> <button type="button" name="button" onclick="setUsername()">Let me chat!</button> </body> </html>
现在,如果您使用相同的用户名连接两个客户端,则会出现错误,如下面的屏幕截图所示 −
一旦您提供了可接受的用户名,您将进入一个带有消息框和发送消息按钮的屏幕。 现在,我们必须处理消息并将其定向到连接的客户端。 为此,修改您的 app.js 文件以包含以下更改 −
var app = require('express')(); var http = require('http').Server(app); var io = require('socket.io')(http); app.get('/', function(req, res){ res.sendFile('E:/test/index.html');}); users = []; io.on('connection', function(socket){ console.log('A user connected'); socket.on('setUsername', function(data){ console.log(data); if(users.indexOf(data) > -1){ socket.emit('userExists', data + ' username is taken! Try some other username.'); } else { users.push(data); socket.emit('userSet', {username: data}); } }); socket.on('msg', function(data){ //Send message to everyone io.sockets.emit('newmsg', data); }) }); http.listen(3000, function(){ console.log('listening on localhost:3000'); });
现在将任意数量的客户端连接到您的服务器,为他们提供用户名并开始聊天! 在下面的示例中,我们连接了两个名为 Ayush 和 Harshit 的客户端,并从这两个客户端发送了一些消息 −