Swoole + websocket 实现简易版网页聊天室

看代码之前呢,先来一张预览图

(因为这里填写图片简书会进行采集,但是采集失败,失败几次,放弃了,所以这里放图片链接)点击查看 预览图

PHP 部分

服务器呢,使用的是swoole 的websocket + table ,websocket 负责与前JS 进行通信,table 则负责记录用户信息,实现进程间的数据共享

<?php

use Swoole\Table;
use Swoole\WebSocket\Server;
use Swoole\Http\Request;
use Swoole\WebSocket\Frame;

/**
 * Class ChatServer
 * @author: gaolei 2021/3/9 2:48 下午
 * @property Server $server
 * @property Table $table
 */
class ChatServer
{

    protected $server;
    protected $table;

    const ACTION_JOIN = 'join';
    const ACTION_SEND = 'send';

    public function __construct(string $host, int $port)
    {
        $this->server = new Server($host, $port);
        $this->server->on('Open', [$this, 'onOpen']);
        $this->server->on('Message', [$this, 'onMessage']);
        $this->server->on('Close', [$this, 'onClose']);

        // 内存表 实现进程间共享数据,也可以使用redis替代
        $this->createTable();
    }

    private function createTable()
    {
        $this->table = new Table(1024);
        $this->table->column('id', Table::TYPE_INT);
        $this->table->column('name', Table::TYPE_STRING, 255);
        $this->table->create();
    }

    public function run()
    {
        $this->server->start();
    }

    public function onOpen(Server $server, Request $request)
    {
        $user = ['id' => $request->fd, 'name' => ''];
        $this->table->set($request->fd, $user);
    }

    public function onMessage(Server $server, Frame $frame)
    {
        $params = json_decode($frame->data, true);
        // 接收到消息,根据消息类型进行不同的处理
        switch ($params['action']) {
            case self::ACTION_JOIN:
                $this->join($frame->fd, $params['value']);
                break;
            case self::ACTION_SEND:
                $this->send($frame->fd, $params['value']);
                break;
        }
    }

    public function onClose(Server $server, int $fd)
    {
        $user = $this->table->get($fd);
        $this->table->del($fd);
        echo "{$user['name']} 退出聊天室 \n";
        $time = date('Y-m-d H:i:s');
        // 将当前用户 '退出的消息' 推送给聊天室 剩余的其他用户
        foreach ($this->table as $member) {
            if ($member['id'] === $fd) {
                continue;
            }
            $res = ['code' => 'someone_out', 'data' => [
                'time' => $time, 'msg' => "{$user['name']} 退出聊天室",
                'member_id' => $user['id']
            ]];
            $this->server->push($member['id'], json_encode($res));
        }
    }

    private function getMembers()
    {
        $members = [];
        foreach ($this->table as $member) {
            $members[] = $member;
        }
        return $members;
    }

    private function join(int $fd, string $name)
    {
        $user = $this->table->get($fd);
        $user['name'] = $name;
        // 将当前用户的信息 记录到 table 中
        $this->table->set($fd, $user);
        $time = date('Y-m-d H:i:s');
        echo "欢迎 {$fd}:{$name} 加入聊天室\n";
        $members = $this->getMembers();
        $membersObj = array_combine(array_column($members, 'id'), array_column($members, 'name'));
        foreach ($this->table as $member) {
            if ($member['id'] === $fd) {
                // 将'加入成功'的消息 与 '当前聊天室的人员信息' 推送给当前用户
                $res = ['code' => 'add_success', 'data' => [
                    'time' => $time, 'msg' => "欢迎 {$name} 加入聊天室",
                    'members' => $members, 'membersObj' => $membersObj,
                    'self' => $member
                ]];
            }else{
                // 将当前用户 '加入的消息' 推送给聊天室 的其他用户
                $res = ['code' => 'new_member', 'data' => [
                    'time' => $time, 'msg' => "欢迎 {$name} 加入聊天室",
                    'user' => $user
                ]];
            }
            $this->server->push($member['id'], json_encode($res));
        }
    }

    private function send(int $fd, string $msg)
    {
        $user = $this->table->get($fd);
        echo "{$user['name']} send : $msg\n";
        $time = date('Y-m-d H:i:s');
        foreach ($this->table as $member) {
            // 只将消息推送个 已注册昵称 且 连接有效的聊天室用户
            if($member['name'] && $this->server->isEstablished($member['id'])){
                if ($member['id'] == $fd) {
                    // 将'发送成功'的消息推送给当前用户
                    echo "send to: {$member['id']} {$member['name']} success" . "\n";
                    $res = ['code' => 'send_success', 'data' => [
                        'time' => $time, 'msg' => $msg, 'author' => $fd
                    ]];
                }else{
                    // 将当前用户 '发送的消息' 推送给聊天室 的其他用户
                    echo "send to: {$member['id']} {$member['name']} success" . "\n";
                    $res = ['code' => 'new_message', 'data' => [
                        'time' => $time, 'msg' => $msg, 'author' => $fd
                    ]];
                }
                $this->server->push($member['id'], json_encode($res));
            }
        }
    }
}
$chat = new ChatServer('0.0.0.0', 9501);
$chat->run();

