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