src/Eccube/Repository/CategoryRepository.php line 94

Open in your IDE?
  1. <?php
  3. /*
  4.  * This file is part of EC-CUBE
  5.  *
  6.  * Copyright(c) EC-CUBE CO.,LTD. All Rights Reserved.
  7.  *
  8.  * http://www.ec-cube.co.jp/
  9.  *
  10.  * For the full copyright and license information, please view the LICENSE
  11.  * file that was distributed with this source code.
  12.  */
  14. namespace Eccube\Repository;
  16. use Doctrine\DBAL\Exception\DriverException;
  17. use Doctrine\DBAL\Exception\ForeignKeyConstraintViolationException;
  18. use Doctrine\Persistence\ManagerRegistry as RegistryInterface;
  19. use Eccube\Common\EccubeConfig;
  20. use Eccube\Entity\Category;
  22. /**
  23.  * CategoryRepository
  24.  *
  25.  * This class was generated by the Doctrine ORM. Add your own custom
  26.  * repository methods below.
  27.  */
  28. class CategoryRepository extends AbstractRepository
  29. {
  30.     /**
  31.      * @var EccubeConfig
  32.      */
  33.     protected $eccubeConfig;
  35.     /**
  36.      * CategoryRepository constructor.
  37.      *
  38.      * @param RegistryInterface $registry
  39.      * @param EccubeConfig $eccubeConfig
  40.      */
  41.     public function __construct(
  42.         RegistryInterface $registry,
  43.         EccubeConfig $eccubeConfig
  44.     ) {
  45.         parent::__construct($registry, Category::class);
  46.         $this->eccubeConfig = $eccubeConfig;
  47.     }
  49.     /**
  50.      * 全カテゴリの合計を取得する.
  51.      *
  52.      * @return int 全カテゴリの合計数
  53.      */
  54.     public function getTotalCount()
  55.     {
  56.         return $this
  57.             ->createQueryBuilder('c')
  58.             ->select('COALESCE(COUNT(c.id), 0)')
  59.             ->getQuery()
  60.             ->getSingleScalarResult();
  61.     }
  63.     /**
  64.      * カテゴリ一覧を取得する.
  65.      *
  66.      * 引数 $Parent を指定した場合は, 指定したカテゴリの子以下を取得する.
  67.      *
  68.      * @param Category|null $Parent 指定の親カテゴリ
  69.      * @param bool $flat trueの場合, 階層化されたカテゴリを一つの配列にまとめる
  70.      *
  71.      * @return Category[] カテゴリの配列
  72.      */
  73.     public function getList(Category $Parent = null, $flat = false)
  74.     {
  75.         $qb = $this->createQueryBuilder('c1')
  76.             ->select('c1, c2, c3, c4, c5')
  77.             ->leftJoin('c1.Children', 'c2')
  78.             ->leftJoin('c2.Children', 'c3')
  79.             ->leftJoin('c3.Children', 'c4')
  80.             ->leftJoin('c4.Children', 'c5')
  81.             ->orderBy('c1.sort_no', 'DESC')
  82.             ->addOrderBy('c2.sort_no', 'DESC')
  83.             ->addOrderBy('c3.sort_no', 'DESC')
  84.             ->addOrderBy('c4.sort_no', 'DESC')
  85.             ->addOrderBy('c5.sort_no', 'DESC');
  87.         if ($Parent) {
  88.             $qb->where('c1.Parent = :Parent')->setParameter('Parent', $Parent);
  89.         } else {
  90.             $qb->where('c1.Parent IS NULL');
  91.         }
  92.         $Categories = $qb->getQuery()
  93.             ->useResultCache(true, $this->getCacheLifetime())
  94.             ->getResult();
  96.         if ($flat) {
  97.             $array = [];
  98.             foreach ($Categories as $Category) {
  99.                 $array = array_merge($array, $Category->getSelfAndDescendants());
  100.             }
  101.             $Categories = $array;
  102.         }
  104.         return $Categories;
  105.     }
  107.     /**
  108.      * カテゴリを保存する.
  109.      *
  110.      * @param  Category $Category カテゴリ
  111.      */
  112.     public function save($Category)
  113.     {
  114.         if (!$Category->getId()) {
  115.             $Parent = $Category->getParent();
  116.             if ($Parent) {
  117.                 $sortNo = $Parent->getSortNo() - 1;
  118.             } else {
  119.                 $sortNo = $this->createQueryBuilder('c')
  120.                     ->select('COALESCE(MAX(c.sort_no), 0)')
  121.                     ->getQuery()
  122.                     ->getSingleScalarResult();
  123.             }
  125.             $Category->setSortNo($sortNo + 1);
  127.             $this
  128.                 ->createQueryBuilder('c')
  129.                 ->update()
  130.                 ->set('c.sort_no', 'c.sort_no + 1')
  131.                 ->where('c.sort_no > :sort_no')
  132.                 ->setParameter('sort_no', $sortNo)
  133.                 ->getQuery()
  134.                 ->execute();
  135.         }
  137.         $em = $this->getEntityManager();
  138.         $em->persist($Category);
  139.         $em->flush();
  140.     }
  142.     /**
  143.      * カテゴリを削除する.
  144.      *
  145.      * @param  Category $Category 削除対象のカテゴリ
  146.      *
  147.      * @throws ForeignKeyConstraintViolationException 外部キー制約違反の場合
  148.      * @throws DriverException SQLiteの場合, 外部キー制約違反が発生すると, DriverExceptionをthrowします.
  149.      */
  150.     public function delete($Category)
  151.     {
  152.         $this
  153.             ->createQueryBuilder('c')
  154.             ->update()
  155.             ->set('c.sort_no', 'c.sort_no - 1')
  156.             ->where('c.sort_no > :sort_no')
  157.             ->setParameter('sort_no', $Category->getSortNo())
  158.             ->getQuery()
  159.             ->execute();
  161.         $em = $this->getEntityManager();
  162.         $em->remove($Category);
  163.         $em->flush();
  164.     }
  165. }