kiri-core/Queue/Abstracts/Queue.php

<?php


namespace Queue\Abstracts;


use Exception;
use Queue\Consumer;
use Snowflake\Abstracts\Component;
use Snowflake\Exception\ComponentException;
use Snowflake\Snowflake;

/**
 * Class Queue
 * @package Queue\Abstracts
 */
abstract class Queue extends Component implements Relyon
{


	/**
	 * @param string $key
	 * @param Consumer $consumer
	 * @param int $score
	 * @return false|int
	 * @throws ComponentException
	 * @throws Exception
	 */
	protected function push(string $key, Consumer $consumer, $score = 0)
	{
		$redis = Snowflake::app()->getRedis();
		try {
			$serialize = serialize($consumer);
			if (!$redis->lock($hash = md5($serialize))) {
				return false;
			}
			$isExists = $redis->zRevRank($key, $serialize);
			if ($isExists !== null) {
				$redis->zAdd($key, $score, $serialize);
			}
			return $redis->unlink($hash);
		} finally {
			$redis->release();
		}
	}


	/**
	 * @param $key
	 * @param Consumer $consumer
	 * @return false|int
	 * @throws ComponentException
	 * @throws Exception
	 */
	protected function pop($key, Consumer $consumer)
	{
		$redis = Snowflake::app()->getRedis();
		try {
			$serialize = serialize($consumer);
			if (!$redis->lock($hash = md5($serialize))) {
				return false;
			}
			$isExists = $redis->zRevRank($key, $serialize);
			if ($isExists === null) {
				return $redis->unlink($hash);
			}
			$redis->zRem($key, $serialize);
			return $redis->unlink($hash);
		} finally {
			$redis->release();
		}
	}


}