class Inbox extends Memcached_DataObject
{
const BOXCAR = 128;
+ const MAX_NOTICES = 1024;
###START_AUTOCODE
/* the code below is auto generated do not remove the above tag */
/**
* Create a new inbox from existing Notice_inbox stuff
*/
-
static function initialize($user_id)
+ {
+ $inbox = Inbox::fromNoticeInbox($user_id);
+
+ unset($inbox->fake);
+
+ $result = $inbox->insert();
+
+ if (!$result) {
+ common_log_db_error($inbox, 'INSERT', __FILE__);
+ return null;
+ }
+
+ return $inbox;
+ }
+
+ static function fromNoticeInbox($user_id)
{
$ids = array();
$ni->selectAdd();
$ni->selectAdd('notice_id');
$ni->orderBy('notice_id DESC');
- $ni->limit(0, 1024);
+ $ni->limit(0, self::MAX_NOTICES);
if ($ni->find()) {
while($ni->fetch()) {
$inbox = new Inbox();
$inbox->user_id = $user_id;
- $inbox->notice_ids = pack('N*', $ids);
-
- $result = $inbox->insert();
-
- if (!$result) {
- common_log_db_error($inbox, 'INSERT', __FILE__);
- return null;
- }
+ $inbox->pack($ids);
+ $inbox->fake = true;
return $inbox;
}
+ /**
+ * Append the given notice to the given user's inbox.
+ * Caching updates are managed for the inbox itself.
+ *
+ * If the notice is already in this inbox, the second
+ * add will be silently dropped.
+ *
+ * @param int @user_id
+ * @param int $notice_id
+ * @return boolean success
+ */
static function insertNotice($user_id, $notice_id)
{
- $inbox = Inbox::staticGet('user_id', $user_id);
-
- if (empty($inbox)) {
+ // Going straight to the DB rather than trusting our caching
+ // during an update. Note: not using DB_DataObject::staticGet,
+ // which is unsafe to use directly (in-process caching causes
+ // memory leaks, which accumulate in queue processes).
+ $inbox = new Inbox();
+ if (!$inbox->get('user_id', $user_id)) {
$inbox = Inbox::initialize($user_id);
}
return false;
}
+ $ids = $inbox->unpack();
+ if (in_array(intval($notice_id), $ids)) {
+ // Already in there, we probably re-ran some inbox adds
+ // due to an error. Skip the dupe silently.
+ return true;
+ }
+
$result = $inbox->query(sprintf('UPDATE inbox '.
'set notice_ids = concat(cast(0x%08x as binary(4)), '.
- 'substr(notice_ids, 1, 4092)) '.
+ 'substr(notice_ids, 1, %d)) '.
'WHERE user_id = %d',
- $notice_id, $user_id));
+ $notice_id,
+ 4 * (self::MAX_NOTICES - 1),
+ $user_id));
if ($result) {
- $c = $this->memcache();
-
- if (!empty($c)) {
- $c->delete($this->cacheKey($this->tableName(), 'user_id', $user_id));
- }
+ self::blow('inbox:user_id:%d', $user_id);
}
return $result;
static function bulkInsert($notice_id, $user_ids)
{
- $cnt = count($user_ids);
-
- for ($off = 0; $off < $cnt; $off += self::BOXCAR) {
-
- $boxcar = array_slice($user_ids, $off, self::BOXCAR);
-
- if (empty($boxcar)) { // jump in, hobo!
- break;
- }
-
- $inbox = new Inbox();
-
- $inbox->query(sprintf('UPDATE inbox '.
- 'set notice_ids = concat(cast(0x%08x as binary(4)), '.
- 'substr(notice_ids, 1, 4092)) '.
- 'WHERE user_id in (%s)',
- $notice_id, implode(',', $boxcar)));
-
- $inbox->free();
+ foreach ($user_ids as $user_id)
+ {
+ Inbox::insertNotice($user_id, $notice_id);
}
}
- function stream($user_id, $offset, $limit, $since_id, $max_id, $since, $own=false)
+ /**
+ * Saves a list of integer notice_ids into a packed blob in this object.
+ * @param array $ids list of integer notice_ids
+ */
+ function pack(array $ids)
{
- $inbox = Inbox::staticGet('user_id', $user_id);
-
- if (empty($inbox)) {
- $inbox = Inbox::initialize($user_id);
- if (empty($inbox)) {
- return array();
- }
- }
-
- $ids = unpack('N*', $inbox->notice_ids);
-
- // XXX: handle since_id
- // XXX: handle max_id
-
- $ids = array_slice($ids, $offset, $limit);
+ $this->notice_ids = call_user_func_array('pack', array_merge(array('N*'), $ids));
+ }
- return $ids;
+ /**
+ * @return array of integer notice_ids
+ */
+ function unpack()
+ {
+ return unpack('N*', $this->notice_ids);
}
}