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('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('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 的客户端，并从两个客户端发送了一些消息 -
