We are no longer offering accounts on this server. Consider https://gitlab.freedesktop.org/ as a place to host projects.

filteringnoticestream.php 3.67 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58
<?php
/**
 * StatusNet - the distributed open-source microblogging tool
 * Copyright (C) 2011, StatusNet, Inc.
 *
 * A notice stream that filters its upstream content
 * 
 * PHP version 5
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 *
 * @category  Stream
 * @package   StatusNet
 * @author    Evan Prodromou <evan@status.net>
 * @copyright 2011 StatusNet, Inc.
 * @license   http://www.fsf.org/licensing/licenses/agpl-3.0.html AGPL 3.0
 * @link      http://status.net/
 */

if (!defined('STATUSNET')) {
    // This check helps protect against security problems;
    // your code file can't be executed directly from the web.
    exit(1);
}

/**
 * A class for presenting a filtered notice stream based on an upstream stream
 *
 * @category  Stream
 * @package   StatusNet
 * @author    Evan Prodromou <evan@status.net>
 * @copyright 2011 StatusNet, Inc.
 * @license   http://www.fsf.org/licensing/licenses/agpl-3.0.html AGPL 3.0
 * @link      http://status.net/
 */

abstract class FilteringNoticeStream extends NoticeStream
{
    protected $upstream;

    function __construct($upstream)
    {
        $this->upstream = $upstream;
    }

    abstract function filter($notice);

59
    function getNotices($offset, $limit, $sinceId=null, $maxId=null)
60 61 62 63 64 65 66 67 68 69 70 71 72
    {
        // "offset" is virtual; we have to get a lot
        $total = $offset + $limit;

        $filtered = array();

        $startAt = 0;
        $askFor  = $total;

        // Keep going till we have $total notices in $notices array,
        // or we get nothing from upstream.

        $results = null;
73
        $round = 0;
74 75

        do {
76

77 78 79 80 81 82 83 84
            $raw = $this->upstream->getNotices($startAt, $askFor, $sinceId, $maxId);

            $results = $raw->N;

            if ($results == 0) {
                break;
            }

Evan Prodromou's avatar
Evan Prodromou committed
85 86 87 88 89
            $notices = $raw->fetchAll();
            
            $this->prefill($notices);

            foreach ($notices as $notice) {
Evan Prodromou's avatar
Evan Prodromou committed
90 91
                if ($this->filter($notice)) {
                    $filtered[] = $notice;
92
                    if (count($filtered) >= $total) {
93 94 95 96 97 98 99 100
                        break;
                    }
                }
            }

            // XXX: make these smarter; factor hit rate into $askFor

            $startAt += $askFor;
Evan Prodromou's avatar
Evan Prodromou committed
101 102 103
            
            $hits = count($filtered);

104 105
            $lastAsk = $askFor;

Evan Prodromou's avatar
Evan Prodromou committed
106 107 108
            if ($hits === 0) {
                $askFor = max(min(2 * $askFor, NOTICES_PER_PAGE * 50), NOTICES_PER_PAGE);
            } else {
109
                $askFor = max(min(intval(ceil(($total - $hits)*$startAt/$hits)), NOTICES_PER_PAGE * 50), NOTICES_PER_PAGE);
Evan Prodromou's avatar
Evan Prodromou committed
110
            }
111

Evan Prodromou's avatar
Evan Prodromou committed
112 113
            $round++;

114
        } while (count($filtered) < $total && $results >= $lastAsk);
115 116 117 118 119 120 121 122 123 124 125

        return new ArrayWrapper(array_slice($filtered, $offset, $limit));
    }

    function getNoticeIds($offset, $limit, $sinceId, $maxId)
    {
        $notices = $this->getNotices($offset, $limit, $sinceId, $maxId);

        $ids = array();

        while ($notices->fetch()) {
126
            $ids[] = $notices->id;
127 128 129 130
        }

        return $ids;
    }
Evan Prodromou's avatar
Evan Prodromou committed
131 132 133 134 135

    function prefill($notices)
    {
        return;
    }
136
}