]>
Commit | Line | Data |
---|---|---|
7d862f83 JB |
1 | <?php |
2 | ||
3 | namespace Wallabag\ImportBundle\Consumer; | |
4 | ||
5 | use Doctrine\ORM\EntityManager; | |
7d862f83 JB |
6 | use Psr\Log\LoggerInterface; |
7 | use Psr\Log\NullLogger; | |
7816eb62 | 8 | use Symfony\Component\EventDispatcher\EventDispatcherInterface; |
f808b016 JB |
9 | use Wallabag\CoreBundle\Entity\Entry; |
10 | use Wallabag\CoreBundle\Entity\Tag; | |
7816eb62 | 11 | use Wallabag\CoreBundle\Event\EntrySavedEvent; |
f808b016 JB |
12 | use Wallabag\ImportBundle\Import\AbstractImport; |
13 | use Wallabag\UserBundle\Repository\UserRepository; | |
7d862f83 JB |
14 | |
15 | abstract class AbstractConsumer | |
16 | { | |
17 | protected $em; | |
18 | protected $userRepository; | |
19 | protected $import; | |
001cc716 | 20 | protected $eventDispatcher; |
7d862f83 JB |
21 | protected $logger; |
22 | ||
7816eb62 | 23 | public function __construct(EntityManager $em, UserRepository $userRepository, AbstractImport $import, EventDispatcherInterface $eventDispatcher, LoggerInterface $logger = null) |
7d862f83 JB |
24 | { |
25 | $this->em = $em; | |
26 | $this->userRepository = $userRepository; | |
27 | $this->import = $import; | |
7816eb62 | 28 | $this->eventDispatcher = $eventDispatcher; |
7d862f83 JB |
29 | $this->logger = $logger ?: new NullLogger(); |
30 | } | |
31 | ||
32 | /** | |
33 | * Handle a message and save it. | |
34 | * | |
35 | * @param string $body Message from the queue (in json) | |
36 | * | |
37 | * @return bool | |
38 | */ | |
39 | protected function handleMessage($body) | |
40 | { | |
41 | $storedEntry = json_decode($body, true); | |
42 | ||
43 | $user = $this->userRepository->find($storedEntry['userId']); | |
44 | ||
45 | // no user? Drop message | |
46 | if (null === $user) { | |
47 | $this->logger->warning('Unable to retrieve user', ['entry' => $storedEntry]); | |
48 | ||
b45b6b67 NL |
49 | // return true to skip message |
50 | return true; | |
7d862f83 JB |
51 | } |
52 | ||
53 | $this->import->setUser($user); | |
54 | ||
55 | $entry = $this->import->parseEntry($storedEntry); | |
56 | ||
57 | if (null === $entry) { | |
1e3d74a9 | 58 | $this->logger->warning('Entry already exists', ['entry' => $storedEntry]); |
7d862f83 | 59 | |
1e3d74a9 JB |
60 | // return true to skip message |
61 | return true; | |
7d862f83 JB |
62 | } |
63 | ||
64 | try { | |
65 | $this->em->flush(); | |
66 | ||
7816eb62 JB |
67 | // entry saved, dispatch event about it! |
68 | $this->eventDispatcher->dispatch(EntrySavedEvent::NAME, new EntrySavedEvent($entry)); | |
69 | ||
7d862f83 JB |
70 | // clear only affected entities |
71 | $this->em->clear(Entry::class); | |
72 | $this->em->clear(Tag::class); | |
73 | } catch (\Exception $e) { | |
74 | $this->logger->warning('Unable to save entry', ['entry' => $storedEntry, 'exception' => $e]); | |
75 | ||
76 | return false; | |
77 | } | |
78 | ||
f808b016 | 79 | $this->logger->info('Content with url imported! (' . $entry->getUrl() . ')'); |
7d862f83 JB |
80 | |
81 | return true; | |
82 | } | |
83 | } |