change application schema (#142)

Co-authored-by: Evgeniy Esaulkov <58729845+EvgeniyEsaulkov@users.noreply.github.com>
fs · Apr 15, 2021 · 8d7ad87 · 8d7ad87
1 parent 86418bb
commit 8d7ad87
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/app/graphql/application_schema.rb b/app/graphql/application_schema.rb
@@ -8,4 +8,10 @@ class ApplicationSchema < GraphQL::Schema
   use GraphQL::Pagination::Connections
   use GraphQL::Analysis::AST
   use GraphQL::Batch
+  use GraphQL::Execution::Errors
+
+  rescue_from(ActiveRecord::RecordNotFound) do |_err, _obj, _args, _ctx, field|
+    raise GraphQL::ExecutionError.new("#{field.type.unwrap.graphql_name} not found",
+                                      extensions: { message: "Not Found", status: 404, code: :not_found })
+  end
 end