Merge pull request #16 from stephenlake/patch-2

Fix #6 for both mysql and postgres - my mistake 💩
igaster · Sep 13, 2018 · adbdabb · adbdabb
2 parents efe28fa + 8b7f583
commit adbdabb
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/src/commands/seedGeoFile.php b/src/commands/seedGeoFile.php
@@ -11,9 +11,14 @@ class seedGeoFile extends Command
     protected $description = 'Load + Parse + Save to DB a geodata file.';
 
     private $pdo;
+    private $driver;
 
     public function __construct() {
         parent::__construct();
+
+        $connection = config('database.default');
+        $this->driver = strtolower(config("database.connections.{$connection}.driver"));
+
         $this->pdo = \DB::connection()->getPdo(\PDO::FETCH_ASSOC);
         if (!\Schema::hasTable('geo'))
             return;
@@ -139,7 +144,12 @@ public function handle() {
 
         // Store Tree in DB
         $this->info("Writing in Database</info>");
-        $stmt = $this->pdo->prepare("INSERT INTO geo (\"id\", \"parent_id\", \"left\", \"right\", \"depth\", \"name\", \"alternames\", \"country\", \"level\", \"population\", \"lat\", \"long\") VALUES (:id, :parent_id, :left, :right, :depth, :name, :alternames, :country, :level, :population, :lat, :long)");
+
+        if ($this->driver == 'mysql') {
+            $stmt = $this->pdo->prepare("INSERT INTO geo (`id`, `parent_id`, `left`, `right`, `depth`, `name`, `alternames`, `country`, `level`, `population`, `lat`, `long`) VALUES (:id, :parent_id, :left, :right, :depth, :name, :alternames, :country, :level, :population, :lat, :long)");
+        } else {
+            $stmt = $this->pdo->prepare("INSERT INTO geo (\"id\", \"parent_id\", \"left\", \"right\", \"depth\", \"name\", \"alternames\", \"country\", \"level\", \"population\", \"lat\", \"long\") VALUES (:id, :parent_id, :left, :right, :depth, :name, :alternames, :country, :level, :population, :lat, :long)");
+        }
 
         $count = 0;
         $totalCount = count($this->geoItems->items);