iv-org-mirror/shard-kemal
1
0
Fork 0
mirror of https://gitea.invidious.io/iv-org/shard-kemal.git synced 2024-08-15 00:53:36 +00:00
Code Issues Projects Releases Packages Wiki Activity

Improve websocket match logic. Fixes #401

Browse source
This commit is contained in:
Serdar Dogruyol 2017-09-14 19:59:22 +03:00
parent e07be72dcf
commit 00981bcf44
2 changed files with 24 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

9
src/kemal/websocket_handler.cr
View file

@ -11,7 +11,7 @@ module Kemal
end
def call(context : HTTP::Server::Context)
return call_next(context) unless context.ws_route_defined?
return call_next(context) unless context.ws_route_defined? && websocket_upgrade_request?(context)
context.request.url_params ||= context.ws_route_lookup.params
content = context.websocket.call(context)
context.response.print(content)
@ -34,5 +34,12 @@ module Kemal
private def radix_path(method, path)
"/#{method.downcase}#{path}"
end
private def websocket_upgrade_request?(context)
return false unless upgrade = context.request.headers["Upgrade"]?
return false unless upgrade.compare("websocket", case_insensitive: true) == 0
context.request.headers.includes_word?("Connection", "Upgrade")
end
end
end