set([
'worker_num' => swoole_cpu_num()
]);
$pool = new DatabasePool();
/**
* The DB test
*
* @param string $database_type
* @param int $queries
*
* @return string
*/
$db = function (string $database_type, int $queries = 0) use ($pool): string {
$db = $pool->get($database_type);
// Read number of queries to run from URL parameter
$query_count = 1;
if ($queries > 1) {
$query_count = $queries > 500 ? 500 : $queries;
}
// Create an array with the response string.
$arr = [];
// Define query
$db->db_test = $db->db_test ?? $db->prepare('SELECT randomNumber FROM World WHERE id = ?');
// For each query, store the result set values in the response array
while ($query_count--) {
$id = mt_rand(1, 10000);
$ret = $db->db_test->execute([$id]);
// Store result in array.
$arr[] = ['id' => $id, 'randomNumber' => $ret[0]['randomNumber']];
}
// Use the PHP standard JSON encoder.
// http://www.php.net/manual/en/function.json-encode.php
if ($queries === -1) {
$arr = $arr[0];
}
$pool->put($db);
return json_encode($arr);
};
/**
* The Fortunes test
*
* @param string $database_type
*
* @return string
*/
$fortunes = function (string $database_type) use ($pool): string {
$db = $pool->get($database_type);
$fortune = [];
$db->fortune_test = $db->fortune_test ?? $db->prepare('SELECT id, message FROM Fortune');
$arr = $db->fortune_test->execute();
foreach ($arr as $row) {
$fortune[$row['id']] = $row['message'];
}
$fortune[0] = 'Additional fortune added at request time.';
asort($fortune);
$html = '';
foreach ($fortune as $id => $message) {
$message = htmlspecialchars($message, ENT_QUOTES, 'UTF-8');
$html .= "| {$id} | {$message} |
";
}
$pool->put($db);
return 'Fortunes';
};
/**
* The Updates test
*
* @param string $database_type
* @param int $queries
*
* @return string
*/
$updates = function (string $database_type, int $queries = 0) use ($pool): string {
$db = $pool->get($database_type);
$query_count = 1;
if ($queries > 1) {
$query_count = $queries > 500 ? 500 : $queries;
}
$arr = [];
$db->updates_test_select = $db->updates_test_select ?? $db->prepare('SELECT randomNumber FROM World WHERE id = ?');
$db->updates_test_update = $db->updates_test_update ?? $db->prepare('UPDATE World SET randomNumber = ? WHERE id = ?');
while ($query_count--) {
$id = mt_rand(1, 10000);
$randomNumber = mt_rand(1, 10000);
$ret = $db->updates_test_select->execute([$id]);
// Store result in array.
$world = ['id' => $id, 'randomNumber' => $ret[0]['randomNumber']];
$world['randomNumber'] = $randomNumber;
$db->updates_test_update->execute([$randomNumber, $id]);
$arr[] = $world;
}
$pool->put($db);
return json_encode($arr);
};
/**
* On start of the PHP worker. One worker per server process is started.
*/
//$server->on('workerStart', function () use ($pool) {
//});
/**
* On every request to the (web)server, execute the following code
*/
$server->on('request', function (Request $req, Response $res) use ($db, $fortunes, $updates) {
try {
switch ($req->server['request_uri']) {
case '/json':
$res->header('Content-Type', 'application/json');
$res->end(json_encode(['message' => 'Hello, World!']));
break;
case '/plaintext':
$res->header('Content-Type', 'text/plain; charset=utf-8');
$res->end('Hello, World!');
break;
case '/db':
$res->header('Content-Type', 'application/json');
if (isset($req->get['queries'])) {
$res->end($db('mysql', (int)$req->get['queries']));
} else {
$res->end($db('mysql', -1));
}
break;
case '/fortunes':
$res->header('Content-Type', 'text/html; charset=utf-8');
$res->end($fortunes('mysql'));
break;
case '/updates':
$res->header('Content-Type', 'application/json');
if (isset($req->get['queries'])) {
$res->end($updates('mysql', (int)$req->get['queries']));
} else {
$res->end($updates('mysql', -1));
}
break;
case '/db_postgres':
$res->header('Content-Type', 'application/json');
if (isset($req->get['queries'])) {
$res->end($db('postgres', (int)$req->get['queries']));
} else {
$res->end($db('postgres', -1));
}
break;
case '/fortunes_postgres':
$res->header('Content-Type', 'text/html; charset=utf-8');
$res->end($fortunes('postgres'));
break;
case '/updates_postgres':
$res->header('Content-Type', 'application/json');
if (isset($req->get['queries'])) {
$res->end($updates('postgres', (int)$req->get['queries']));
} else {
$res->end($updates('postgres', -1));
}
break;
default:
$res->status(404);
$res->end('Error 404');
}
} catch (\Throwable $e) {
$res->status(500);
$res->end('Error 500');
}
});
$server->start();
/**
* Class DatabasePool
*
* Deal with the fact that Swoole 2.1.3 has no build in database pooling
*/
class DatabasePool
{
private $server = [
'host' => '',
'user' => 'benchmarkdbuser',
'password' => 'benchmarkdbpass',
'database' => 'hello_world'
];
private $pool;
private $pool_count = 0;
function __construct()
{
$this->pool = new \SplQueue;
$this->server['host'] = gethostbyname('tfb-database');
}
function put($db)
{
$this->pool->enqueue($db);
$this->pool_count++;
}
function get(string $server_type)
{
if ($this->pool_count > 0) {
$this->pool_count--;
return $this->pool->dequeue();
}
// No idle connection, time to create a new connection
if ($server_type === 'mysql') {
$db = new Swoole\Coroutine\Mysql;
}
if ($server_type === 'postgres') {
$db = new Swoole\Coroutine\PostgreSql;
}
$db->connect($this->server);
return $db;
}
}