Html

<!DOCTYPE html>
<html lang="en">
<head>
    <meta charset="UTF-8">
    <title>ChatRoom</title>
    <link rel="stylesheet" href="statics/chat-room.css">
</head>
<body>
<div class="chart-room">
    <div class="room-header" style="line-height: 30px">
        <div class="welcome-msg">欢迎 <span id="nickname"></span> 来到 高雷 的 网页聊天室</div>
    </div>
    <div class="room-body">
        <div id="msg-list"></div>
        <div id="members"></div>
    </div>
    <div class="room-footer">
        <textarea name="message" id="message"></textarea>
        <div id="send_btn">
            发送<span class="notice">快捷键:Ctrl/command + Enter</span>
        </div>
    </div>
</div>
<script src="statics/chat-room.js"></script>
</body>
</html>

Script

var msgIpt = document.getElementById('message'),
    sendBtn = document.getElementById('send_btn'),
    msgList = document.getElementById('msg-list'),
    nickname, members, membersObj,mouseIn = false;

// 用户加入之前需要填写昵称,昵称显示在用户列表
nickname = getNick()
document.getElementById('nickname').innerText = nickname;
var wsAddress = 'ws://127.0.0.1:9501',
    socket = new WebSocket(wsAddress),
    mine;
socket.onopen = function (evt) {
    addNotice('连接成功');
    register();
};
socket.onclose = function (evt) {
    console.log("Disconnected");
};
socket.onerror = function (evt, e) {
    console.log('Error occured: ' + evt.data);
};
socket.onmessage = recv;

// 获取用户的昵称
function getNick() {
    var nick = (prompt('请输入您的昵称') || '').replace(/(^\s*)|(\s*$)/g, "");
    if (nick.length > 0) {
        addNotice('连接中...')
        return nick;
    }
    return getNick();
}

// JSON 对象 转 json 字符串
function toJSON(data) {
    return JSON.stringify(data);
}

// 注册并加入聊天室
function register() {
    socket.send(toJSON({action: 'join', value: nickname}));
}

// 监听 发送按钮
sendBtn.onclick = function () {
    var msg = msgIpt.value;
    if (msgIpt.value.length < 1) {
        return msgIpt.focus();
    }
    socket.send(toJSON({action: 'send', value: msg}));
    msgIpt.value = '';
}

// 监听消息发送组合键
document.onkeydown = function (oEvent) {
    var e = event || window.event || arguments.callee.caller.arguments[0];
    if ((e.keyCode === 13) && (e.ctrlKey || e.metaKey)) {
        sendBtn.click();
    }
}

msgList.onmouseover = function () {
    mouseIn = true;
}
msgList.onmouseleave = function () {
    mouseIn = false;
}

// 接受处理信息
function recv(evt) {
    try {
        var response = JSON.parse(evt.data)
        switch (response.code) {
            case "add_success":
                return addSuccess(response.data)
            case "new_member":
                return newMember(response.data)
            case 'new_message':
                return newMessage(response.data)
            case 'send_success':
                return sendSuccess(response.data)
            case 'someone_out':
                return someoneOut(response.data)
        }
    } catch (e) {
        // 打印错误信息
        console.log([e, evt]);
    }
}

// 加入聊天室成功,将已有的人员写入列表
function addSuccess(data) {
    console.log(['addSuccess', data])
    addNotice(data.time + ' ' + data.msg)
    mine = data.self;
    members = data.members;
    membersObj = data.membersObj;
    console.log([mine, members, membersObj])
    members.forEach(function (v) {
        addMemberLine(v)
    })
}

// 消息发送成功
function sendSuccess(data) {
    console.log(['sendSuccess', data])
    addMessageLine(data, 1)
}

// 接受到新的消息
function newMessage(data) {
    console.log(['newMessage', data])
    addMessageLine(data, 2)
}

// 接收到有新的人加入到聊天室
function newMember(data) {
    console.log(['newMember', data])
    addNotice(data.time + ' ' + data.msg)
    members.push(data.user);
    membersObj[data.user.id] = data.user.name;
    console.log([mine, members, membersObj])
    addMemberLine(data.user);
}

// 有人退出聊天室
function someoneOut(data) {
    console.log(['someoneOut', data])
    addNotice(data.time + ' ' + data.msg)
    removeMemberLine(data.member_id)
}

// 添加 提示信息
function addNotice(msg) {
    var dom = document.createElement('div');
    dom.className = 'notice';
    dom.innerText = msg;
    msgList.appendChild(dom);
}

