3 * StatusNet - the distributed open-source microblogging tool
4 * Copyright (C) 2011, StatusNet, Inc.
10 * This program is free software: you can redistribute it and/or modify
11 * it under the terms of the GNU Affero General Public License as published by
12 * the Free Software Foundation, either version 3 of the License, or
13 * (at your option) any later version.
15 * This program is distributed in the hope that it will be useful,
16 * but WITHOUT ANY WARRANTY; without even the implied warranty of
17 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
18 * GNU Affero General Public License for more details.
20 * You should have received a copy of the GNU Affero General Public License
21 * along with this program. If not, see <http://www.gnu.org/licenses/>.
25 * @author Evan Prodromou <evan@status.net>
26 * @copyright 2011 StatusNet, Inc.
27 * @license http://www.fsf.org/licensing/licenses/agpl-3.0.html AGPL 3.0
28 * @link http://status.net/
31 if (!defined('STATUSNET')) {
32 // This check helps protect against security problems;
33 // your code file can't be executed directly from the web.
38 * Class for notice streams
42 * @author Evan Prodromou <evan@status.net>
43 * @copyright 2011 StatusNet, Inc.
44 * @license http://www.fsf.org/licensing/licenses/agpl-3.0.html AGPL 3.0
45 * @link http://status.net/
50 const CACHE_WINDOW = 200;
52 public $generator = null;
54 public $cachekey = null;
56 function __construct($generator, $args, $cachekey)
58 $this->generator = $generator;
60 $this->cachekey = $cachekey;
63 function getNotices($offset=0, $limit=20, $sinceId=0, $maxId=0)
65 $ids = $this->getNoticeIds($offset, $limit, $sinceId, $maxId);
67 $notices = self::getStreamByIds($ids);
72 function getNoticeIds($offset=0, $limit=20, $sinceId=0, $maxId=0)
74 $cache = Cache::instance();
76 // We cache self::CACHE_WINDOW elements at the tip of the stream.
77 // If the cache won't be hit, just generate directly.
80 $sinceId != 0 || $maxId != 0 ||
82 ($offset + $limit) > self::CACHE_WINDOW) {
83 return $this->generate($offset, $limit, $sinceId, $maxId);
86 // Check the cache to see if we have the stream.
88 $idkey = Cache::key($this->cachekey);
90 $idstr = $cache->get($idkey);
92 if ($idstr !== false) {
94 $window = explode(',', $idstr);
95 $ids = array_slice($window, $offset, $limit);
99 // Check the cache to see if we have a "last-known-good" version.
100 // The actual cache gets blown away when new notices are added, but
101 // the "last" value holds a lot of info. We might need to only generate
102 // a few at the "tip", which can bound our queries and save lots
105 $laststr = $cache->get($idkey.';last');
107 if ($laststr !== false) {
108 $window = explode(',', $laststr);
109 $last_id = $window[0];
110 $new_ids = $this->generate(0, self::CACHE_WINDOW, $last_id, 0);
112 $new_window = array_merge($new_ids, $window);
114 $new_windowstr = implode(',', $new_window);
116 $result = $cache->set($idkey, $new_windowstr);
117 $result = $cache->set($idkey . ';last', $new_windowstr);
119 $ids = array_slice($new_window, $offset, $limit);
124 // No cache hits :( Generate directly and stick the results
125 // into the cache. Note we generate the full cache window.
127 $window = $this->generate(0, self::CACHE_WINDOW, 0, 0);
129 $windowstr = implode(',', $window);
131 $result = $cache->set($idkey, $windowstr);
132 $result = $cache->set($idkey . ';last', $windowstr);
134 // Return just the slice that was requested
136 $ids = array_slice($window, $offset, $limit);
141 static function getStreamByIds($ids)
143 $cache = Cache::instance();
145 if (!empty($cache)) {
147 foreach ($ids as $id) {
148 $n = Notice::staticGet('id', $id);
153 return new ArrayWrapper($notices);
155 $notice = new Notice();
157 //if no IDs requested, just return the notice object
160 $notice->whereAdd('id in (' . implode(', ', $ids) . ')');
166 while ($notice->fetch()) {
167 $temp[$notice->id] = clone($notice);
172 foreach ($ids as $id) {
173 if (array_key_exists($id, $temp)) {
174 $wrapped[] = $temp[$id];
178 return new ArrayWrapper($wrapped);
182 function generate($offset, $limit, $sinceId, $maxId)
184 $args = array_merge($this->args, array($offset,
189 return call_user_func_array($this->generator, $args);