Add the getNewQueryBuilder and the getQueryBuilderResult functions

thomasbeaujean · Mar 20, 2015 · 7bb17ac · 7bb17ac
1 parent e1eb469
commit 7bb17ac
Show file tree

Hide file tree

Showing 2 changed files with 32 additions and 2 deletions.
diff --git a/Generator/RepositoryGenerator.php b/Generator/RepositoryGenerator.php
@@ -81,13 +81,15 @@ public function generateFiles($directory)
             $tableNames = $description['tableNames'];
 
             foreach ($tableNames as $tableIndex => $tableName) {
-                $renderedTemplate = $twig->render($this->topRepositoryTemple, array('tableName' => $tableName, 'bundleName' => $bundleName));
+                $entityDql = lcfirst($tableName);
+
+                $renderedTemplate = $twig->render($this->topRepositoryTemple, array('tableName' => $tableName, 'bundleName' => $bundleName, 'entityDql' => $entityDql));
 
                 //parse the columns
                 foreach ($fieldNames[$tableIndex] as $columnName) {
                     $parameters = array(
                         'entity' => $tableName,
-                        'entityDql' => lcfirst($tableName),
+                        'entityDql' => $entityDql,
                         'column' => ucfirst($columnName),
                         'columnDql' => $columnName
                     );

diff --git a/Resources/views/Generator/TopRepositoryTemplate.html.twig b/Resources/views/Generator/TopRepositoryTemplate.html.twig
@@ -28,3 +28,31 @@ class {{ tableName }}Repository extends EntityRepository
 
         return self::$parameterIndex;
     }
+
+    /**
+     * Get a new query builder
+     *
+     * @return integer The index
+     */
+    public function getNewQueryBuilder()
+    {
+        $qb = $this->createQueryBuilder('{{ entityDql }}');
+
+        return $qb;
+    }
+
+    /**
+     * Get the result of the query builder
+     *
+     * @return unknown The results
+     */
+    public static function getQueryBuilderResult(QueryBuilder $qb)
+    {
+         $query = $qb->getQuery();
+         $results = $query->getResult();
+
+         return $results;
+    }
+
+
+