Merge pull request #3 from ellgreen/replace-and-ignore

Added replace and ignore support

Author: ellgreen

README.md

@@ -41,7 +41,7 @@ LoadFile::file('/path/to/employees.csv', $local = true)
 LoadFile::file('/path/to/employees.csv', $local = true)
     ->into('employees')
     ->columns(['forename', 'surname', 'employee_id'])
-    ->ignore(1)
+    ->ignoreLines(1)
     ->load();
 ```
 

src/Builder.php

@@ -14,6 +14,10 @@ class Builder
     public ?string $file = null;
     public ?string $table = null;
     public bool $local = false;
+
+    public bool $replace = false;
+    public bool $ignore = false;
+
     public ?string $charset = null;
 
     public ?string $fieldsTerminatedBy = null;
@@ -75,6 +79,28 @@ public function local(bool $local): self
         return $this;
     }
 
+    public function replace(bool $replace = true): self
+    {
+        $this->replace = $replace;
+
+        if ($replace) {
+            $this->ignore = false;
+        }
+
+        return $this;
+    }
+
+    public function ignore(bool $ignore = true): self
+    {
+        $this->ignore = $ignore;
+
+        if ($ignore) {
+            $this->replace = false;
+        }
+
+        return $this;
+    }
+
     public function charset(?string $charset): self
     {
         $this->charset = $charset;
@@ -143,9 +169,9 @@ public function lines(string $startingBy, string $terminatedBy): self
         return $this;
     }
 
-    public function ignore(int $lines): self
+    public function ignoreLines(int $count): self
     {
-        $this->ignoreLines = $lines;
+        $this->ignoreLines = $count;
         return $this;
     }
 

src/Grammar.php

@@ -12,10 +12,20 @@ public function compileLoadFile(Builder $query): array
         $querySegments = collect();
 
         $querySegments->push(
-            'load data ' . ($query->local ? 'local' : '') . ' infile ' . $this->quoteString($query->file),
-            'into table ' . $this->wrapTable($query->table),
+            'load data' . ($query->local ? ' local' : ''),
+            'infile ' . $this->quoteString($query->file),
         );
 
+        if ($query->replace) {
+            $querySegments->push('replace');
+        }
+
+        if ($query->ignore) {
+            $querySegments->push('ignore');
+        }
+
+        $querySegments->push('into table ' . $this->wrapTable($query->table));
+
         if (isset($query->charset)) {
             $querySegments->push('character set ' . $this->quoteString($query->charset));
         }

tests/Feature/LoadFileTest.php

@@ -61,7 +61,7 @@ public function testIgnoreRow()
     {
         LoadFile::file(realpath(__DIR__ . '/../data/people.csv'), true)
             ->into('people')
-            ->ignore(1)
+            ->ignoreLines(1)
             ->columns([DB::raw('@forename'), DB::raw('@surname'), 'dob'])
             ->set([
                 'greeting' => 'Hello',
@@ -83,4 +83,48 @@ public function testIgnoreRow()
             'greeting' => 'Hello',
         ]);
     }
+
+    public function testReplace()
+    {
+        LoadFile::file(realpath(__DIR__ . '/../data/people-simple.csv'), true)
+            ->replace()
+            ->into('people')
+            ->columns(['name', 'dob', 'greeting'])
+            ->fields(',', '"', '\\\\', true)
+            ->load();
+
+        $this->assertDatabaseHas('people', [
+            'name' => 'John Doe',
+            'dob' => '1980-01-01',
+            'greeting' => 'Bonjour',
+        ]);
+
+        $this->assertDatabaseHas('people', [
+            'name' => 'Jane Doe',
+            'dob' => '1975-06-30',
+            'greeting' => 'Hello',
+        ]);
+    }
+
+    public function testIgnore()
+    {
+        LoadFile::file(realpath(__DIR__ . '/../data/people-simple.csv'), true)
+            ->ignore()
+            ->into('people')
+            ->columns(['name', 'dob', 'greeting'])
+            ->fields(',', '"', '\\\\', true)
+            ->load();
+
+        $this->assertDatabaseHas('people', [
+            'name' => 'John Doe',
+            'dob' => '1980-01-01',
+            'greeting' => 'Bonjour',
+        ]);
+
+        $this->assertDatabaseHas('people', [
+            'name' => 'Jane Doe',
+            'dob' => '1975-06-30',
+            'greeting' => 'Hello',
+        ]);
+    }
 }

tests/Unit/BuilderTest.php

@@ -53,6 +53,20 @@ public function testSetLocal()
         $this->assertTrue($this->builder->local);
     }
 
+    public function testSetReplace()
+    {
+        $this->builder->replace();
+
+        $this->assertTrue($this->builder->replace);
+    }
+
+    public function testSetIgnore()
+    {
+        $this->builder->ignore();
+
+        $this->assertTrue($this->builder->ignore);
+    }
+
     public function testSetCharset()
     {
         $this->builder->charset($charset = 'utf8mb4');
@@ -114,9 +128,9 @@ public function testSetLines()
         $this->assertSame('\\n', $this->builder->linesTerminatedBy);
     }
 
-    public function testSetIgnore()
+    public function testSetIgnoreLines()
     {
-        $this->builder->ignore(1);
+        $this->builder->ignoreLines(1);
 
         $this->assertSame(1, $this->builder->ignoreLines);
     }

tests/Unit/GrammarTest.php

@@ -37,9 +37,29 @@ public function testSimpleCompile()
         SQL);
     }
 
+    public function testReplace()
+    {
+        $this->builder->replace();
+
+        $this->assertSqlAndBindings(<<<'SQL'
+            load data local infile '/path/to/employees.csv' replace
+            into table `employees` (`forename`, `surname`, `employee_id`)
+        SQL);
+    }
+
+    public function testIgnore()
+    {
+        $this->builder->ignore();
+
+        $this->assertSqlAndBindings(<<<'SQL'
+            load data local infile '/path/to/employees.csv' ignore
+            into table `employees` (`forename`, `surname`, `employee_id`)
+        SQL);
+    }
+
     public function testIgnoreLines()
     {
-        $this->builder->ignore(1);
+        $this->builder->ignoreLines(1);
 
         $this->assertSqlAndBindings(<<<'SQL'
             load data local infile '/path/to/employees.csv'