rcpp_framework/modules/drogon/trantor/tests/SendfileTest.cc

#include <trantor/net/TcpServer.h>
#include <trantor/utils/Logger.h>
#include <trantor/net/EventLoopThread.h>
#include <string>
#include <iostream>
#include <thread>
#include <sys/stat.h>
#include <fcntl.h>
#ifndef _WIN32
#include <unistd.h>
#endif

using namespace trantor;
#define USE_IPV6 0
int main(int argc, char *argv[])
{
    if (argc < 2)
    {
        std::cout << "usage:" << argv[0] << " filename" << std::endl;
        return 1;
    }
    std::cout << "filename:" << argv[1] << std::endl;
    struct stat filestat;
    if (stat(argv[1], &filestat) < 0)
    {
        perror("");
        exit(1);
    }
    std::cout << "file len=" << filestat.st_size << std::endl;

    auto fp = fopen(argv[1], "rb");

    if (fp == nullptr)
    {
        perror("");
        exit(1);
    }
    fclose(fp);

    LOG_DEBUG << "test start";

    Logger::setLogLevel(Logger::kTrace);
    EventLoopThread loopThread;
    loopThread.run();

#if USE_IPV6
    InetAddress addr(1207, true, true);
#else
    InetAddress addr(1207);
#endif
    TcpServer server(loopThread.getLoop(), addr, "test");
    server.setRecvMessageCallback(
        [](const TcpConnectionPtr &connectionPtr, MsgBuffer *buffer) {
            // LOG_DEBUG<<"recv callback!";
        });
    int counter = 0;
    server.setConnectionCallback(
        [argv, &counter](const TcpConnectionPtr &connPtr) {
            if (connPtr->connected())
            {
                LOG_DEBUG << "New connection";
                std::thread t([connPtr, argv, &counter]() {
                    for (int i = 0; i < 5; ++i)
                    {
                        connPtr->sendFile(argv[1]);
                        char str[64];
                        ++counter;
                        sprintf(str, "\n%d files sent!\n", counter);
                        connPtr->send(str, strlen(str));
                    }
                });
                t.detach();

                for (int i = 0; i < 3; ++i)
                {
                    connPtr->sendFile(argv[1]);
                    char str[64];
                    ++counter;
                    sprintf(str, "\n%d files sent!\n", counter);
                    connPtr->send(str, strlen(str));
                }
            }
            else if (connPtr->disconnected())
            {
                LOG_DEBUG << "connection disconnected";
            }
        });
    server.setIoLoopNum(3);
    server.start();
    loopThread.wait();
}