swoole 服務(wù)端120行代碼構(gòu)建一個websocket 聊天室.

韓冰發(fā)布于2019-06-27 17:05 / 550人閱讀

摘要：的異步并行高性能網(wǎng)絡(luò)通信引擎，使用純語言編寫，提供了語言的異步多線程服務(wù)器，異步網(wǎng)絡(luò)客戶端，異步，異步，數(shù)據(jù)庫連接池，，消息隊列，毫秒定時器，異步文件讀寫，異步查詢。內(nèi)置了服務(wù)器端客戶端服務(wù)器端。

swoole :http://www.swoole.com/
PHP的異步、并行、高性能網(wǎng)絡(luò)通信引擎，使用純C語言編寫，提供了PHP語言的異步多線程服務(wù)器，異步TCP/UDP網(wǎng)絡(luò)客戶端，異步MySQL，異步Redis，數(shù)據(jù)庫連接池，AsyncTask，消息隊列，毫秒定時器，異步文件讀寫，異步DNS查詢。 Swoole內(nèi)置了Http/WebSocket服務(wù)器端/客戶端、Http2.0服務(wù)器端。

7-22更新------ 昨天已經(jīng)用面向?qū)ο蠛蛂edis重做了一次,現(xiàn)在支持切換分組,私信了.并且對其他地方也做了一些優(yōu)化. 視頻演示: http://www.bilibili.com/video... 視頻中的列表沒有正常刷新漏洞已修復 github : https://github.com/buffge/buf...

demo 地址 http://www.buffge.xin/buffcha... end------ 7-27更新

今天服務(wù)器被人攻擊了他攻擊也許就是為了測試或者玩我,而且他也沒有得逞
他利用我的redis的漏洞來攻擊我,因為我本地用一個可視化軟件去查看我服務(wù)器的數(shù)據(jù)庫,所以我
開放的權(quán)限并且沒有加密,之后有人想透過redis 給我服務(wù)器上傳腳本. 不過這個腳本就是重復運行而已,應(yīng)該是個
練手的.他這個腳本沒有運行起來,因為我redis 的權(quán)限很低. 他首先給我redis里面加入了這樣一句話

然后想讓這個腳本在我服務(wù)器中循環(huán)運行.真的可惡.我好心開源給大家看了玩玩,我也沒空去搞那么安全那么好但是就是有這種人渣來黑我.
等過幾天我忙完了手上的事重新開啟這個聊天室并添加新功能和新樣式不說了,我先舉報去了,我已經(jīng)找到他的ip和他的群了.就看tx管不管了.bye~
-------end

昨晚一直在學習swoole. 終于搞出了一個多人聊天室.新手可以看看,高玩請直接右上角.

思路:

 1:用戶登錄主頁面,此時后臺將用戶放入數(shù)據(jù)庫(我不會redis memcached 等),
 數(shù)據(jù)庫格式為

第一次登錄只需要將fd插入就行了.

2:此時如果不進行注冊,那么就無法發(fā)送信息給全局.
如果注冊了 那么填寫用戶名,ajax請求 將用戶名進行加鹽信息摘要獲得token.
注冊成功后發(fā)送token:117fdba5e4d4050ed18ffd85ac86ed5b5d72dec6a271f4e4f6a6e03f4b957cd4:user_name:張三123
這樣的一個字符串給服務(wù)器,服務(wù)器會進行token 和名稱驗證,如果是正確的token.那么將用戶信息插入到數(shù)據(jù)庫.
并通知其他websocket 有新用戶上線.

3:快樂的聊天吧.

路上遇到的坑:

        1:我原本準備在server.php(就是swoole 后臺服務(wù))創(chuàng)建一個對象,
        存儲所有訪客的信息 姓名等等,但是他不共享這個對象,我覺得可以用redis 緩存起來,
        我這里就寫了玩的,就用mysql弄了,也沒有面向?qū)ο?
        2:不能用session  反正用了 之后就是各種問題,我原本是準備用session 存儲用戶名稱的.
        后臺沒辦法想出一個加密token 然后存入數(shù)據(jù)庫,
        具體的token是這樣的.

