EightPoints\Bundle\GuzzleBundle\Middleware\RequestTimeMiddleware

Complexity

Total Complexity

5

Size/Duplication

Total Lines	47
Duplicated Lines	0 %

Importance

Changes

0

3 Methods

Rating	Name	Duplication	Size	Complexity
A	getOnStatsCallback()	0	8	2
A	__construct()	0	3	1
A	__invoke()	0	10	2

<?php

namespace EightPoints\Bundle\GuzzleBundle\Middleware;

use EightPoints\Bundle\GuzzleBundle\DataCollector\HttpDataCollector;
use Psr\Http\Message\RequestInterface;
use GuzzleHttp\TransferStats;

class RequestTimeMiddleware
{
    /** @var \EightPoints\Bundle\GuzzleBundle\DataCollector\HttpDataCollector */
    private $dataCollector;

    /**
     * @param \EightPoints\Bundle\GuzzleBundle\DataCollector\HttpDataCollector $dataCollector
     */
    public function __construct(HttpDataCollector $dataCollector)
    {
        $this->dataCollector = $dataCollector;
    }

    /**
     * @param callable $handler
     *
     * @return \Closure
     */
    public function __invoke(callable $handler) : \Closure
    {
        return function (
            RequestInterface $request,
            array $options
        ) use ($handler) {
            $options['on_stats'] = $this->getOnStatsCallback(isset($options['on_stats']) ? $options['on_stats'] : null);

            // Continue the handler chain.
            return $handler($request, $options);
        };
    }

    /**
     * Create callback for on_stats options.
     * If request has on_stats option, it will be called inside of this callback.
     *
     * @param null|\Closure $initialOnStats
     *
     * @return \Closure
     */
    protected function getOnStatsCallback(\Closure $initialOnStats = null) : \Closure
    {
        return function (TransferStats $stats) use ($initialOnStats) {
            if (is_callable($initialOnStats)) {
                call_user_func($initialOnStats, $stats);
            }

            $this->dataCollector->addTotalTime((float)$stats->getTransferTime());
        };
    }
}