TP5.1 使用websocket实现聊天功能(windows环境)

核心类(用到了模型)

<?php
/**
 * Created by PhpStorm.
 * User: Administrator
 * Date: 2020/11/11
 * Time: 16:37
 */

namespace app\common\controller;

use app\model\Chat;

class SocketService
{
    private $address  = '0.0.0.0';
    private $port = 1999;
    private $_sockets;

    public function __construct($address = '', $port='')
    {
        if(!empty($address)){
            $this->address = $address;
        }
        if(!empty($port)) {
            $this->port = $port;
        }
    }

    public function service(){
        //获取tcp协议号码。
        $tcp = getprotobyname("tcp");
        $sock = socket_create(AF_INET, SOCK_STREAM, $tcp);
        socket_set_option($sock, SOL_SOCKET, SO_REUSEADDR, 1);
        if($sock < 0)
        {
            throw new Exception("failed to create socket: ".socket_strerror($sock)."\n");
        }
        socket_bind($sock, $this->address, $this->port);
        socket_listen($sock, $this->port);
        echo "listen on $this->address $this->port ... \n";
        $this->_sockets = $sock;
    }

    public function run(){
        $this->service();
        $clients[] = $this->_sockets;
        while (true){
            $changes = $clients;
            $write = NULL;
            $except = NULL;
            socket_select($changes,  $write,  $except, 0, 10);
            foreach ($changes as $key => $_sock){
                if($this->_sockets == $_sock){ //判断是不是新接入的socket
                    if(($newClient = socket_accept($_sock))  === false){
                        die('failed to accept socket: '.socket_strerror($_sock)."\n");
                    }
                    $line = trim(socket_read($newClient, 1024));
                    $this->handshaking($newClient, $line);
                    //获取client ip
                    socket_getpeername ($newClient, $ip);
                    $clients[$ip] = $newClient;
                } else {
                    socket_recv($_sock, $buffer,  2048, 0); // 接收数据
                    $msg = $this->message($buffer); // 转码数据
                    $data = json_decode($msg, true);
                    if(!empty($data)){
                        echo "{$key} clinet msg:",$msg,"\n";
                        $this->send($clients, $msg);

                        $db = new Chat();
                        $db->allowField(true)->save($data);
                    }else{
                        unset($clients[$_sock]);
                    }
                }
            }
        }
        // 关闭监听的socket
        socket_close($this->_sockets);
    }

    /**
     * 握手处理
     * @param $newClient socket
     * @return int  接收到的信息
     */
    public function handshaking($newClient, $line){

        $headers = array();
        $lines = preg_split("/\r\n/", $line);
        foreach($lines as $line)
        {
            $line = chop($line);
            if(preg_match('/\A(\S+): (.*)\z/', $line, $matches))
            {
                $headers[$matches[1]] = $matches[2];
            }
        }
        $secKey = $headers['Sec-WebSocket-Key'];
        $secAccept = base64_encode(pack('H*', sha1($secKey . '258EAFA5-E914-47DA-95CA-C5AB0DC85B11')));
        $upgrade  = "HTTP/1.1 101 Web Socket Protocol Handshake\r\n" .
            "Upgrade: websocket\r\n" .
            "Connection: Upgrade\r\n" .
            "WebSocket-Origin: $this->address\r\n" .
            "WebSocket-Location: ws://$this->address:$this->port/websocket/websocket\r\n".
            "Sec-WebSocket-Accept:$secAccept\r\n\r\n";
        return socket_write($newClient, $upgrade, strlen($upgrade));
    }

    /**
     * 解析接收数据
     * @param $buffer
     * @return null|string
     */
    public function message($buffer){
        $len = $masks = $data = $decoded = null;
        $len = ord($buffer[1]) & 127;
        if ($len === 126)  {
            $masks = substr($buffer, 4, 4);
            $data = substr($buffer, 8);
        } else if ($len === 127)  {
            $masks = substr($buffer, 10, 4);
            $data = substr($buffer, 14);
        } else  {
            $masks = substr($buffer, 2, 4);
            $data = substr($buffer, 6);
        }
        for ($index = 0; $index < strlen($data); $index++) {
            $decoded .= $data[$index] ^ $masks[$index % 4];
        }
        return $decoded;
    }

    /**
     * 发送数据
     * @param $newClinet 新接入的socket
     * @param $msg   要发送的数据
     * @return int|string
     */
    public function send($sockets, $msg){
        $msg = $this->frame($msg);
        foreach ($sockets as $changed_socket) {
            @socket_write($changed_socket, $msg, strlen($msg));
        }
    }


