SwooleTW\Http\Websocket\Rooms\RedisRoom

Complexity

Total Complexity

24

Size/Duplication

Total Lines	211
Duplicated Lines	0 %

Importance

Changes	4
Bugs	1	Features	1

14 Methods

Rating	Name	Duplication	Size	Complexity
A	connection()	0	3	1
A	addValue()	0	12	2
A	prepare()	0	5	1
A	checkTable()	0	4	2
A	getKey()	0	3	1
A	removeValue()	0	12	2
A	cleanRooms()	0	4	2
A	add()	0	8	3
A	getRooms()	0	3	1
A	getRedis()	0	3	1
A	delete()	0	9	4
A	getClients()	0	3	1
A	__construct()	0	4	1
A	getValue()	0	8	2

<?php

namespace SwooleTW\Http\Websocket\Rooms;

use Illuminate\Support\Arr;
use Illuminate\Contracts\Redis\Factory as RedisFactory;

/**
 * Class RedisRoom
 */
class RedisRoom implements RoomContract
{
	/**
     * The Redis factory implementation.
     *
     * @var \Illuminate\Contracts\Redis\Factory
     */
    public $redis;

	/**
	 * @var array
	 */
	protected $config;

	/**
	 * RedisRoom constructor.
	 *
	 * @param array $config
	 */
	public function __construct(array $config, RedisFactory $redis)
	{
		$this->config = $config;
		$this->redis = $redis;
	}

	/**
	 * @param \Illuminate\Contracts\Redis\Factory|null $redis
	 *
	 * @return \Ofcold\HttpSwoole\Websocket\Rooms\RoomContract

The type Ofcold\HttpSwoole\Websocket\Rooms\RoomContract was not found. Maybe you did not declare it correctly or list all dependencies?

	 */
	public function prepare(RedisFactory $redis = null): RoomContract
	{
		$this->cleanRooms();

		return $this;

The expression return $this returns the type SwooleTW\Http\Websocket\Rooms\RedisRoom which is incompatible with the documented return type Ofcold\HttpSwoole\Websocket\Rooms\RoomContract.

	}

	/**
	 * Get redis client.
	 */
	public function getRedis()
	{
		return $this->connection();
	}

	/**
	 * Add multiple socket fds to a room.
	 *
	 * @param int fd

The type SwooleTW\Http\Websocket\Rooms\fd was not found. Maybe you did not declare it correctly or list all dependencies?

	 * @param array|string rooms

The type SwooleTW\Http\Websocket\Rooms\rooms was not found. Maybe you did not declare it correctly or list all dependencies?

	 */
	public function add(int $fd, $rooms)
	{
		$rooms = is_array($rooms) ? $rooms : [$rooms];

		$this->addValue($fd, $rooms, RoomContract::DESCRIPTORS_KEY);

		foreach ($rooms as $room) {
			$this->addValue($room, [$fd], RoomContract::ROOMS_KEY);
		}
	}

	/**
	 * Delete multiple socket fds from a room.
	 *
	 * @param int fd
	 * @param array|string rooms
	 */
	public function delete(int $fd, $rooms)
	{
		$rooms = is_array($rooms) ? $rooms : [$rooms];
		$rooms = count($rooms) ? $rooms : $this->getRooms($fd);

		$this->removeValue($fd, $rooms, RoomContract::DESCRIPTORS_KEY);

		foreach ($rooms as $room) {
			$this->removeValue($room, [$fd], RoomContract::ROOMS_KEY);
		}
	}

	/**
	 * Add value to redis.
	 *
	 * @param $key
	 * @param array $values
	 * @param string $table
	 *
	 * @return $this
	 */
	public function addValue($key, array $values, string $table)
	{
		$this->checkTable($table);
		$redisKey = $this->getKey($key, $table);

		$this->connection()->pipeline(function ($pipe) use ($redisKey, $values) {
			foreach ($values as $value) {
				$pipe->sadd($redisKey, $value);
			}
		});

		return $this;
	}

	/**
	 * Remove value from reddis.
	 *
	 * @param $key
	 * @param array $values
	 * @param string $table
	 *
	 * @return $this
	 */
	public function removeValue($key, array $values, string $table)
	{
		$this->checkTable($table);
		$redisKey = $this->getKey($key, $table);

		$this->connection()->pipeline(function ($pipe) use ($redisKey, $values) {
			foreach ($values as $value) {
				$pipe->srem($redisKey, $value);
			}
		});

		return $this;
	}

	/**
	 * Get all sockets by a room key.
	 *
	 * @param string room
	 *
	 * @return array
	 */
	public function getClients(string $room)
	{
		return $this->getValue($room, RoomContract::ROOMS_KEY) ?? [];
	}

	/**
	 * Get all rooms by a fd.
	 *
	 * @param int fd
	 *
	 * @return array
	 */
	public function getRooms(int $fd)
	{
		return $this->getValue($fd, RoomContract::DESCRIPTORS_KEY) ?? [];
	}

	/**
	 * Check table for rooms and descriptors.
	 *
	 * @param string $table
	 */
	protected function checkTable(string $table)
	{
		if (! in_array($table, [RoomContract::ROOMS_KEY, RoomContract::DESCRIPTORS_KEY])) {
			throw new \InvalidArgumentException("Invalid table name: `{$table}`.");
		}
	}

	/**
	 * Get value.
	 *
	 * @param string $key
	 * @param string $table
	 *
	 * @return array
	 */
	public function getValue(string $key, string $table)
	{
		$this->checkTable($table);

		$result = $this->connection()->smembers($this->getKey($key, $table));

		// Try to fix occasional non-array returned result
		return is_array($result) ? $result : [];
	}

	/**
	 * Get key.
	 *
	 * @param string $key
	 * @param string $table
	 *
	 * @return string
	 */
	public function getKey(string $key, string $table)
	{
		return "{$table}:{$key}";
	}

	/**
	 * Clean all rooms.
	 */
	protected function cleanRooms(): void
	{
		if (count($keys = $this->connection()->keys("*"))) {
			$this->connection()->del($keys);
		}
	}

	/**
	 * Get the Redis connection instance.
	 *
	 * @return \Illuminate\Redis\Connections\Connection

The type Illuminate\Redis\Connections\Connection was not found. Maybe you did not declare it correctly or list all dependencies?

	 */
	public function connection()
	{
		return $this->redis->connection('swoole');
	}
}