Add better error handling and message for exchange access refuse

cloudamqp · Jan 13, 2025 · 853a545 · 853a545
1 parent 1b50ad1
commit 853a545
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/lavinmq/http/handler/error_handler.cr b/src/lavinmq/http/handler/error_handler.cr
@@ -18,6 +18,10 @@ module LavinMQ
         Log.info { "method=#{context.request.method} path=#{context.request.path} status=#{context.response.status_code} message=\"#{ex.message}\"" }
       rescue ex : IO::Error | ::HTTP::Server::ClientError
         Log.info { "method=#{context.request.method} path=#{context.request.path} error=\"#{ex.inspect}\"" }
+      rescue ex : LavinMQ::Exchange::AccessRefused
+        Log.error { "method=#{context.request.method} path=#{context.request.path} status=403 error=#{ex.inspect_with_backtrace}" }
+        context.response.status_code = 403
+        {error: "access_refused", reason: "Access Refused"}.to_json(context.response)
       rescue ex : Exception
         Log.error { "method=#{context.request.method} path=#{context.request.path} status=500 error=#{ex.inspect_with_backtrace}" }
         context.response.status_code = 500