zap/wrk/cpp/main.cpp

#include <iostream>
#include <fstream>
#include <string>
#include <boost/beast.hpp>
#include <boost/asio/thread_pool.hpp>

namespace beast = boost::beast;
namespace http = beast::http;
namespace net = boost::asio;
using tcp = net::ip::tcp;

std::string read_html_file(const std::string& file_path) {
    std::ifstream file(file_path);
    if (!file) {
        return "File not found: " + file_path;
    }
    std::string content((std::istreambuf_iterator<char>(file)), std::istreambuf_iterator<char>());
    file.close();
    return content;
}

void handle_client(tcp::socket socket, const std::string& msg) {
    try {
        // Construct an HTTP response with the HTML content
        http::response<http::string_body> response;
        response.version(11);
        response.result(http::status::ok);
        response.reason("OK");
        response.set(http::field::server, "C++ Server");
        response.set(http::field::content_type, "text/html");
        response.body() = msg;
        response.prepare_payload();

        // Send the response to the client
        http::write(socket, response);
    } catch (const std::exception& e) {
        std::cerr << "Error: " << e.what() << std::endl;
    }
}

int main() {
    try {
        net::io_context io_context{BOOST_ASIO_CONCURRENCY_HINT_UNSAFE_IO};

        // Create an endpoint to bind to
        tcp::endpoint endpoint(tcp::v4(), 8070);

        // Create and bind the acceptor
        tcp::acceptor acceptor(io_context, endpoint);
        std::cout << "Server listening on port 8070..." << std::endl;

        // static 17-byte string
        std::string msg = "Hello from C++!!!";
        // or
        // Read HTML content from a file (e.g., "index.html")
        // std::string html_content = read_html_file("hello.html");

        // std::cout << "str len: " << (html_content.length() == msg.length()) << std::boolalpha << "\n";

        // Create a thread pool with 4 threads
        net::thread_pool pool(4);

        while (true) {
            // Wait for a client to connect
            tcp::socket socket(io_context);
            acceptor.accept(socket);

            // Post a task to the thread pool to handle the client request
            net::post(pool, [socket = std::move(socket), msg]() mutable {
                handle_client(std::move(socket), msg);
            });
        }

        // The thread pool destructor will ensure that all threads are joined properly.
    } catch (const std::exception& e) {
        std::cerr << "Error: " << e.what() << std::endl;
    }

    return 0;
}