websocket实现简单聊天程序

Posted

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了websocket实现简单聊天程序相关的知识,希望对你有一定的参考价值。

程序的流程图:

技术分享

主要代码:

服务端 app.js

先加载所需要的通信模块:

var express = require(‘express‘);
var app = express();
var http = require(‘http‘).createServer(app);
var io = require(‘socket.io‘).listen(http);
var fs = require(‘fs‘);

创建用户列表和消息列表:

var person = [];
var history = [];

绑定并监听80端口:

app.get(‘/‘,function (req,res) {
    res.sendFile(__dirname + ‘/login.html);
});

http.listen(80,function () {
    console.log(‘listening on *:80‘);
});

客户端连接成功后,触发响应事件connection,完成要绑定的事件并实现客户端出发的事件:

io.sockets.on(‘connection‘,function (socket) {
    var user = ‘‘;

    socket.emit(‘history‘, history);
    io.sockets.emit(‘updatePerson‘, person); 

    socket.on(‘sendMsg‘, function (data) {
        var obj = {};
        obj.content = data;
        obj.time = now();
        obj.name = user;
        if (history.length === history_num) {
            history.shift();
        }
        history.push(obj);
        io.sockets.emit(‘news‘, obj);
    });

    socket.on(‘setUserName‘, function (data) {
        user = data;
        person.push(user);
        io.sockets.emit(‘loginsucess‘);
        io.sockets.emit(‘updatePerson‘, person);
        io.sockets.emit(‘news‘, {content: user + ‘进入房间‘, time: now(), name: ‘系统消息‘});
    });

    socket.on(‘disconnect‘, function () {
        if (user !== ‘‘) {
            person.forEach(function (value, index) {
                if (value === user) {
                    person.splice(index, 1);
                }
            });
            io.sockets.emit(‘news‘, {content: user + ‘离开房间‘, time: now(), name: ‘系统消息‘});
            io.sockets.emit(‘updatePerson‘, person);
        }
    });
});

客户端 index.js:

先初始化用户信息:

 $scope.data = [];     //消息队列  
 $scope.name = ‘‘;    //用户名
 $scope.content = ‘‘;  //用户输入的消息
 $scope.personlist = []; //用户队列

然后连接服务器端:

const  socket_url = ‘http://localhost‘;
var socket = io.connect(socket_url);

连接成功后,对用户昵称输入的验证:

$scope.checkName = function () {
    if($scope.name!==‘‘){
        if($scope.personlist.length!==0){
            if($scope.personlist.indexOf($scope.name)>-1) {
                document.getElementById("info").textContent = "该昵称已被占用,请重新输入";
            }
            else{
                socket.emit(‘setUserName‘, $scope.name);
            }
        }
        else{
            socket.emit(‘setUserName‘, $scope.name);
        }
    }
    else{
        document.getElementById(‘name‘).focus();
    }
};

验证成功后,对用户输入消息要触发的事件:

$scope.sendMsg = function(data){
    var date = new Date();
    data = $scope.content;
    if($scope.content !== ‘‘){
        socket.emit(‘sendMsg‘,data);
    }
    $scope.content = ‘‘;
};

程序的部分运行测试结果:

浏览器输入localhost后展示的用户登录界面:

技术分享

昵称重复后的提示:

技术分享

昵称输入成功后进入当前用户的聊天界面:

技术分享

 

源码地址:github

参考学习:

Node.js + Web Socket 打造即时聊天程序嗨聊

基于websocket的一个简单的聊天室

以上是关于websocket实现简单聊天程序的主要内容,如果未能解决你的问题,请参考以下文章

Vue+Websocket<简单实现聊天功能>

使用node+vue实现简单的WebSocket聊天功能

WebSocket实现简单的在线聊天

Node + H5 + WebSocket + Koa2 实现简单的多人聊天

websocket IM 聊天教程-教你用 GoEasy 快速实现 IM 聊天

实现websocket多人聊天,很简单