From d0fda5f1b465754bad0cb828ed42cd28feb5195a Mon Sep 17 00:00:00 2001 From: Frederic Guillot Date: Sun, 1 May 2016 22:32:50 -0400 Subject: [PATCH] Added limit option for producer --- producer.php | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/producer.php b/producer.php index 53facaf..aca112f 100644 --- a/producer.php +++ b/producer.php @@ -8,9 +8,14 @@ if (php_sapi_name() !== 'cli') { die('This script can run only from the command line.'.PHP_EOL); } +$options = getopt('', array( + 'limit::', +)); + +$limit = ! empty($options['limit']) && ctype_digit($options['limit']) ? (int) $options['limit'] : Model\Feed\LIMIT_ALL; $connection = new Pheanstalk(BEANSTALKD_HOST); -foreach (Model\Feed\get_ids() as $feed_id) { +foreach (Model\Feed\get_ids($limit) as $feed_id) { $connection ->useTube(BEANSTALKD_QUEUE) ->put($feed_id, Pheanstalk::DEFAULT_PRIORITY, Pheanstalk::DEFAULT_DELAY, BEANSTALKD_TTL);