add direction option

orangehill · Jul 8, 2017 · 839c034 · 839c034
1 parent 68e560d
commit 839c034
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 5 deletions.
diff --git a/src/Orangehill/Iseed/Iseed.php b/src/Orangehill/Iseed/Iseed.php
@@ -59,7 +59,7 @@ public function readStubFile($file)
      * @return bool
      * @throws Orangehill\Iseed\TableNotFoundException
      */
-    public function generateSeed($table, $database = null, $max = 0, $exclude = null, $prerunEvent = null, $postrunEvent = null, $dumpAuto = true, $indexed = true, $orderBy = null)
+    public function generateSeed($table, $database = null, $max = 0, $exclude = null, $prerunEvent = null, $postrunEvent = null, $dumpAuto = true, $indexed = true, $orderBy = null, $direction = 'ASC')
     {
         if (!$database) {
             $database = config('database.default');
@@ -73,7 +73,7 @@ public function generateSeed($table, $database = null, $max = 0, $exclude = null
         }
 
         // Get the data
-        $data = $this->getData($table, $max, $exclude, $orderBy);
+        $data = $this->getData($table, $max, $exclude, $orderBy, $direction);
 
         // Repack the data
         $dataArray = $this->repackSeedData($data);
@@ -128,7 +128,7 @@ public function getSeedPath()
      * @param  string $table
      * @return Array
      */
-    public function getData($table, $max, $exclude = null, $orderBy = null)
+    public function getData($table, $max, $exclude = null, $orderBy = null, $direction = 'ASC')
     {
         $result = \DB::connection($this->databaseName)->table($table);
 
@@ -138,7 +138,7 @@ public function getData($table, $max, $exclude = null, $orderBy = null)
         }
 
         if($orderBy) {
-            $result = $result->orderBy($orderBy, 'DESC');
+            $result = $result->orderBy($orderBy, $direction);
         }
 
         if ($max) {

diff --git a/src/Orangehill/Iseed/IseedCommand.php b/src/Orangehill/Iseed/IseedCommand.php
@@ -52,6 +52,7 @@ public function fire()
         $dumpAuto = intval($this->option('dumpauto'));
         $indexed = !$this->option('noindex');
         $orderBy = $this->option('orderby');
+        $direction = $this->option('direction');
 
         if ($chunkSize < 1) {
             $chunkSize = null;
@@ -85,7 +86,8 @@ public function fire()
                         $postrunEvent,
                         $dumpAuto,
                         $indexed,
-                        $orderBy
+                        $orderBy,
+                        $direction
                     ),
                     $table
                 );
@@ -143,6 +145,7 @@ protected function getOptions()
             array('dumpauto', null, InputOption::VALUE_OPTIONAL, 'run composer dump-autoload', true),
             array('noindex', null, InputOption::VALUE_NONE, 'no indexing in the seed', null),
             array('orderby', null, InputOption::VALUE_OPTIONAL, 'orderby desc by column', null),
+            array('direction', null, InputOption::VALUE_OPTIONAL, 'orderby direction', null),
         );
     }