add not-found handler

LafayetteCollegeLibraries · Oct 9, 2017 · 51dcb21 · 51dcb21
1 parent f31aedb
commit 51dcb21
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/app/controllers/application_controller.rb b/app/controllers/application_controller.rb
@@ -1,2 +1,8 @@
 class ApplicationController < ActionController::API
+  rescue_from ActiveRecord::RecordNotFound, with: :not_found
+
+  private
+    def not_found(error)
+      render json: { error: error.message, status: 404 }, status: 404
+    end
 end