Adding OptimisticLockException to KernelExceptionListener

dontdrinkandroot · Mar 27, 2017 · df0c5bd · df0c5bd
1 parent 80cb77b
commit df0c5bd
Showing 1 changed file with 24 additions and 9 deletions.
diff --git a/Listener/KernelExceptionListener.php b/Listener/KernelExceptionListener.php
@@ -2,6 +2,7 @@
 
 namespace Dontdrinkandroot\RestBundle\Listener;
 
+use Doctrine\ORM\OptimisticLockException;
 use Symfony\Component\HttpFoundation\JsonResponse;
 use Symfony\Component\HttpFoundation\Request;
 use Symfony\Component\HttpFoundation\Response;
@@ -47,15 +48,7 @@ public function onKernelException(GetResponseForExceptionEvent $event)
             ];
         }
 
-        $statusCode = Response::HTTP_INTERNAL_SERVER_ERROR;
-
-        if ($exception instanceof InsufficientAuthenticationException) {
-            $statusCode = Response::HTTP_UNAUTHORIZED;
-        }
-
-        if ($exception instanceof BadCredentialsException) {
-            $statusCode = Response::HTTP_UNAUTHORIZED;
-        }
+        $statusCode = $this->resolveStatusCode($exception);
 
         $response = new JsonResponse($data);
 
@@ -100,4 +93,26 @@ public function setDebug(bool $debug)
     {
         $this->debug = $debug;
     }
+
+    /**
+     * @param $exception
+     *
+     * @return int
+     */
+    protected function resolveStatusCode($exception): int
+    {
+        if ($exception instanceof InsufficientAuthenticationException) {
+            return Response::HTTP_UNAUTHORIZED;
+        }
+
+        if ($exception instanceof BadCredentialsException) {
+            return Response::HTTP_UNAUTHORIZED;
+        }
+
+        if ($exception instanceof OptimisticLockException) {
+            return Response::HTTP_PRECONDITION_FAILED;
+        }
+
+        return Response::HTTP_INTERNAL_SERVER_ERROR;
+    }
 }