89 lines
2.5 KiB
PHP
89 lines
2.5 KiB
PHP
<?php
|
|
|
|
namespace BeyondCode\LaravelWebSockets\Statistics\Logger;
|
|
|
|
use BeyondCode\LaravelWebSockets\Statistics\Http\Controllers\WebSocketStatisticsEntriesController;
|
|
use BeyondCode\LaravelWebSockets\Statistics\Statistic;
|
|
use BeyondCode\LaravelWebSockets\WebSockets\Channels\ChannelManager;
|
|
use Clue\React\Buzz\Browser;
|
|
use function GuzzleHttp\Psr7\stream_for;
|
|
use Ratchet\ConnectionInterface;
|
|
|
|
class HttpStatisticsLogger implements StatisticsLogger
|
|
{
|
|
/** @var Statistic[] */
|
|
protected $statistics = [];
|
|
|
|
/** @var \BeyondCode\LaravelWebSockets\WebSockets\Channels\ChannelManager */
|
|
protected $channelManager;
|
|
|
|
/** @var Browser */
|
|
protected $browser;
|
|
|
|
public function __construct(ChannelManager $channelManager, Browser $browser)
|
|
{
|
|
$this->channelManager = $channelManager;
|
|
|
|
$this->browser = $browser;
|
|
}
|
|
|
|
public function webSocketMessage(ConnectionInterface $connection)
|
|
{
|
|
$this->initializeStatistics($connection->app->id);
|
|
|
|
$this->statistics[$connection->app->id]->webSocketMessage();
|
|
}
|
|
|
|
public function apiMessage($appId)
|
|
{
|
|
$this->initializeStatistics($appId);
|
|
|
|
$this->statistics[$appId]->apiMessage();
|
|
}
|
|
|
|
public function connection(ConnectionInterface $connection)
|
|
{
|
|
$this->initializeStatistics($connection->app->id);
|
|
|
|
$this->statistics[$connection->app->id]->connection();
|
|
}
|
|
|
|
public function disconnection(ConnectionInterface $connection)
|
|
{
|
|
$this->initializeStatistics($connection->app->id);
|
|
|
|
$this->statistics[$connection->app->id]->disconnection();
|
|
}
|
|
|
|
protected function initializeStatistics($id)
|
|
{
|
|
if (!isset($this->statistics[$id])) {
|
|
$this->statistics[$id] = new Statistic($id);
|
|
}
|
|
}
|
|
|
|
public function save()
|
|
{
|
|
foreach ($this->statistics as $appId => $statistic) {
|
|
|
|
if (!$statistic->isEnabled()) {
|
|
continue;
|
|
}
|
|
|
|
$this->browser
|
|
->post(
|
|
action([WebSocketStatisticsEntriesController::class, 'store']),
|
|
['Content-Type' => 'application/json'],
|
|
stream_for(json_encode($statistic->toArray()))
|
|
);
|
|
|
|
// Reset connection and message count
|
|
$currentConnectionCount = collect($this->channelManager->getChannels($appId))
|
|
->sum(function ($channel) {
|
|
return count($channel->getSubscribedConnections());
|
|
});
|
|
|
|
$statistic->reset($currentConnectionCount);
|
|
}
|
|
}
|
|
} |