src/Wallabag/ImportBundle/Controller/ImportController.php

   1 <?php
   2
   3 namespace Wallabag\ImportBundle\Controller;
   4
   5 use Symfony\Bundle\FrameworkBundle\Controller\Controller;
   6 use Sensio\Bundle\FrameworkExtraBundle\Configuration\Route;
   7
   8 class ImportController extends Controller
   9 {
  10     /**
  11      * @Route("/", name="import")
  12      */
  13     public function importAction()
  14     {
  15         return $this->render('WallabagImportBundle:Import:index.html.twig', [
  16             'imports' => $this->get('wallabag_import.chain')->getAll(),
  17         ]);
  18     }
  19
  20     /**
  21      * Display how many messages are queue (both in Redis and RabbitMQ).
  22      * Only for admins.
  23      */
  24     public function checkQueueAction()
  25     {
  26         $nbRedisMessages = null;
  27         $nbRabbitMessages = null;
  28         $redisNotInstalled = false;
  29         $rabbitNotInstalled = false;
  30
  31         if (!$this->get('security.authorization_checker')->isGranted('ROLE_SUPER_ADMIN')) {
  32             return $this->render('WallabagImportBundle:Import:check_queue.html.twig');
  33         }
  34
  35         if ($this->get('craue_config')->get('import_with_rabbitmq')) {
  36             // in case rabbit is activated but not installed
  37             try {
  38                 $nbRabbitMessages = $this->getTotalMessageInRabbitQueue('pocket')
  39                     + $this->getTotalMessageInRabbitQueue('readability')
  40                     + $this->getTotalMessageInRabbitQueue('wallabag_v1')
  41                     + $this->getTotalMessageInRabbitQueue('wallabag_v2');
  42             } catch (\Exception $e) {
  43                 $rabbitNotInstalled = true;
  44             }
  45         } elseif ($this->get('craue_config')->get('import_with_redis')) {
  46             $redis = $this->get('wallabag_core.redis.client');
  47
  48             try {
  49                 $nbRedisMessages = $redis->llen('wallabag.import.pocket')
  50                     + $redis->llen('wallabag.import.readability')
  51                     + $redis->llen('wallabag.import.wallabag_v1')
  52                     + $redis->llen('wallabag.import.wallabag_v2');
  53             } catch (\Exception $e) {
  54                 $redisNotInstalled = true;
  55             }
  56         }
  57
  58         return $this->render('WallabagImportBundle:Import:check_queue.html.twig', [
  59             'nbRedisMessages' => $nbRedisMessages,
  60             'nbRabbitMessages' => $nbRabbitMessages,
  61             'redisNotInstalled' => $redisNotInstalled,
  62             'rabbitNotInstalled' => $rabbitNotInstalled,
  63         ]);
  64     }
  65
  66     /**
  67      * Count message in RabbitMQ queue.
  68      * It get one message without acking it (so it'll stay in the queue)
  69      * which will include the total of *other* messages in the queue.
  70      * Adding one to that messages will result in the full total message.
  71      *
  72      * @param string $importService The import service related: pocket, readability, wallabag_v1 or wallabag_v2
  73      *
  74      * @return int
  75      */
  76     private function getTotalMessageInRabbitQueue($importService)
  77     {
  78         $message = $this
  79             ->get('old_sound_rabbit_mq.import_'.$importService.'_consumer')
  80             ->getChannel()
  81             ->basic_get('wallabag.import.'.$importService);
  82
  83         if (null === $message) {
  84             return 0;
  85         }
  86
  87         return $message->delivery_info['message_count'] + 1;
  88     }
  89 }