老雷socket編程之PHP利用socket擴展實現聊天服務

老雷socket編程之PHP利用socket擴展實現聊天服務

socket聊天服務原理

PHP有兩個socket的擴展 sockets和streams
sockets
socket_create(AF_INET, SOCK_STREAM, SOL_TCP)
socket_write
socket_read
socket_close
客戶端
socket_connect($socket, $address, $service_port)；

服務端
socket_bind($sock, $address, $port)
socket_listen($sock)
socket_accept

Streams
客戶端
stream_socket_client
fwrite
fread
fclose($fp);
服務端
stream_socket_server
stream_set_blocking
stream_select
stream_socket_accept conn
stream_socket_recvfrom
stream_socket_sendto
stream_socket_shutdown

課後練習
使用sockets和streams擴展實現客戶端跟服務端的功能。

服務端代碼 客服端可使用udp&tcp測試工具

<?php

class SocketService
{
	public $host="tcp://0.0.0.0:8000";
    private $address;
    private $port;
    private $_sockets;
	public $clients;
	public $maxid=1000;
    public function __construct($address = '', $port='')
    {
            if(!empty($address)){
                $this->address = $address;
            }
            if(!empty($port)) {
                $this->port = $port;
            }
    }
	
	public function onConnect($client_id){
		echo  "Client client_id:{$client_id}   \n";
         
	}
	
	public function onMessage($client_id,$msg){
		//發給所有的
		foreach($this->clients as $kk=>$cc){
			if($kk>0){
				$this->send($cc, $msg);
			}								
		}	
	}
	
	public function onClose($client_id){
		echo "$client_id close \n";
	}
	
    public function service(){
        //獲取tcp協議號碼。
        $tcp = getprotobyname("tcp");
        $sock = stream_socket_server($this->host, $errno, $errstr);;
        
        if(!$sock)
        {
            throw new Exception("failed to create socket: ".socket_strerror($sock)."\n");
        }
		stream_set_blocking($sock,0);
        $this->_sockets = $sock;
		 echo "listen on $this->address $this->host ... \n";
    }
 
    public function run(){
        $this->service();
        $this->clients[] = $this->_sockets;
        while (true){
            $changes = $this->clients;
            //$write = NULL;
            //$except = NULL;
            stream_select($changes,  $write,  $except, NULL);
            foreach ($changes as $key => $_sock){
                if($this->_sockets == $_sock){ //判斷是不是新接入的socket
                    if(($newClient = stream_socket_accept($_sock))  === false){
						unset($this->clients[$key]);
						continue;
                    }
                    $line = trim(stream_socket_recvfrom($newClient, 1024));
                     
					$this->maxid++;
                    $this->clients[$this->maxid] = $newClient;
                    $this->onConnect($this->maxid);
                } else {
					$msg=@stream_socket_recvfrom($_sock,  2048);
					if(!$msg){
						stream_socket_shutdown($this->clients[$key],STREAM_SHUT_RDWR);
						unset($this->clients[$key]);
						$this->onClose($key);
					}else{
						$msg=$this->decode($msg);
						$this->onMessage($key,$msg);
					}
					
                    
                }
            }
        }
    }
 
	
	
	
	/**
     * 發送數據
     * @param $newClinet 新接入的socket
     * @param $msg   要發送的數據
     * @return int|string
     */
    public function send($newClinet, $msg){
		$msg=$this->encode($msg);
		if($msg){
			stream_socket_sendto($newClinet, $msg);
		}
        
    }
    
	public function encode($msg){
		  return $msg . "\n";
	}
	public function decode($msg){
		return rtrim($msg, "\r\n");
	}
 
    /**
     * 關閉socket
     */
    public function close(){
        return socket_close($this->_sockets);
    }
}
 
$sock = new SocketService('127.0.0.1','9000');
$sock->run();