    public function frame($s) {
        $a = str_split($s, 125);
        if (count($a) == 1) {
            return "\x81" . chr(strlen($a[0])) . $a[0];
        }
        $ns = "";
        foreach ($a as $o) {
            $ns .= "\x81" . chr(strlen($o)) . $o;
        }
        return $ns;
    }

    /**
     * 关闭socket
     */
    public function close(){
        return socket_close($this->_sockets);
    }
}

$sock = new SocketService();
$sock->run();

 

绑定入口

 

 

在application创建common/controller

把核心类放进去

在public创建server.php

<?php
/**
 * Created by PhpStorm.
 * User: Administrator
 * Date: 2020/11/11
 * Time: 16:13
 */// [ 应用入口文件 ]


namespace think;

// 加载基础文件
require __DIR__ . '/../thinkphp/base.php';

// 执行应用并响应(绑定)
Container::get('app')->bind('common/Socket_service')->run()->send();

运行服务端

在cmd或者treminal中

cd到这个应用的public目录中

php server.php

 

 

 服务端已启动,正在监听1999端口

 

客户端

<{include file="public/header" /}>
<style>
    .fl{ float: left; }
    .fr{ float: right; }
    li{ display: block; background-color: #ccc; line-height: 30px; width: 100%; height: 30px; }
    .block{ width: 90%; height: 500px; float: left; margin: 0 auto; }
    .msg{ width: 100%; height: 400px; background-color: #fff; margin-bottom: 400px; }
    .receive{ width: 51%; }
    .send{ width: 51%; }
    .red{ color: red; float: right; }
</style>
</head>
<body>
<div class="block">
    <div class="msg">
        <div class="send fl">
            <ul>
                <li>1111111 <span class="red">2020</span></li>
            </ul>
        </div>
        <div class="receive fr">
            <ul>
                <li>1111111 <span class="red">2020</span></li>
            </ul>
        </div>
        <div class="bottom">
            <textarea name="" id="content" cols="30" rows="10" class="layui-textarea"></textarea>
            <button class="layui-btn" id="send">发送</button>
        </div>
    </div>
</div>
</body>
</html>

<script>

    $(function () {
        var wsurl = 'ws://127.0.0.1:1999';
        var websocket;
        var i = 0;
        var uid = 2;
        if (window.WebSocket) {
            websocket = new WebSocket(wsurl);

            //连接建立
            websocket.onopen = function (evevt) {
                console.log("Connected to WebSocket server.");
                // $('.receive').append('<p class="bg-info message"><i class="glyphicon glyphicon-info-sign"></i>Connected to WebSocket server!</p>');
                $('.receive').append('<li>Connected to WebSocket server! <span class="red">'+ getTime(new Date().getTime()) +'</span></li>');
            }
            //收到消息
            websocket.onmessage = function (event) {
                var msg = JSON.parse(event.data); //解析收到的json消息数据

                i++;
                if (msg.send_uid == uid) {
                    $('.send').append('<li> '+ msg.content +' <span class="red">'+ getTime(new Date().getTime()) +'</span></li>');
                }else{
                    $('.receive').append('<li> '+ msg.content +'<span class="red">'+ getTime(new Date().getTime()) +'</span></li>');
                }

                $('#message').val('');
                window.location.hash = '#' + i;
            }

            //发生错误
            websocket.onerror = function (event) {
                i++;
                console.log("Connected to WebSocket server error");
                window.location.hash = '#' + i;
            }

            //连接关闭
            websocket.onclose = function (event) {
                i++;
                console.log('websocket Connection Closed. ');
                window.location.hash = '#' + i;
            }

            function send() {
                var msg = {
                    type: 'send',
                    send_uid: 1,
                    receive_uid: 2,
                    content: '111111',
                    img: '',
                    voice: ''
                };
                try {
                    websocket.send(JSON.stringify(msg));
                } catch (ex) {
                    console.log(ex);
                }
            }
            //按下enter键发送消息
            $(window).keydown(function (event) {
                if (event.keyCode == 13) {
                    console.log('user enter');
                    send();
                }
            });
            //点发送按钮发送消息
            $('#send').bind('click', function () {
                send();
            });
        }
        else {
            alert('该浏览器不支持web socket');
        }
    });
</script>

 

结果

 

 

 

 

posted @ 2020-11-13 13:03  张永峰z  阅读(2548)  评论(4编辑  收藏  举报