entityManager = $entityManager; $this->classMetadata = $entityManager->getClassMetadata($this->getEntityClassName()); $this->doctrineRepository = new DoctrineEntityRepository($this->entityManager, $this->classMetadata); } /** * @param array $criteria * @param array|null $orderBy * @param int|null $limit * @param int|null $offset * @return T[] */ public function findBy(array $criteria, array $orderBy = null, $limit = null, $offset = null) { return $this->doctrineRepository->findBy($criteria, $orderBy, $limit, $offset); } public function countBy(array $criteria): int { return $this->doctrineRepository->count($criteria); } /** * @param array $criteria * @param array|null $orderBy * @return T|null */ public function findOneBy(array $criteria, array $orderBy = null) { return $this->doctrineRepository->findOneBy($criteria, $orderBy); } /** * @param mixed $id * @return T|null */ public function findOneById($id) { return $this->doctrineRepository->find($id); } /** * @return T[] */ public function findAll() { return $this->doctrineRepository->findAll(); } /** * @param T $entity */ public function persist($entity) { $this->entityManager->persist($entity); } public function truncate() { $tableName = $this->getTableName(); $connection = $this->entityManager->getConnection(); $connection->query('SET FOREIGN_KEY_CHECKS=0'); $q = "TRUNCATE $tableName"; $connection->executeUpdate($q); $connection->query('SET FOREIGN_KEY_CHECKS=1'); } /** * @param T $entity */ public function remove($entity) { $this->entityManager->remove($entity); } /** * @param T $entity */ public function refresh($entity) { $this->entityManager->refresh($entity); } public function flush() { $this->entityManager->flush(); } /** * @return class-string */ abstract protected function getEntityClassName(); protected function getTableName(): string { return $this->entityManager->getClassMetadata($this->getEntityClassName())->getTableName(); } public function insertOrUpdateMultiple(array $columns, array $data, ?DateTime $updatedAt = null): int { $tableName = $this->getTableName(); $entityColumns = $this->entityManager->getClassMetadata($this->getEntityClassName())->getColumnNames(); if (!$columns || !$data) { return 0; } $rows = []; $parameters = []; foreach ($data as $key => $item) { $paramNames = array_map(function (string $parameter) use ($key): string { return ":{$parameter}_{$key}"; }, $columns); foreach ($item as $columnKey => $column) { $parameters[$paramNames[$columnKey]] = $column; } $rows[] = "(" . implode(', ', $paramNames) . ")"; } $updateColumns = array_map(function (string $column): string { return "{$column} = VALUES($column)"; }, array_diff($columns, $this->ignoreColumnsForUpdate)); if ($updatedAt && in_array('updated_at', $entityColumns, true)) { $parameters['updated_at'] = $updatedAt; $updateColumns[] = "updated_at = :updated_at"; } // we want to reset deleted_at for updated rows if (in_array('deleted_at', $entityColumns, true)) { $updateColumns[] = 'deleted_at = NULL'; } return $this->entityManager->getConnection()->executeUpdate(" INSERT INTO {$tableName} (`" . implode("`, `", $columns) . "`) VALUES " . implode(", \n", $rows) . " ON DUPLICATE KEY UPDATE " . implode(", \n", $updateColumns) . " ", $parameters); } }