Psr7Middlewares\Utils\CacheMessageTrait

Complexity

Total Complexity

7

Size/Duplication

Total Lines	70
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	2

Importance

Changes	2
Bugs	0	Features	1

3 Methods

Rating	Name	Duplication	Size	Complexity
A	cache()	0	6	1
A	getFromCache()	0	20	4
A	saveIntoCache()	0	13	2

<?php

namespace Psr7Middlewares\Utils;

use Psr\Http\Message\MessageInterface;

/**
 * Trait used by all middlewares that needs to a psr-7 message into a psr-6 cache.
 */
trait CacheMessageTrait
{
    /**
     * @var CacheItemPoolInterface The cache implementation used
     */
    private $cache;

    /**
     * Provide the cache implementation.
     * 
     * @param CacheItemPoolInterface $cache
     * 
     * @return self
     */
    public function cache(CacheItemPoolInterface $cache)
    {
        $this->cache = $cache;

        return $this;
    }

    /**
     * Restore a message from the cache.
     * 
     * @param string           $key     The message key
     * @param MessageInterface $message
     *
     * @return MessageInterface|false
     */
    private function getFromCache($key, MessageInterface $message)
    {
        if ($this->cache) {
            $item = $this->cache->getItem($key);

            if ($item->isHit()) {
                list($headers, $body) = $item->get();

                foreach ($headers as $name => $header) {
                    $message = $message->withHeader($name, $header);
                }

                $message->getBody()->write($body);

                return $message;
            }
        }

        return false;
    }

    /**
     * Store a message in the cache.
     * 
     * @param string           $key     The message key
     * @param MessageInterface $message
     */
    private function saveIntoCache($key, MessageInterface $message)
    {
        if ($this->cache) {
            $item = $this->cache->getItem($key);

            $item->set([
                $message->getHeaders(),
                (string) $message->getBody(),
            ]);

            $this->cache->save($item);
        }
    }
}