Use repository instead of query builder

src/Console/UpdateFormatsCommand.php

@@ -31,6 +31,7 @@ use Symfony\Component\Console\Input\ArrayInput;
 use Symfony\Component\Console\Input\InputInterface;
 use Symfony\Component\Console\Output\NullOutput;
 use Symfony\Component\Console\Output\OutputInterface;
+use Symfony\Component\Validator\Exception\ValidationFailedException;
 
 /**
  * Synchronize metadata formats in database with configuration.
@@ -59,7 +60,8 @@ class UpdateFormatsCommand extends Command
                     continue;
                 }
             }
-            if (Database::getInstance()->addOrUpdateMetadataFormat($prefix, $format['namespace'], $format['schema'])) {
+            try {
+                Database::getInstance()->addOrUpdateMetadataFormat($prefix, $format['namespace'], $format['schema']);
                 ++$added;
                 $output->writeln([
                     sprintf(
@@ -67,12 +69,13 @@ class UpdateFormatsCommand extends Command
                         $prefix
                     )
                 ]);
-            } else {
+            } catch (ValidationFailedException $exception) {
                 $output->writeln([
                     sprintf(
                         ' [ERROR] Could not add or update metadata format "%s". ',
                         $prefix
-                    )
+                    ),
+                    $exception->getMessage()
                 ]);
             }
         }

src/Database.php

@@ -32,7 +32,6 @@ use Doctrine\ORM\EntityManager;
 use Doctrine\ORM\Mapping\Driver\AttributeDriver;
 use Doctrine\ORM\Proxy\ProxyFactory;
 use Doctrine\ORM\Tools\Pagination\Paginator;
-use Exception;
 use OCC\Basics\Traits\Singleton;
 use OCC\OaiPmh2\Database\Format;
 use OCC\OaiPmh2\Database\Record;
@@ -41,6 +40,7 @@ use OCC\OaiPmh2\Database\Set;
 use OCC\OaiPmh2\Database\Token;
 use Symfony\Component\Cache\Adapter\PhpFilesAdapter;
 use Symfony\Component\Filesystem\Path;
+use Symfony\Component\Validator\Exception\ValidationFailedException;
 
 /**
  * Handles all database shenanigans.
@@ -76,37 +76,29 @@ class Database
      * @param string $namespace The namespace URI
      * @param string $schema The schema URL
      *
-     * @return bool Whether the format was inserted/updated successfully
+     * @return void
+     *
+     * @throws ValidationFailedException
      */
-    public function addOrUpdateMetadataFormat(string $prefix, string $namespace, string $schema): bool
+    public function addOrUpdateMetadataFormat(string $prefix, string $namespace, string $schema): void
     {
-        $inDatabase = $this->getMetadataFormats()->getQueryResult();
+        $format = $this->entityManager->find(Format::class, $prefix);
-        if (in_array($prefix, array_keys($inDatabase), true)) {
+        if (isset($format)) {
             try {
-                $dql = $this->entityManager->createQueryBuilder();
+                $format->setNamespace($namespace);
-                $dql->update(Format::class, 'format')
+                $format->setSchema($schema);
-                    ->set('format.namespace', ':namepsace')
+            } catch (ValidationFailedException $exception) {
-                    ->set('format.xmlSchema', ':schema')
+                throw $exception;
-                    ->where($dql->expr()->eq('format.prefix', ':prefix'))
-                    ->setParameter('prefix', $prefix)
-                    ->setParameter('namespace', $namespace)
-                    ->setParameter('schema', $schema);
-                $query = $dql->getQuery();
-                $query->execute();
-                return true;
-            } catch (Exception) {
-                return false;
             }
         } else {
             try {
                 $format = new Format($prefix, $namespace, $schema);
+            } catch (ValidationFailedException $exception) {
+                throw $exception;
+            }
+        }
         $this->entityManager->persist($format);
         $this->entityManager->flush();
-                return true;
-            } catch (Exception) {
-                return false;
-            }
-        }
     }
 
     /**
@@ -183,17 +175,13 @@ class Database
      */
     public function getRecord(string $identifier, string $metadataPrefix): ?Record
     {
-        $dql = $this->entityManager->createQueryBuilder();
+        return $this->entityManager->find(
-        $dql->select('record')
+            Record::class,
-            ->from(Record::class, 'record')
+            [
-            ->where($dql->expr()->eq('record.identifier', ':identifier'))
+                'identifier' => $identifier,
-            ->andWhere($dql->expr()->eq('record.format', ':format'))
+                'format' => $metadataPrefix
-            ->setParameter('identifier', $identifier)
+            ]
-            ->setParameter('format', $metadataPrefix)
+        );
-            ->setMaxResults(1);
-        $query = $dql->getQuery();
-        /** @var ?Record */
-        return $query->getOneOrNullResult();
     }
 
     /**
@@ -365,15 +353,12 @@ class Database
      */
     public function removeMetadataFormat(string $prefix): bool
     {
-        $dql = $this->entityManager->createQueryBuilder();
+        $format = $this->entityManager->find(Format::class, $prefix);
-        $dql->delete(Format::class, 'format')
+        if (isset($format)) {
-            ->where($dql->expr()->eq('format.prefix', ':prefix'))
+            $this->entityManager->remove($format);
-            ->setParameter('prefix', $prefix);
+            $this->entityManager->flush();
-        $query = $dql->getQuery();
-        try {
-            $query->execute();
             return true;
-        } catch (Exception) {
+        } else {
             return false;
         }
     }