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 public $generator = null;
52 public $cachekey = null;
54 function __construct($generator, $args, $cachekey)
56 $this->generator = $generator;
58 $this->cachekey = $cachekey;
61 function getNotices($offset=0, $limit=20, $sinceId=0, $maxId=0)
63 $ids = $this->getNoticeIds($offset, $limit, $sinceId, $maxId);
65 $notices = self::getStreamByIds($ids);
70 function getNoticeIds($offset=0, $limit=20, $sinceId=0, $maxId=0)
72 $cache = Cache::instance();
74 // We cache NOTICE_CACHE_WINDOW elements at the tip of the stream.
75 // If the cache won't be hit, just generate directly.
78 $sinceId != 0 || $maxId != 0 ||
80 ($offset + $limit) > NOTICE_CACHE_WINDOW) {
81 return $this->generate($offset, $limit, $sinceId, $maxId);
84 // Check the cache to see if we have the stream.
86 $idkey = Cache::key($this->cachekey);
88 $idstr = $cache->get($idkey);
90 if ($idstr !== false) {
92 $window = explode(',', $idstr);
93 $ids = array_slice($window, $offset, $limit);
97 // Check the cache to see if we have a "last-known-good" version.
98 // The actual cache gets blown away when new notices are added, but
99 // the "last" value holds a lot of info. We might need to only generate
100 // a few at the "tip", which can bound our queries and save lots
103 $laststr = $cache->get($idkey.';last');
105 if ($laststr !== false) {
106 $window = explode(',', $laststr);
107 $last_id = $window[0];
108 $new_ids = $this->generate(0, NOTICE_CACHE_WINDOW, $last_id, 0);
110 $new_window = array_merge($new_ids, $window);
112 $new_windowstr = implode(',', $new_window);
114 $result = $cache->set($idkey, $new_windowstr);
115 $result = $cache->set($idkey . ';last', $new_windowstr);
117 $ids = array_slice($new_window, $offset, $limit);
122 // No cache hits :( Generate directly and stick the results
123 // into the cache. Note we generate the full cache window.
125 $window = $this->generate(0, NOTICE_CACHE_WINDOW, 0, 0);
127 $windowstr = implode(',', $window);
129 $result = $cache->set($idkey, $windowstr);
130 $result = $cache->set($idkey . ';last', $windowstr);
132 // Return just the slice that was requested
134 $ids = array_slice($window, $offset, $limit);
139 static function getStreamByIds($ids)
141 $cache = Cache::instance();
143 if (!empty($cache)) {
145 foreach ($ids as $id) {
146 $n = Notice::staticGet('id', $id);
151 return new ArrayWrapper($notices);
153 $notice = new Notice();
155 //if no IDs requested, just return the notice object
158 $notice->whereAdd('id in (' . implode(', ', $ids) . ')');
164 while ($notice->fetch()) {
165 $temp[$notice->id] = clone($notice);
170 foreach ($ids as $id) {
171 if (array_key_exists($id, $temp)) {
172 $wrapped[] = $temp[$id];
176 return new ArrayWrapper($wrapped);
180 function generate($offset, $limit, $sinceId, $maxId)
182 $args = array_merge($this->args, array($offset,
187 return call_user_func_array($this->generator, $args);