]>
Commit | Line | Data |
---|---|---|
d51b38ed NL |
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 | /** | |
7019c7cf | 11 | * @Route("/", name="import") |
d51b38ed | 12 | */ |
252ebd60 | 13 | public function importAction() |
d51b38ed | 14 | { |
7019c7cf JB |
15 | return $this->render('WallabagImportBundle:Import:index.html.twig', [ |
16 | 'imports' => $this->get('wallabag_import.chain')->getAll(), | |
17 | ]); | |
d51b38ed | 18 | } |
e01a3c98 JB |
19 | |
20 | /** | |
21 | * Display how many messages are queue (both in Redis and RabbitMQ). | |
4fc99824 | 22 | * Only for admins. |
e01a3c98 JB |
23 | */ |
24 | public function checkQueueAction() | |
25 | { | |
26 | $nbRedisMessages = null; | |
27 | $nbRabbitMessages = null; | |
13a522df JB |
28 | $redisNotInstalled = false; |
29 | $rabbitNotInstalled = false; | |
e01a3c98 | 30 | |
4fc99824 | 31 | if (!$this->get('security.authorization_checker')->isGranted('ROLE_SUPER_ADMIN')) { |
13a522df | 32 | return $this->render('WallabagImportBundle:Import:check_queue.html.twig'); |
4fc99824 JB |
33 | } |
34 | ||
e01a3c98 | 35 | if ($this->get('craue_config')->get('import_with_rabbitmq')) { |
13a522df JB |
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') | |
ff1a5362 JB |
41 | + $this->getTotalMessageInRabbitQueue('wallabag_v2') |
42 | + $this->getTotalMessageInRabbitQueue('firefox') | |
43 | + $this->getTotalMessageInRabbitQueue('chrome') | |
44 | + $this->getTotalMessageInRabbitQueue('instapaper') | |
9ab024b4 | 45 | + $this->getTotalMessageInRabbitQueue('pinboard') |
ff1a5362 | 46 | ; |
13a522df JB |
47 | } catch (\Exception $e) { |
48 | $rabbitNotInstalled = true; | |
49 | } | |
e01a3c98 JB |
50 | } elseif ($this->get('craue_config')->get('import_with_redis')) { |
51 | $redis = $this->get('wallabag_core.redis.client'); | |
52 | ||
13a522df JB |
53 | try { |
54 | $nbRedisMessages = $redis->llen('wallabag.import.pocket') | |
55 | + $redis->llen('wallabag.import.readability') | |
56 | + $redis->llen('wallabag.import.wallabag_v1') | |
ff1a5362 JB |
57 | + $redis->llen('wallabag.import.wallabag_v2') |
58 | + $redis->llen('wallabag.import.firefox') | |
59 | + $redis->llen('wallabag.import.chrome') | |
60 | + $redis->llen('wallabag.import.instapaper') | |
9ab024b4 | 61 | + $redis->llen('wallabag.import.pinboard') |
ff1a5362 | 62 | ; |
13a522df JB |
63 | } catch (\Exception $e) { |
64 | $redisNotInstalled = true; | |
65 | } | |
e01a3c98 JB |
66 | } |
67 | ||
68 | return $this->render('WallabagImportBundle:Import:check_queue.html.twig', [ | |
69 | 'nbRedisMessages' => $nbRedisMessages, | |
70 | 'nbRabbitMessages' => $nbRabbitMessages, | |
13a522df JB |
71 | 'redisNotInstalled' => $redisNotInstalled, |
72 | 'rabbitNotInstalled' => $rabbitNotInstalled, | |
e01a3c98 JB |
73 | ]); |
74 | } | |
75 | ||
76 | /** | |
77 | * Count message in RabbitMQ queue. | |
78 | * It get one message without acking it (so it'll stay in the queue) | |
79 | * which will include the total of *other* messages in the queue. | |
80 | * Adding one to that messages will result in the full total message. | |
81 | * | |
82 | * @param string $importService The import service related: pocket, readability, wallabag_v1 or wallabag_v2 | |
83 | * | |
84 | * @return int | |
85 | */ | |
86 | private function getTotalMessageInRabbitQueue($importService) | |
87 | { | |
88 | $message = $this | |
89 | ->get('old_sound_rabbit_mq.import_'.$importService.'_consumer') | |
90 | ->getChannel() | |
91 | ->basic_get('wallabag.import.'.$importService); | |
92 | ||
93 | if (null === $message) { | |
94 | return 0; | |
95 | } | |
96 | ||
97 | return $message->delivery_info['message_count'] + 1; | |
98 | } | |
d51b38ed | 99 | } |