swoole和websocket如何實(shí)現(xiàn)聊天室功能

這篇文章給大家分享的是有關(guān)swoole和websocket如何實(shí)現(xiàn)聊天室功能的內(nèi)容。小編覺得挺實(shí)用的，因此分享給大家做個(gè)參考，一起跟隨小編過來看看吧。

首先，定義一個(gè) swoole_lock 和 swoole_websocket_server ，并且配置參數(shù)，具體參數(shù)詳情可以去swoole官網(wǎng)查看。

public function start()
{
  $this->lock = new swoole_lock(SWOOLE_MUTEX);　　　　　　　　　　　　

  // 對(duì)文件或數(shù)組進(jìn)行鎖操作，已達(dá)到同步
  $this->server = new swoole_websocket_server($this->addr, $this->port);　　

  // swoole提供的Websocket Server
  $this->server->set(array(
   'daemonize' => 0,
   'worker_num' => 4,
   'task_worker_num' => 10,
   'max_request' => 1000,
   'log_file' => ROOT_PATH . 'storage\\logs\\swoole.log'  

  // swoole日志路徑，必須是絕對(duì)路徑
  ));

  $this->server->on('open', array($this, 'onOpen'));
  $this->server->on('message', array($this, 'onMessage'));
  $this->server->on('task', array($this, 'onTask'));
  $this->server->on('finish', array($this, 'onFinish'));
  $this->server->on('close', array($this, 'onClose'));

　　 // 啟動(dòng)服務(wù)
  $this->server->start();
}

當(dāng)有客戶端鏈接時(shí)，簡(jiǎn)單記錄客戶端的信息。

public function onOpen($server, $request)
  {
   $message = array(
    'remote_addr' => $request->server['remote_addr'],
    'request_time' => date('Y-m-d H:i:s', $request->server['request_time'])
   );
   write_log($message);
  }

當(dāng)有客戶端發(fā)送信息時(shí)，對(duì)信息進(jìn)行處理。

public function onMessage($server, $frame)
  {
   $data = json_decode($frame->data);

   switch ($data->type) {
    case 'init':
    case 'INIT':
     $this->users[$frame->fd] = $data->message;　　、

   // 記錄每個(gè)鏈接的信息，同樣不要嘗試打印出來看，因?yàn)槟阒荒芸吹阶约旱逆溄有畔?
     $message = '歡迎' . $data->message . '加入了聊天室';
     $response = array(
      'type' => 1, // 1代表系統(tǒng)消息，2代表用戶聊天
      'message' => $message
     );
     break;
    case 'chat':
    case 'CHAT':
     $message = $data->message;
     $response = array(
      'type' => 2, // 1代表系統(tǒng)消息，2代表用戶聊天
      'username' => $this->users[$frame->fd],
      'message' => $message
     );
     break;
    default:
     return false;
   }
　　　　　　　　
　　　　　　　// 將信息交給task處理
   $this->server->task($response);
  }

  public function onTask($server, $task_id, $from_id, $message)
  {
　　　　　　　// 迭代所有的客戶端鏈接，將消息推送過去。（如果你嘗試將 $this->server->connections 打印出來，那么你會(huì)發(fā)現(xiàn)他是空的。但當(dāng)時(shí)用 foreach 去循環(huán)時(shí)，它確實(shí)有用。）
   foreach ($this->server->connections as $fd) {
    $this->server->push($fd, json_encode($message));
   }
   $server->finish( 'Task' . $task_id . 'Finished' . PHP_EOL);
  }

最后，當(dāng)客戶端斷開鏈接時(shí)，利用鎖機(jī)制，同步刪除客戶端信息，并記錄日志。

public function onClose($server, $fd)
  {
   $username = $this->users[$fd];
   // 釋放客戶端，利用鎖進(jìn)行同步
   $this->lock->lock();
   unset($this->users[$fd]);
   $this->lock->unlock();

   if( $username ) {
    $response = array(
     'type' => 1, // 1代表系統(tǒng)消息，2代表用戶聊天
     'message' => $username . '離開了聊天室'
    );
    $this->server->task($response);
   }


   write_log( $fd . ' disconnected');
  }

服務(wù)端完了，下面就是客戶端，很簡(jiǎn)單，只需要用websocket鏈接就ok！

// websocket
  let address = 'ws://<?php echo CLIENT_CONNECT_ADDR . ':' . CLIENT_CONNECT_PORT ?>';
  let webSocket = new WebSocket(address);
  webSocket.onerror = function (event) {
   alert('服務(wù)器連接錯(cuò)誤，請(qǐng)稍后重試');
  };
  webSocket.onopen = function (event) {
   if(!sessionStorage.getItem('username')) {
    setName();
   }else {
    username = sessionStorage.getItem('username')
    webSocket.send(JSON.stringify({
     'message': username,
     'type': 'init'
    }));
   }
  };
  webSocket.onmessage = function (event) {
   console.log(event);
   let data = JSON.parse(event.data);
   if (data.type == 1) {
    $('#chat-list2').append('<li class="ui-border-tb"><span class="username">系統(tǒng)消息:</span><span class="message">' + data.message + '</span></li>');
   } else if (data.type == 2) {
    $('#chat-list2').append('<li class="ui-border-tb"><span class="username">' + data.username + ':</span><span class="message">' + data.message + '</span></li>');
   }

  };
  webSocket.onclose = function (event) {
   alert('散了吧，服務(wù)器都關(guān)了');
  };

感謝各位的閱讀！關(guān)于“swoole和websocket如何實(shí)現(xiàn)聊天室功能”這篇文章就分享到這里了，希望以上內(nèi)容可以對(duì)大家有一定的幫助，讓大家可以學(xué)到更多知識(shí)，如果覺得文章不錯(cuò)，可以把它分享出去讓更多的人看到吧！