1
0
mirror of https://github.com/danog/amp.git synced 2024-12-12 09:29:45 +01:00
amp/examples/005_tcp_server_obj.php
2014-09-23 00:06:03 -04:00

143 lines
4.8 KiB
PHP

<?php
require __DIR__ . '/../vendor/autoload.php';
date_default_timezone_set('UTC'); // F U Derick.
define('SERVER_ADDRESS', '127.0.0.1:1337');
/**
* A simple struct to hold a client's state while it's connected to our server
*/
class Client {
public $id;
public $socket;
public $readWatcher;
public $writeWatcher;
public $outputBuffer;
}
/**
* A simple TCP server that broadcasts the current time once per second to all connected clients
*/
class Server {
private $reactor;
private $clients = [];
private $timeBroadcastWatcher;
private $ioGranularity = 8192;
public function __construct(Amp\Reactor $reactor = null) {
$this->reactor = $reactor ?: Amp\reactor();
}
public function start($address) {
if (!$server = @stream_socket_server($address, $errno, $errstr)) {
throw new RuntimeException(
sprintf('Failed binding server on %s; [%d] %s', $address, $errno, $errstr)
);
}
printf("Server socket listening on %s\n", $address);
stream_set_blocking($server, false);
$this->reactor->onReadable($server, function() use ($server) {
$this->acceptClients($server);
});
// Let's schedule a broadcast of the current time to all connected sockets every three seconds
$this->timeBroadcastWatcher = $this->reactor->repeat(function() {
$this->broadcastTime();
}, $msInterval = 3000);
// Release the hounds!
$this->reactor->run();
}
private function acceptClients($server) {
while ($socket = @stream_socket_accept($server, $timeout=0, $name)) {
$client = new Client;
$client->id = (int) $socket;
$client->socket = $socket;
// What to do when the client socket is readable
$client->readWatcher = $this->reactor->onReadable($socket, function() use ($client) {
$this->readFromClient($client);
});
// What to do when the socket is writable
$client->writeWatcher = $this->reactor->onWritable($socket, function() use ($client) {
$this->writeToClient($client);
});
// Buffer something to send to the client. The writability watcher we just enabled
// above will take care of sending this data automatically.
$client->outputBuffer = "--- Welcome to the example server! ---\n\n";
printf("Client socket accepted: %s\n", $name);
// Store the client using its integer ID
$this->clients[$client->id] = $client;
}
}
private function broadcastTime() {
$data = date('r') . "\n";
foreach ($this->clients as $client) {
$client->outputBuffer .= $data;
$this->reactor->enable($client->writeWatcher);
}
}
private function readFromClient(Client $client) {
$data = @fread($client->socket, $this->ioGranularity);
// This only happens on EOF. If the socket has died we need to unload it.
if ($data == '' && $this->isSocketDead($client->socket)) {
$this->unloadClient($client);
} else {
printf("Data rcvd from client %d: %s\n", $client->id, $data);
}
}
private function isSocketDead($socket) {
return (!is_resource($socket) || feof($socket));
}
private function writeToClient(Client $client) {
$bytesWritten = @fwrite($client->socket, $client->outputBuffer);
if ($bytesWritten === strlen($client->outputBuffer)) {
// All data written. Disable the writability watcher. Sockets are essentially "always"
// writable, so it's important to disable write watchers when you don't have any data
// remaining to write. Otherwise you'll just hammer your CPU.
$client->outputBuffer = '';
$this->reactor->disable($client->writeWatcher);
} elseif ($bytesWritten > 0) {
// Data was partially written -- truncate the buffer
$client->outputBuffer = substr($client->outputBuffer, $bytesWritten);
} elseif ($this->isSocketDead($client->socket)) {
// Otherwise the client is dead and we just unload it
$this->unloadClient($client);
}
}
/**
* We have to clean up after ourselves or we'll create memory leaks. Always be sure to cancel
* any stream IO watchers or repeating timer events once they're no longer needed!
*/
private function unloadClient(Client $client) {
$this->reactor->cancel($client->readWatcher);
$this->reactor->cancel($client->writeWatcher);
if (is_resource($client->socket)) {
@fclose($client->socket);
}
unset($this->clients[$client->id]);
printf("Client %d disconnected\n", $client->id);
}
}
(new Server)->start(SERVER_ADDRESS);