Fixed reference to registry to get EM

mridgway · Sep 15, 2010 · 4ecf152 · 4ecf152
1 parent 82f610a
commit 4ecf152
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/lib/ZendX/Doctrine2/Paginator.php b/lib/ZendX/Doctrine2/Paginator.php
@@ -47,7 +47,7 @@ public function count()
             if (null === $this->_countQb) {
                 $this->_countQb = clone $this->_qb;
                 $from = $this->_countQb->getDqlPart('from');
-                $identifierNames = \Zend_Registry::get('doctrine')->getClassMetadata($from[0]->getFrom())->getIdentifierFieldNames();
+                $identifierNames = $this->_qb->getEntityManager()->getClassMetadata($from[0]->getFrom())->getIdentifierFieldNames();
                 $this->_countQb->select('count('.$this->_countQb->getRootAlias().'.' . $identifierNames[0] . ')');
             }
             return $this->_rowCount = $this->_countQb->getQuery()->getSingleScalarResult();
@@ -67,6 +67,11 @@ public function getItems($offset, $itemCountPerPage)
         return $this->_qb->getQuery()->getResult();
     }
 
+    /**
+     * Set a custom query builder to get the count
+     *
+     * @param Doctrine\ORM\QueryBuilder $qb
+     */
     public function setCountQueryBuilder(\Doctrine\ORM\QueryBuilder $qb)
     {
         $this->_countQb = $qb;