q792602257/qlg.tsgz.moe
0
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
Files
63aa29c5f2e92576f879336393d483eca284afe1
qlg.tsgz.moe/vendor/swoole/benchmark/run.php
root f0ef89dfbb Init Repo
2019-09-06 23:53:10 +08:00

575 lines
14 KiB
PHP
Executable File
Raw Blame History

<?php
require __DIR__.'/../examples/websocket/WebSocketClient.php';
//关闭错误输出
//error_reporting(0);
$shortopts = "c:";
$shortopts .= "n:";
$shortopts .= "s:";
$shortopts .= "f:";
$shortopts .= "p::";
$opt = getopt($shortopts);
//并发数量
if(!isset($opt['c'])) exit("require -c [process_num]. ep: -c 100\n");
if(!isset($opt['n'])) exit("require -n [request_num]. ep: -n 10000\n");
if(!isset($opt['s'])) exit("require -s [server_url]. ep: -s tcp://127.0.0.1:9999\n");
if(!isset($opt['f'])) exit("require -f [test_function]. ep: -f short_tcp\n");
$bc = new Swoole_Benchmark(trim($opt['f']));
$bc->process_num = (int)$opt['c'];
$bc->request_num = (int)$opt['n'];
$bc->server_url = trim($opt['s']);
$bc->server_config = parse_url($bc->server_url);
$bc->send_data = "GET / HTTP/1.1\r\n";
$bc->send_data .= "Host: www.baidu.com\r\n";
$bc->send_data .= "Connection: keep-alive\r\n";
$bc->send_data .= "Accept: text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,*/*;q=0.8\r\n";
$bc->send_data .= "User-Agent: Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/34.0.1847.116 Safari/537.36\r\n\r\n";
$bc->read_len = 65536;
if(!empty($opt['p'])) $bc->show_detail = true;
function eof(Swoole_Benchmark $bc)
{
static $client = null;
static $i;
$start = microtime(true);
if (empty($client)) {
$client = new swoole_client(SWOOLE_SOCK_TCP, SWOOLE_SOCK_SYNC);
$client->set(array('open_eof_check' => true, "package_eof" => "\r\n\r\n"));
$end = microtime(true);
$conn_use = $end - $start;
$bc->max_conn_time = $conn_use;
$i = 0;
//echo "connect {$bc->server_url} \n";
if (!$client->connect($bc->server_config['host'], $bc->server_config['port'], 2)) {
error:
echo "Error: " . swoole_strerror($client->errCode) . "[{$client->errCode}]\n";
$client = null;
return false;
}
$start = $end;
}
/*--------写入Sokcet-------*/
$data = str_repeat('A', rand(100, 200))."\r\n\r\n";
if (!$client->send($data))
{
goto error;
}
$end = microtime(true);
$write_use = $end - $start;
if ($write_use > $bc->max_write_time) $bc->max_write_time = $write_use;
$start = $end;
/*--------读取Sokcet-------*/
$i ++;
$ret = $client->recv();
if (empty($ret))
{
echo $bc->pid, "#$i", " is lost\n";
return false;
}
elseif(strlen($ret) != strlen($data))
{
echo "#$i\tlength error\n";
var_dump($ret);
echo "-----------------------------------\n";
var_dump($data);
}
$end = microtime(true);
$read_use = $end - $start;
if ($read_use > $bc->max_read_time) $bc->max_read_time = $read_use;
return true;
}
function long_tcp(Swoole_Benchmark $bc)
{
static $fp = null;
static $i;
$start = microtime(true);
if(empty($fp))
{
$fp = new swoole_client(SWOOLE_SOCK_TCP, SWOOLE_SOCK_SYNC);
$end = microtime(true);
$conn_use = $end-$start;
$bc->max_conn_time = $conn_use;
$i = 0;
//echo "connect {$bc->server_url} \n";
if (!$fp->connect($bc->server_config['host'], $bc->server_config['port'], 2))
{
error:
echo "Error: ".swoole_strerror($fp->errCode)."[{$fp->errCode}]\n";
$fp = null;
return false;
}
$start = $end;
}
/*--------写入Sokcet-------*/
if (!$fp->send($bc->send_data))
{
goto error;
}
$end = microtime(true);
$write_use = $end - $start;
if ($write_use > $bc->max_write_time)
{
$bc->max_write_time = $write_use;
}
$start = $end;
/*--------读取Sokcet-------*/
while(true)
{
$ret = $fp->recv(65530);
if (empty($ret) or substr($ret, -1, 1) == "\n")
{
break;
}
}
//var_dump($ret);
$i++;
if (empty($ret))
{
echo $bc->pid, "#$i@", " is lost\n";
return false;
}
$end = microtime(true);
$read_use = $end - $start;
if ($read_use > $bc->max_read_time)
{
$bc->max_read_time = $read_use;
}
return true;
}
function websocket(Swoole_Benchmark $bc)
{
static $client = null;
static $i;
$start = microtime(true);
if (empty($client))
{
$client = new WebSocketClient($bc->server_config['host'], $bc->server_config['port']);
if (!$client->connect())
{
echo "connect failed\n";
return false;
}
$end = microtime(true);
$conn_use = $end - $start;
$bc->max_conn_time = $conn_use;
$i = 0;
$start = $end;
}
/*--------写入Sokcet-------*/
if (!$client->send($bc->send_data))
{
echo "send failed\n";
return false;
}
$end = microtime(true);
$write_use = $end - $start;
if ($write_use > $bc->max_write_time)
{
$bc->max_write_time = $write_use;
}
$start = $end;
/*--------读取Sokcet-------*/
$ret = $client->recv();
//var_dump($ret);
$i++;
if (empty($ret))
{
echo $bc->pid, "#$i@", " is lost\n";
return false;
}
$end = microtime(true);
$read_use = $end - $start;
if ($read_use > $bc->max_read_time)
{
$bc->max_read_time = $read_use;
}
return true;
}
/**
* 去掉计时信息的UDP
* @param $bc
* @return bool
*/
function udp(Swoole_Benchmark $bc)
{
static $fp;
if (empty($fp))
{
$fp = stream_socket_client($bc->server_url, $errno, $errstr, 1);
if (!$fp)
{
echo "{$errstr}[{$errno}]\n";
return false;
}
}
/*--------写入Sokcet-------*/
fwrite($fp, $bc->send_data);
/*--------读取Sokcet-------*/
$ret = fread($fp, $bc->read_len);
if (empty($ret))
{
return false;
}
return true;
}
function udp2(Swoole_Benchmark $bc)
{
static $fp;
$start = microtime(true);
if (empty($fp))
{
$u = parse_url($bc->server_url);
$fp = new swoole_client(SWOOLE_SOCK_UDP);
$fp->connect($u['host'], $u['port'], 0.5, 0);
$end = microtime(true);
$conn_use = $end - $start;
$bc->max_conn_time = $conn_use;
$start = $end;
}
/*--------写入Sokcet-------*/
$fp->send($bc->send_data);
$end = microtime(true);
$write_use = $end - $start;
if ($write_use > $bc->max_write_time)
{
$bc->max_write_time = $write_use;
}
$start = $end;
/*--------读取Sokcet-------*/
$ret = $fp->recv();
if (empty($ret))
{
return false;
}
$end = microtime(true);
$read_use = $end - $start;
if ($read_use > $bc->max_read_time)
{
$bc->max_read_time = $read_use;
}
return true;
}
function short_tcp($bc)
{
$fp = new swoole_client(SWOOLE_SOCK_TCP, SWOOLE_SOCK_SYNC);
if(!$fp->connect($bc->server_config['host'], $bc->server_config['port'], 1))
{
error:
echo "Error: ".socket_strerror($fp->errCode)."[{$fp->errCode}]\n";
return false;
}
else
{
if(!$fp->send($bc->send_data))
{
goto error;
}
$ret = $fp->recv();
$fp->close();
if(!empty($ret)) return true;
else return false;
}
}
function long_socks5($bc)
{
static $fp = null;
static $i;
$start = microtime(true);
if(empty($fp))
{
$fp = new swoole_client(SWOOLE_SOCK_TCP, SWOOLE_SOCK_SYNC,5);
$end = microtime(true);
$conn_use = $end-$start;
$bc->max_conn_time = $conn_use;
$i = 0;
//echo "connect {$bc->server_url} \n";
if (!$fp->connect($bc->server_config['host'], $bc->server_config['port'], 2))
{
error:
echo "Error: ".swoole_strerror($fp->errCode)."[{$fp->errCode}]\n";
$fp = null;
return false;
}
$fp->send(pack("C3", 0x05, 0x01, 0x00));//greet
$data = $fp->recv();
$response = unpack("Cversion/Cmethod", $data);
if ($response['version'] != 0x05)
{
exit('SOCKS version is not supported.');
}
$headers = getHeader($bc->send_data);
if (empty($headers['port'])) {
$headers['port'] = 80;
}
$g = pack("C5", 0x05, 0x01, 0x00, 0x03, strlen($headers['host'])) . $headers['host'] . pack("n", $headers['port']);
$fp->send($g);
$data = $fp->recv();
$response = unpack("Cversion/Cresult/Creg/Ctype/Lip/Sport", $data);
if ($response['result'] != 0x00)
{
echo 'SOCKS connection request failed: ' . getSocksRefusalMsg($response['result']), $response['result'];exit;
}
$start = $end;
}
/*--------写入Sokcet-------*/
if (!$fp->send($bc->send_data))
{
goto error;
}
$end = microtime(true);
$write_use = $end - $start;
if ($write_use > $bc->max_write_time)
{
$bc->max_write_time = $write_use;
}
$start = $end;
/*--------读取Sokcet-------*/
while(true)
{
$ret = $fp->recv(65530);
if (empty($ret) or substr($ret, -1, 1) == "\n")
{
break;
}
}
//var_dump($ret);
$i++;
if (empty($ret))
{
echo $bc->pid, "#$i@", " is lost\n";
return false;
}
$end = microtime(true);
$read_use = $end - $start;
if ($read_use > $bc->max_read_time)
{
$bc->max_read_time = $read_use;
}
return true;
}
function getHeader($message)
{
// 标准每行应该以"\r\n"行终止，这里兼容以"\n"作为行终止的情况，所以按"\n"分割行
$lines = explode("\n", $message);
foreach ($lines as &$line)
{
// 按"\n"分割行以后，某些行末可能存在"\r"字符，这里将其过滤掉
$line = rtrim($line, "\r");
}
unset($line);
if (count($lines) <= 0)
{
return false;
}
$headers = [];
foreach ($lines as $line)
{
$pos = strpos($line, ':');
// 非标准首部，抛弃
if ($pos === false)
{
continue;
}
$field = trim(substr($line, 0, $pos));
$value = trim(substr($line, $pos + 1));
// 如果有host头部，重新设置host和port
if (strtolower($field) === 'host')
{
$segments = explode(':', $value);
$host = $segments[0];
$headers['host'] = $host;
if (isset($segments[1]))
{
$port = intval($segments[1]);
$headers['port'] = $port;
}
}
$headers[$field] = $value;
}
return $headers;
}
//请求数量最好是进程数的倍数
$bc->process_req_num = intval($bc->request_num/$bc->process_num);
$bc->run();
$bc->report();
$bc->end();
class Swoole_Benchmark
{
public $test_func;
public $process_num;
public $request_num;
public $server_url;
public $server_config;
public $send_data;
public $read_len;
public $time_end;
private $shm_key;
public $main_pid;
public $child_pid = array();
public $show_detail = false;
public $max_write_time = 0;
public $max_read_time = 0;
public $max_conn_time = 0;
public $pid;
protected $tmp_dir = '/tmp/swoole_bench/';
function __construct($func)
{
if (!function_exists($func))
{
exit(__CLASS__ . ": function[$func] not exists\n");
}
if (!is_dir($this->tmp_dir))
{
mkdir($this->tmp_dir);
}
$this->test_func = $func;
}
function end()
{
unlink($this->shm_key);
foreach($this->child_pid as $pid)
{
$f = $this->tmp_dir . 'lost_' . $pid . '.log';
if (is_file($f)) unlink($f);
}
}
function run()
{
$this->main_pid = posix_getpid();
$this->shm_key = $this->tmp_dir.'t.log';
for ($i = 0; $i < $this->process_num; $i++)
{
$this->child_pid[] = $this->start(array($this, 'worker'));
}
for ($i = 0; $i < $this->process_num; $i++)
{
$status = 0;
$pid = pcntl_wait($status);
}
$this->time_end = microtime(true);
}
function init_signal()
{
pcntl_signal(SIGUSR1,array($this, "sig_handle"));
}
function sig_handle($sig)
{
switch ($sig)
{
case SIGUSR1:
return;
}
$this->init_signal();
}
function start($func)
{
$pid = pcntl_fork();
if($pid>0)
{
return $pid;
}
elseif($pid==0)
{
$this->worker();
}
else
{
echo "Error:fork fail\n";
}
}
function worker()
{
$lost = 0;
if(!file_exists($this->shm_key))
{
file_put_contents($this->shm_key,microtime(true));
}
if($this->show_detail) $start = microtime(true);
$this->pid = posix_getpid();
for ($i = 0; $i < $this->process_req_num; $i++)
{
$func = $this->test_func;
if(!$func($this)) $lost++;
}
if ($this->show_detail)
{
$log = $this->pid . "#\ttotal_use(s):" . substr(microtime(true) - $start, 0, 5);
$log .= "\tconnect(ms):" . substr($this->max_conn_time * 1000, 0, 5);
$log .= "\twrite(ms):" . substr($this->max_write_time * 1000, 0, 5);
$log .= "\tread(ms):" . substr($this->max_read_time * 1000, 0, 5);
file_put_contents($this->tmp_dir.'lost_' . $this->pid . '.log', $lost . "\n" . $log);
}
else
{
file_put_contents($this->tmp_dir.'lost_'.$this->pid.'.log', $lost);
}
exit(0);
}
function report()
{
$time_start = file_get_contents($this->shm_key);
$usetime = $this->time_end - $time_start;
$lost = 0;
foreach ($this->child_pid as $f)
{
$file = $this->tmp_dir.'lost_'.$f.'.log';
if (is_file($file))
{
$_lost = file_get_contents($file);
$log = explode("\n",$_lost,2);
}
if (!empty($log))
{
$lost += intval($log[0]);
if ($this->show_detail) echo $log[1], "\n";
}
}
//并发量
echo "concurrency:\t".$this->process_num,"\n";
//请求量
echo "request num:\t".$this->request_num,"\n";
//请求量
echo "lost num:\t".$lost,"\n";
//请求量
echo "success num:\t".($this->request_num-$lost),"\n";
//总时间
echo "total time:\t".substr($usetime,0,5),"\n";
//每秒处理能力
echo "req per second:\t".intval($this->request_num/$usetime),"\n";
//每次请求平均时间ms
echo "one req use(ms):\t".substr($usetime/$this->request_num*1000,0,5),"\n";
}
}
Reference in New Issue View Git Blame Copy Permalink