注冊時候用鹽和sha256生成token,然后去swoole 服務(wù)器那邊驗證的時候也用同樣的用戶名和鹽進行計算,看是否一樣.如果一樣說明是正確的登錄了. 我不知道這個方法好不好.以前只會session,這個是昨晚突然想到的.

服務(wù)器那邊的日志如下:

4個用戶的聊天界面如下:

服務(wù)器端代碼看注釋.寫的渣別說,我是菜鳥.

   
  <#時間 = "00:46:16">
  <#人物 = "buff" >
  <#備注 = " ">
 */
if (php_sapi_name() !== "cli") {
    exit("使用cli模式");
}
$serv = new SwooleWebsocketServer("192.168.1.109", 9501);
//回調(diào)函數(shù) 新建一個websocket連接時 觸發(fā)的事件
$serv->on("Open", function($server, $req) {
    $mysqli = new mysqli("127.0.0.1", "buffge", "daimin", "buffchat");
    $charsetsql = "set names utf8";
    $mysqli->query($charsetsql);
    //將新登陸的用戶 保存到mysql
    $sql = "INSERT INTO `users` (`fd`) VALUES ("{$req->fd}")";
    $mysqli->query($sql);
    if ($mysqli->affected_rows !== 1) {
        echo "插入數(shù)據(jù)失敗!sql== {$sql}";
    }
    echo "新客戶端連接: " . $req->fd . "時間:" . date("Y-n-j H:i:s") . "
";
    $userlist = "";
    //檢查當前共有多少用戶在線
    $sql = "select `user_name` from users where `user_name`!=""";
    $res = $mysqli->query($sql);
    for ($i = 0; $i < $res->num_rows; $i++) {
        $result = $res->fetch_assoc();
        $userlist .= (""" . $result["user_name"] . "",");
    }
    $userlist = substr($userlist, 0, strlen($userlist) - 1);
    //通知用戶 當前在線用戶列表
    $server->push($req->fd, "{"code":"4","users":[{$userlist}]}");
    $res->free();
    $mysqli->close();
});
//當收到用戶的消息時 觸發(fā)事件
$serv->on("Message", function($server, $frame) {
    $mysqli = new mysqli("127.0.0.1", "buffge", "daimin", "buffchat");
    $charsetsql = "set names utf8";
    $mysqli->query($charsetsql);
    $sql = "select * from users where fd={$frame->fd}";
    $res = $mysqli->query($sql);
    $result = $res->fetch_assoc();
    //獲取當前發(fā)消息的人的名稱
    $user_name = $result["user_name"];
    echo "收到來自客戶端{$frame->fd}的消息: " . $frame->data . "
";
    //當用戶是第一個注冊時候(發(fā)送的語句前面5個字是token)
    if (strpos($frame->data, "token") === 0) {
        //如果數(shù)據(jù)庫中沒有令牌
        if ($result["token"] === null) {
            $userData = explode(":", $frame->data);
            $hash = hash("sha256", "daimin" . $userData[3]);
            if ($userData[1] == $hash) {
                $sql = "UPDATE `users` SET `token` = "{$hash}",`user_name` = "{$userData[3]}" WHERE `fd` = {$frame->fd}";
                $mysqli->query($sql);
                if ($mysqli->affected_rows !== 1) {
                    echo "更新用戶信息失敗, sql === {$sql}";
                    $server->push($frame->fd, "{"code":"5","mes":"更新用戶信息失敗"}");
                }
                $server->push($frame->fd, "{"code":"3","user_name":"{$userData[3]}"}");
                $res->free();
                $userlist = "";
                //檢查當前共有多少用戶在線
                $sql = "select `user_name` from users where `user_name`!=""";
                $res = $mysqli->query($sql);
                for ($i = 0; $i < $res->num_rows; $i++) {
                    $result = $res->fetch_assoc();
                    $userlist .= (""" . $result["user_name"] . "",");
                }
                $userlist = substr($userlist, 0, strlen($userlist) - 1);
                //通知所有用戶 當前在線用戶列表
                foreach ($server->connections as $fd) {
                    $server->push($fd, "{"code":"4","users":[{$userlist}]}");
                }
                echo "新注冊用戶 {$userData[3]}
";
            }
            else {
                $server->push($frame->fd, "{"code":"5","mes":"token錯誤"}");
            }
        }
        //如果只是發(fā)送包含token這個字符串的語句 群發(fā)
        else {
            foreach ($server->connections as $fd) {
                $server->push($fd, "{"code":"2","mes":"{$frame->data}","user_name":"{$user_name}"}");
            }
        }
    }
    //如果不是注冊用戶
    else {
        //如果沒有令牌
        if ($result["token"] === null) {
            $server->push($frame->fd, "{"mes":"請先登錄!"}");
            return;
        }
        //將換行轉(zhuǎn)換為br
        $mes = nl2br($frame->data);
        //格式化json
        $mes = str_replace("
", "", $mes);
        //群發(fā)消息
        foreach ($server->connections as $fd) {
            $server->push($fd, "{"code":"2","mes":"{$mes}","user_name":"{$user_name}"}");
        }
    }
    $res->free();
    $mysqli->close();
});
//當websocket 斷開連接時 觸發(fā)事件
$serv->on("Close", function($server, $fd) {
    $mysqli = new mysqli("127.0.0.1", "buffge", "daimin", "buffchat");
    $charsetsql = "set names utf8";
    $mysqli->query($charsetsql);
    $sql = "DELETE FROM `users` WHERE `users`.`fd` = {$fd}";
    //當用戶退出時 刪除信息
    $mysqli->query($sql);
    if ($mysqli->affected_rows !== 1) {
        echo "刪除用戶信息失敗, sql === {$sql}";
    }
    echo "客戶端{$fd}已斷開連接
";
    $mysqli->close();
});

$serv->start();

前端js 核心

websocket.onmessage = function (evt) {
    console.log(evt.data);
    var data = JSON.parse(evt.data);
    //根據(jù)code 判斷業(yè)務(wù)
    switch (data.code) {
        //個人消息
        case "1":
            break;
            //全局消息
        case "2":
            var $time = bf_get_time();//這是我自己寫的一個獲取時間函數(shù)
            var $mes = data.mes;//消息內(nèi)容
            var $user_name = data.user_name;//發(fā)消息的人
            var $who = $user_name === selfName ? "self" : "other";//根據(jù)名稱設(shè)置格式
            var $append = "" + $user_name + " " + $time + "
" + $mes + ""
            $(".gui_content").append($append);
            var $cont_scrTop = $(".gui_content").scrollTop();
            var $list_height = $(".mes_item:last-of-type").height();
            $(".gui_content").animate({"scrollTop": $cont_scrTop + $list_height}, 100);
            break;
            //通知注冊用戶成功消息
        case "3":
            var $append = "";
            $append += data.user_name;
            $append += "";
            $(".user_list ul").append($append);
            $(".gui_user").html("歡迎 " + data.user_name + "");
            selfName = data.user_name;
            break;
            //更新當前在線列表 刪除的我沒寫~
        case "4":
            var $append = "";
            var $nowUser = $(".user_list ul li");
            outer:
                    for (var user in data.users) {
                for (var i = 0; i < $nowUser.length; i++) {
                    if ($nowUser[i].innerText === data.users[user]) {
                        continue outer;
                    }
                }
                console.log(data.users[user]);
                $append += "";
                $append += data.users[user];
                $append += "";
            }
            $(".user_list ul").append($append);
            break;
        case "5":
            alert(data.mes);
            break;
    }
    console.log("收到來自服務(wù)器的消息: " + data.code);

};

等有空了用面向?qū)ο髮懸淮伟裮ysql 換成redis,現(xiàn)在的代碼有點亂,一些功能也沒有寫,因為太困了,連續(xù)寫了8小時.

如果有大神看的話能不能告訴我哪里該改進. 比如怎么驗證用戶,用哪種方法存儲之類的.謝謝了