forked from GNUsocial/gnu-social
		
	
		
			
				
	
	
		
			68 lines
		
	
	
		
			2.4 KiB
		
	
	
	
		
			PHP
		
	
	
	
	
	
			
		
		
	
	
			68 lines
		
	
	
		
			2.4 KiB
		
	
	
	
		
			PHP
		
	
	
	
	
	
<?php
 | 
						|
/**
 | 
						|
 * Store last poll time in db, then check if they should be renewed (if so, enqueue).
 | 
						|
 * Can be called from a queue handler on a per-feed status to poll stuff.
 | 
						|
 *
 | 
						|
 * Used as internal feed polling mechanism (atom/rss)
 | 
						|
 *
 | 
						|
 * @category OStatus
 | 
						|
 * @package  GNUsocial
 | 
						|
 * @author   Mikael Nordfeldth <mmn@hethane.se>
 | 
						|
 * @license  http://www.fsf.org/licensing/licenses/agpl.html AGPLv3
 | 
						|
 * @link     http://www.gnu.org/software/social/
 | 
						|
 */
 | 
						|
 | 
						|
if (!defined('GNUSOCIAL')) { exit(1); }
 | 
						|
 | 
						|
class FeedPoll {
 | 
						|
    const DEFAULT_INTERVAL = 5; // in minutes
 | 
						|
 | 
						|
    const QUEUE_CHECK = 'feedpoll-check';
 | 
						|
 | 
						|
    // TODO: Find some smart way to add feeds only once, so they don't get more than 1 feedpoll in the queue each
 | 
						|
    //       probably through sub_start sub_end trickery.
 | 
						|
    public static function enqueueNewFeeds(array $args=array()) {
 | 
						|
        if (!isset($args['interval']) || !is_int($args['interval']) || $args['interval']<=0) {
 | 
						|
            $args['interval'] = self::DEFAULT_INTERVAL;
 | 
						|
        }
 | 
						|
 | 
						|
        $args['interval'] *= 60;    // minutes to seconds
 | 
						|
 | 
						|
        $feedsub = new FeedSub();
 | 
						|
        $feedsub->sub_state = 'nohub';
 | 
						|
        // Find feeds that haven't been polled within the desired interval,
 | 
						|
        // though perhaps we're abusing the "last_update" field here?
 | 
						|
        $feedsub->whereAdd(sprintf('last_update < "%s"', common_sql_date(time()-$args['interval'])));
 | 
						|
        $feedsub->find();
 | 
						|
 | 
						|
        $qm = QueueManager::get();
 | 
						|
        while ($feedsub->fetch()) {
 | 
						|
            $orig = clone($feedsub);
 | 
						|
            $item = array('id' => $feedsub->id);
 | 
						|
            $qm->enqueue($item, self::QUEUE_CHECK);
 | 
						|
            $feedsub->last_update = common_sql_now();
 | 
						|
            $feedsub->update($orig);
 | 
						|
        }
 | 
						|
    }
 | 
						|
 | 
						|
    public function setupFeedSub(FeedSub $feedsub, $interval=300)
 | 
						|
    {
 | 
						|
        $orig = clone($feedsub);
 | 
						|
        $feedsub->sub_state = 'nohub';
 | 
						|
        $feedsub->sub_start = common_sql_date(time());
 | 
						|
        $feedsub->sub_end   = '';
 | 
						|
        $feedsub->last_update = common_sql_date(time()-$interval);  // force polling as soon as we can
 | 
						|
        $feedsub->update($orig);
 | 
						|
    }
 | 
						|
 | 
						|
    public function checkUpdates(FeedSub $feedsub)
 | 
						|
    {
 | 
						|
        $request = new HTTPClient();
 | 
						|
        $feed = $request->get($feedsub->uri);
 | 
						|
        if (!$feed->isOk()) {
 | 
						|
            throw new ServerException('FeedSub could not fetch id='.$feedsub->id.' (Error '.$feed->getStatus().': '.$feed->getBody());
 | 
						|
        }
 | 
						|
        $feedsub->receive($feed->getBody(), null);
 | 
						|
    }
 | 
						|
}
 |