CppNet/test/http/http_server.cpp

#include "http_server.h"
#include "http_request.h"
#include "http_context.h"
#include "http_response.h"

#include "common/util/time.h"

using namespace cppnet;

HttpServer::HttpServer() {

}

HttpServer::~HttpServer() {

}

void HttpServer::OnConnection(cppnet::Handle handle, uint32_t err) {
    if (err == CEC_SUCCESS) {
        auto context = new HttpContext();
        handle->SetContext((void*)context);
    }
}

void HttpServer::OnDisConnection(cppnet::Handle handle, uint32_t err) {
    auto context = (HttpContext*)handle->GetContext();
    delete context;
}

void HttpServer::OnMessage(cppnet::Handle handle, cppnet::BufferPtr data, 
                          uint32_t) {
    
    auto context = (HttpContext*)handle->GetContext();

    if (!context->ParseRequest(data, cppnet::UTCTimeMsec())) {
        handle->Write("HTTP/1.1 400 Bad Request\r\n\r\n", sizeof("HTTP/1.1 400 Bad Request\r\n\r\n"));
        handle->Close();
    }

    if (context->IsGotAll()) {
        OnRequest(handle, context->GetRequest());
        context->Reset();
    }
}

void HttpServer::OnMessageSend(cppnet::Handle , uint32_t) {
    // do nothing.
}

void HttpServer::OnRequest(cppnet::Handle handle, const HttpRequest& req) {
    const std::string& connection = req.GetHeader("Connection");
    bool close = connection == "close" ||
      (req.GetVersion() == Http10 && connection != "Keep-Alive");

    HttpResponse response(close);
    _http_call_back(req, response);

    std::string res = response.GetSendBuffer();
    handle->Write(res.c_str(), (uint32_t)res.length());
    if (response.GetCloseConnection()) {
        handle->Close();
    }
}