PHRAS-4085_data-volumes-api

bin/maintenance

@@ -15,6 +15,7 @@ use Alchemy\Phrasea\Command\Maintenance\CleanRightsCommand;
 use Alchemy\Phrasea\Command\Maintenance\CleanWebhookLogsCommand;
 use Alchemy\Phrasea\Command\Maintenance\CleanWorkerRunningJobCommand;
 use Alchemy\Phrasea\Command\Maintenance\SessionsCommand;
+use Alchemy\Phrasea\Command\Maintenance\LazaretFilesSetSizeCommand;
 
 require_once __DIR__ . '/../lib/autoload.php';
 
@@ -59,4 +60,6 @@ $cli->command(new CleanLogViewCommand());
 
 $cli->command(new CleanWebhookLogsCommand());
 
+$cli->command(new LazaretFilesSetSizeCommand());
+
 $cli->run();

lib/Alchemy/Phrasea/Border/Manager.php

@@ -405,6 +405,8 @@ protected function createLazaret(File $file, Visa $visa, LazaretSession $session
 
         $lazaretFile->setSession($session);
 
+        $lazaretFile->setSize($file->getFile()->getSize());
+
         $this->app['orm.em']->persist($lazaretFile);
 
         foreach ($file->getAttributes() as $fileAttribute) {

lib/Alchemy/Phrasea/Command/Maintenance/LazaretFilesSetSizeCommand.php

@@ -0,0 +1,54 @@
+<?php
+
+namespace Alchemy\Phrasea\Command\Maintenance;
+
+use Alchemy\Phrasea\Command\Command;
+use Alchemy\Phrasea\Model\Entities\LazaretFile;
+use Alchemy\Phrasea\Model\Repositories\LazaretFileRepository;
+use Doctrine\ORM\EntityManager;
+use Symfony\Component\Console\Input\InputInterface;
+use Symfony\Component\Console\Input\InputOption;
+use Symfony\Component\Console\Output\OutputInterface;
+
+class LazaretFilesSetSizeCommand extends Command
+{
+    public function __construct()
+    {
+        parent::__construct('lazaret:set_sizes');
+
+        $this
+            ->setDescription('Set the null size in the LazaretFiles table')
+            ->addOption('dry-run', null, InputOption::VALUE_NONE, 'dry run, count')
+
+            ->setHelp('');
+    }
+
+    public function doExecute(InputInterface $input, OutputInterface $output)
+    {
+        /** @var LazaretFileRepository $lazaretRepository */
+        $lazaretRepository = $this->container['repo.lazaret-files'];
+
+        $lazaretNullSizes = $lazaretRepository->findBy(['size' => null]);
+
+        $path = $this->container['tmp.lazaret.path'];
+        /** @var EntityManager $em */
+        $em = $this->container['orm.em'];
+
+        if (!$input->getOption('dry-run')) {
+            /** @var LazaretFile $lazaretNullSize */
+            foreach ($lazaretNullSizes as $lazaretNullSize) {
+                $lazaretFileName = $path .'/'.$lazaretNullSize->getFilename();
+                $media = $this->container->getMediaFromUri($lazaretFileName);
+
+                $lazaretNullSize->setSize($media->getFile()->getSize());
+                $em->persist($lazaretNullSize);
+            }
+
+            $em->flush();
+
+            $output->writeln(sprintf("%d LazaretFiles done!", count($lazaretNullSizes)));
+        } else {
+            $output->writeln(sprintf("%d LazaretFiles to update!", count($lazaretNullSizes)));
+        }
+    }
+}

lib/Alchemy/Phrasea/Model/Entities/LazaretFile.php

@@ -96,6 +96,11 @@ class LazaretFile
      */
     private $session;
 
+    /**
+     * @ORM\Column(type="bigint", nullable=true)
+     */
+    private $size;
+
     /**
      * Constructor
      */
@@ -322,6 +327,29 @@ public function setUpdated(\DateTime $updated)
         return $this;
     }
 
+    /**
+     * Set size
+     *
+     * @param  integer     $size
+     * @return LazaretFile
+     */
+    public function setSize($size)
+    {
+        $this->size = $size;
+
+        return $this;
+    }
+
+    /**
+     * Get size
+     *
+     * @return integer
+     */
+    public function getSize()
+    {
+        return $this->size;
+    }
+
     /**
      * Get updated
      *