139 lines
4.0 KiB
PHP
Executable File
139 lines
4.0 KiB
PHP
Executable File
<?php
|
||
//$server = new swoole_websocket_server("0.0.0.0", 9501);
|
||
$server = new swoole_websocket_server("0.0.0.0", 9501, SWOOLE_BASE);
|
||
//$server->addlistener('0.0.0.0', 9502, SWOOLE_SOCK_UDP);
|
||
//$server->set(['worker_num' => 4,
|
||
// 'task_worker_num' => 4,
|
||
//]);
|
||
|
||
function user_handshake(swoole_http_request $request, swoole_http_response $response)
|
||
{
|
||
//自定定握手规则，没有设置则用系统内置的（只支持version:13的）
|
||
if (!isset($request->header['sec-websocket-key']))
|
||
{
|
||
//'Bad protocol implementation: it is not RFC6455.'
|
||
$response->end();
|
||
return false;
|
||
}
|
||
if (0 === preg_match('#^[+/0-9A-Za-z]{21}[AQgw]==$#', $request->header['sec-websocket-key'])
|
||
|| 16 !== strlen(base64_decode($request->header['sec-websocket-key']))
|
||
)
|
||
{
|
||
//Header Sec-WebSocket-Key is illegal;
|
||
$response->end();
|
||
return false;
|
||
}
|
||
|
||
$key = base64_encode(sha1($request->header['sec-websocket-key']
|
||
. '258EAFA5-E914-47DA-95CA-C5AB0DC85B11',
|
||
true));
|
||
$headers = array(
|
||
'Upgrade' => 'websocket',
|
||
'Connection' => 'Upgrade',
|
||
'Sec-WebSocket-Accept' => $key,
|
||
'Sec-WebSocket-Version' => '13',
|
||
'KeepAlive' => 'off',
|
||
);
|
||
foreach ($headers as $key => $val)
|
||
{
|
||
$response->header($key, $val);
|
||
}
|
||
$response->status(101);
|
||
$response->end();
|
||
global $server;
|
||
$fd = $request->fd;
|
||
$server->defer(function () use ($fd, $server)
|
||
{
|
||
$server->push($fd, "hello, welcome\n");
|
||
});
|
||
return true;
|
||
}
|
||
|
||
$server->on('handshake', 'user_handshake');
|
||
$server->on('open', function (swoole_websocket_server $_server, swoole_http_request $request) {
|
||
echo "server#{$_server->worker_pid}: handshake success with fd#{$request->fd}\n";
|
||
var_dump($_server->exist($request->fd), $_server->getClientInfo($request->fd));
|
||
// var_dump($request);
|
||
});
|
||
|
||
$server->on('message', function (swoole_websocket_server $_server, $frame) {
|
||
var_dump($frame->data);
|
||
echo "received ".strlen($frame->data)." bytes\n";
|
||
if ($frame->data == "close")
|
||
{
|
||
$_server->close($frame->fd);
|
||
}
|
||
elseif($frame->data == "task")
|
||
{
|
||
$_server->task(['go' => 'die']);
|
||
}
|
||
else
|
||
{
|
||
//echo "receive from {$frame->fd}:{$frame->data}, opcode:{$frame->opcode}, finish:{$frame->finish}\n";
|
||
// for ($i = 0; $i < 100; $i++)
|
||
{
|
||
$_send = str_repeat('B', rand(100, 800));
|
||
$_server->push($frame->fd, $_send);
|
||
// echo "#$i\tserver sent " . strlen($_send) . " byte \n";
|
||
}
|
||
$fd = $frame->fd;
|
||
$_server->tick(2000, function($id) use ($fd, $_server) {
|
||
$_send = str_repeat('B', rand(100, 5000));
|
||
$ret = $_server->push($fd, $_send);
|
||
if (!$ret)
|
||
{
|
||
var_dump($id);
|
||
var_dump($_server->clearTimer($id));
|
||
}
|
||
});
|
||
}
|
||
});
|
||
|
||
$server->on('close', function ($_server, $fd) {
|
||
echo "client {$fd} closed\n";
|
||
});
|
||
|
||
$server->on('task', function ($_server, $worker_id, $task_id, $data)
|
||
{
|
||
var_dump($worker_id, $task_id, $data);
|
||
return "hello world\n";
|
||
});
|
||
|
||
$server->on('finish', function ($_server, $task_id, $result)
|
||
{
|
||
var_dump($task_id, $result);
|
||
});
|
||
|
||
$server->on('packet', function ($_server, $data, $client) {
|
||
echo "#".posix_getpid()."\tPacket {$data}\n";
|
||
var_dump($client);
|
||
});
|
||
|
||
$server->on('request', function (swoole_http_request $request, swoole_http_response $response) {
|
||
$response->end(<<<HTML
|
||
<h1>Swoole WebSocket Server</h1>
|
||
<script>
|
||
var wsServer = 'ws://127.0.0.1:9501';
|
||
var websocket = new WebSocket(wsServer);
|
||
websocket.onopen = function (evt) {
|
||
console.log("Connected to WebSocket server.");
|
||
};
|
||
|
||
websocket.onclose = function (evt) {
|
||
console.log("Disconnected");
|
||
};
|
||
|
||
websocket.onmessage = function (evt) {
|
||
console.log('Retrieved data from server: ' + evt.data);
|
||
};
|
||
|
||
websocket.onerror = function (evt, e) {
|
||
console.log('Error occured: ' + evt.data);
|
||
};
|
||
</script>
|
||
HTML
|
||
);
|
||
});
|
||
|
||
$server->start();
|