// 添加一行消息展示
function addMessageLine(data, type) {
    if(!membersObj[data.author]){
        return;
    }
    var box = document.createElement('div'),
        author = document.createElement('div'),
        msg = document.createElement('div'),
        name = document.createElement('span'),
        time = document.createElement('span');
    box.className = type === 1 ? 'mine' : 'other';
    author.className = 'author';
    msg.className = 'msg';
    name.className = 'name';
    time.className = 'time';
    name.innerText = membersObj[data.author];
    time.innerText = data.time;
    author.appendChild(name);
    author.innerHTML += ':'
    author.appendChild(time);
    msg.innerText = data.msg;
    box.appendChild(author);
    box.appendChild(msg);
    msgList.appendChild(box);
    if(mouseIn){
        return;
    }
    msgList.scrollTop = msgList.scrollHeight;
}

// 新的人加入,将用户人员信息写入列表
function addMemberLine(user) {
    var memberList = document.getElementById('members'),
        item = document.createElement('div');
    if (user.id == mine.id) {
        item.className = 'member-item self'
    } else {
        item.className = 'member-item'
    }
    item.innerText = user.name;
    item.id = 'member_' + user.id;
    memberList.appendChild(item);
}

// 有人退出网页,在用户列表将其移除
function removeMemberLine(id) {
    var line = document.getElementById('member_'+id);
    if(line)line.remove();
}


Css

* {margin: 0;padding: 0}
html, body {height: 100%;}
body {
    display: flex;justify-content: center;align-items: center;background: #000
}

.chart-room {
    width: 1000px;height: 100%;max-height: 800px;margin: auto;background: #333;box-shadow: 0 0 10px #222;
    padding: 5px;color: #888;display: flex;flex-wrap: wrap;align-items: center;
}
.chart-room > div {
    width: 100%;background: #333;
}

/* 头部信息展示部分 */
.room-header {
    height: 80px;border-bottom: 1px solid #888;
}
.room-header #nickname {
    color: red
}

/* 消息展示部分 */

.room-body {
    display: flex;width: 100%;height: calc(100% - 185px);
}
.room-body #members {
    width: 150px;background: #2d3a4c;padding: 8px 0;overflow-y: scroll;
}
.room-body #members .member-item {
    line-height: 20px;padding: 0 5px;font-size: 13px;
}
.room-body #members .member-item.self {
    color: #f54cd8;
}
.room-body #members .member-item:hover {
    background: #384153;
}

.room-body #msg-list {
    width: calc(100% - 150px);overflow-y: scroll;padding: 5px;
}
.room-body #msg-list > * {margin-bottom: 10px}
.room-body #msg-list .notice{
    text-align: center;font-size: 13px;
}

.room-body #msg-list .other,.room-body #msg-list .mine{
    display: flex;
}
.room-body #msg-list .msg,.room-body #msg-list .author{
    display: flex;align-items: center;padding: 5px;line-height: 27px;
}
.room-body #msg-list .author .time{font-size: 12px}
.room-body #msg-list .msg{
    background: #666;border-radius: 3px;font-size: 13px;line-height: 17px;
    max-width: 200px;padding: 3px 10px;height: auto;
}
.room-body #msg-list .msg::selection{
    background: #336666;color: red;
}

/* 其他人发送的消息*/
.room-body #msg-list .other{flex-direction: row}

/* 我自己发送的消息 */
.room-body #msg-list .mine{flex-direction: row-reverse}

/* 发送消息部分 */
.room-footer {
    height: 100px;border-top: 1px solid #888;display: flex;align-items: center;
}

.room-footer #message {
    resize: none;width: calc(100% - 180px);height: 89px;line-height: 20px;border-top: none;
    outline: none;padding: 5px;overflow-y: scroll;color: #888;caret-color: #aaa;
    background: #333;
}

.room-footer #send_btn {
    width: 180px;text-align: center;height: 100px;line-height: 100px;
    user-select: none;background: #366;color: #ccc;position: relative;
}

.room-footer #send_btn:hover {
    background: #355;
}

.room-footer #send_btn:active {
    background: #344;
}

.room-footer #send_btn .notice {
    position: absolute;bottom: 61px;width: 100%;font-size: 12px;color: #999;
    background: none;height: 20px;left: 0;
}

原文链接 Swoole + websocket 实现简易版网页聊天室

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 214,776评论 6 496
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 91,527评论 3 389
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 160,361评论 0 350
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 57,430评论 1 288
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 66,511评论 6 386
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 50,544评论 1 293
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,561评论 3 414
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,315评论 0 270
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,763评论 1 307
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,070评论 2 330
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,235评论 1 343
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,911评论 5 338
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,554评论 3 322
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,173评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,424评论 1 268
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 47,106评论 2 365
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,103评论 2 352

推荐阅读更多精彩内容