pixelfed/app/Services/PublicTimelineService.php

<?php

namespace App\Services;

use Illuminate\Support\Facades\Redis;
use App\{
	Profile,
	Status,
	UserFilter
};

class PublicTimelineService {

	const CACHE_KEY = 'pf:services:timeline:public';

	public static function get($start = 0, $stop = 10)
	{
		if($stop > 100) {
			$stop = 100;
		}

		return Redis::zrevrange(self::CACHE_KEY, $start, $stop);
	}

	public static function getRankedMaxId($start = null, $limit = 10)
	{
		if(!$start) {
			return [];
		}

		return array_keys(Redis::zrevrangebyscore(self::CACHE_KEY, $start, '-inf', [
			'withscores' => true,
			'limit' => [1, $limit]
		]));
	}

	public static function getRankedMinId($end = null, $limit = 10)
	{
		if(!$end) {
			return [];
		}

		return array_keys(Redis::zrevrangebyscore(self::CACHE_KEY, '+inf', $end, [
			'withscores' => true,
			'limit' => [0, $limit]
		]));
	}

	public static function add($val)
	{
		if(self::count() > 400) {
			if(config('database.redis.client') === 'phpredis') {
				Redis::zpopmin(self::CACHE_KEY);
			}
		}

		return Redis::zadd(self::CACHE_KEY, $val, $val);
	}

	public static function rem($val)
	{
		return Redis::zrem(self::CACHE_KEY, $val);
	}

	public static function del($val)
	{
		return self::rem($val);
	}

	public static function count()
	{
		return Redis::zcard(self::CACHE_KEY);
	}

	public static function warmCache($force = false, $limit = 100)
	{
		if(self::count() == 0 || $force == true) {
			Redis::del(self::CACHE_KEY);
			$ids = Status::whereNull('uri')
				->whereNull('in_reply_to_id')
				->whereNull('reblog_of_id')
				->whereIn('type', ['photo', 'photo:album', 'video', 'video:album', 'photo:video:album'])
				->whereScope('public')
				->orderByDesc('id')
				->limit($limit)
				->pluck('id');
			foreach($ids as $id) {
				self::add($id);
			}
			return 1;
		}
		return 0;
	}
}