Merge pull request #51 from wave-framework/patch-handle-duplicate-keys

Adds duplicate key exception on insert

Author: phindmarsh

src/Wave/DB.php

@@ -466,8 +466,14 @@ public static function insert(Model &$object) {
             implode(',', $placeholders)
         );
 
-        $connection->prepare($sql)->execute($params);
-
+        try {
+            $connection->prepare($sql)->execute($params);
+        } catch (\PDOException $e){
+            if($e->getCode() === DB\Exception::SQLSTATE_DUPLICATE_KEY){
+                throw new DuplicateKeyException($e->getCode(), $e->getMessage());
+            }
+            throw $e;
+        }
         $primary_key = $object::_getPrimaryKey();
         if($primary_key !== null && count($primary_key) === 1) {
             $column = current($primary_key);

src/Wave/DB/Exception.php

@@ -4,5 +4,7 @@
 
 use Wave;
 
-class Exception extends Wave\Exception {
+class Exception extends Wave\Exception
+{
+    const SQLSTATE_DUPLICATE_KEY = 1062;
 }

src/Wave/DB/Exception/DuplicateKeyException.php

@@ -0,0 +1,11 @@
+<?php
+
+namespace Wave;
+
+class DuplicateKeyException extends \Wave\DB\Exception
+{
+    public function __construct($message, $code = null)
+    {
+        parent::__construct($message, $code);
+    }
+}