GithubStar/workflow
1
0
Fork 0
mirror of https://github.com/sogou/workflow.git synced 2026-10-28 02:56:43 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix set_request_uri for websocket

Browse Source
This commit is contained in:
holmes1412
2021-10-31 13:33:55 +08:00
parent 07eb9debb3
commit ea11a674fc
1 changed files with 9 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/factory/WebSocketTaskImpl.cc
+9 -1
View File
@@ -139,6 +139,8 @@ SubTask *ComplexWebSocketOutTask::done()
SubTask *ComplexWebSocketOutTask::upgrade()
{
auto *channel = (ComplexWebSocketChannel *)this->get_request_channel();
const ParsedURI *uri = channel->get_uri();
std::string request_uri;
auto *http_task = new WFChannelOutTask<HttpRequest>(this->channel,
WFGlobal::get_scheduler(),
@@ -152,10 +154,16 @@ SubTask *ComplexWebSocketOutTask::upgrade()
this->ready = true;
});
if (uri->path && uri->path[0])
request_uri = uri->path;
else
request_uri = "/";
HttpRequest *req = http_task->get_msg();
req->set_method(HttpMethodGet);
req->set_http_version("HTTP/1.1");
req->set_request_uri("/");
req->set_request_uri(request_uri);
req->add_header_pair("Host", channel->get_uri()->host);
req->add_header_pair("Upgrade", "websocket");
req->add_header_pair("Connection", "Upgrade");