111 lines
2.7 KiB
C++
111 lines
2.7 KiB
C++
|
#include "http_server.h"
|
||
|
|
|
||
|
|
#include "request.h"
|
||
|
|
#include "application.h"
|
||
|
|
|
||
|
|
#define LOG_VERBOSE 0
|
||
|
|
|
||
|
|
void HTTPServer::http_callback_handler(Request *request) {
|
||
|
|
Application::handle_request(request);
|
||
|
|
}
|
||
|
|
|
||
|
|
void HTTPServer::httpEnterCallbackDefault(const HTTPParser &httpParser, const HttpSession::Ptr &session) {
|
||
|
|
//(void)httpParser; needed?
|
||
|
|
|
||
|
|
Request *request = RequestPool::get_request();
|
||
|
|
|
||
|
|
request->http_parser = &httpParser;
|
||
|
|
request->session = &session;
|
||
|
|
|
||
|
|
#if LOG_VERBOSE
|
||
|
|
std::cout << "method:" << http_method_str(static_cast<http_method>(httpParser.method())) << std::endl;
|
||
|
|
#endif
|
||
|
|
|
||
|
|
http_callback_handler(request);
|
||
|
|
|
||
|
|
if (httpParser.isKeepAlive()) {
|
||
|
|
request->response->addHeadValue("Connection", "Keep-Alive");
|
||
|
|
|
||
|
|
std::string result = request->response->getResult();
|
||
|
|
|
||
|
|
session->send(result.c_str(), result.size());
|
||
|
|
} else {
|
||
|
|
request->response->addHeadValue("Connection", "Close");
|
||
|
|
|
||
|
|
std::string result = request->response->getResult();
|
||
|
|
|
||
|
|
session->send(result.c_str(), result.size(), [session]() { session->postShutdown(); });
|
||
|
|
}
|
||
|
|
|
||
|
|
RequestPool::return_request(request);
|
||
|
|
}
|
||
|
|
|
||
|
|
void HTTPServer::wsEnterCallbackDefault(const HttpSession::Ptr &httpSession, WebSocketFormat::WebSocketFrameType opcode, const std::string &payload) {
|
||
|
|
|
||
|
|
std::cout << "frame enter of type:" << int(opcode) << std::endl;
|
||
|
|
std::cout << "payload is:" << payload << std::endl;
|
||
|
|
// echo frame
|
||
|
|
auto frame = std::make_shared<std::string>();
|
||
|
|
|
||
|
|
WebSocketFormat::wsFrameBuild(payload.c_str(), payload.size(), *frame, WebSocketFormat::WebSocketFrameType::TEXT_FRAME, true, false);
|
||
|
|
|
||
|
|
httpSession->send(frame);
|
||
|
|
}
|
||
|
|
|
||
|
|
void HTTPServer::configure() {
|
||
|
|
}
|
||
|
|
|
||
|
|
void HTTPServer::initialize() {
|
||
|
|
if (service)
|
||
|
|
return;
|
||
|
|
|
||
|
|
configure();
|
||
|
|
|
||
|
|
service = TcpService::Create();
|
||
|
|
service->startWorkerThread(threads);
|
||
|
|
|
||
|
|
int p_port = port;
|
||
|
|
|
||
|
|
if (listenBuilder)
|
||
|
|
delete listenBuilder;
|
||
|
|
|
||
|
|
listenBuilder = new wrapper::HttpListenerBuilder();
|
||
|
|
listenBuilder->configureService(service);
|
||
|
|
|
||
|
|
listenBuilder->configureSocketOptions({
|
||
|
|
[](TcpSocket &socket) {
|
||
|
|
socket.setNodelay();
|
||
|
|
},
|
||
|
|
});
|
||
|
|
|
||
|
|
listenBuilder->configureConnectionOptions({ AddSocketOption::WithMaxRecvBufferSize(1024) });
|
||
|
|
|
||
|
|
listenBuilder->configureListen([p_port](wrapper::BuildListenConfig builder) {
|
||
|
|
builder.setAddr(false, "0.0.0.0", p_port);
|
||
|
|
});
|
||
|
|
|
||
|
|
listenBuilder->configureEnterCallback([](const HttpSession::Ptr &httpSession, HttpSessionHandlers &handlers) {
|
||
|
|
handlers.setHttpCallback(HTTPServer::httpEnterCallbackDefault);
|
||
|
|
handlers.setWSCallback(HTTPServer::wsEnterCallbackDefault);
|
||
|
|
});
|
||
|
|
|
||
|
|
listenBuilder->asyncRun();
|
||
|
|
}
|
||
|
|
|
||
|
|
void HTTPServer::main_loop() {
|
||
|
|
while (true) {
|
||
|
|
std::this_thread::sleep_for(std::chrono::seconds(1));
|
||
|
|
if (brynet::base::app_kbhit()) {
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
HTTPServer::HTTPServer() {
|
||
|
|
port = 80;
|
||
|
|
threads = 4;
|
||
|
|
}
|
||
|
|
|
||
|
|
HTTPServer::~HTTPServer() {
|
||
|
|
delete listenBuilder;
|
||
|
|
}
|