ydb/yt/cpp/mapreduce/http/ut/simple_server.h

#pragma once

#include <util/generic/ptr.h>

#include <util/stream/input.h>
#include <util/stream/output.h>

#include <util/thread/pool.h>

#include <functional>

class TInetStreamSocket;

// Simple server listens on the specified port and launches
// requestHandler in the separate thread for each incoming connection.
class TSimpleServer
{
public:
    using TRequestHandler = std::function<void(IInputStream* input, IOutputStream* output)>;

public:
    TSimpleServer(int port, TRequestHandler requestHandler);
    ~TSimpleServer();

    void Stop();

    int GetPort() const;
    TString GetAddress() const;

private:
    const int Port_;
    THolder<IThreadPool> ThreadPool_;
    THolder<IThreadFactory::IThread> ListenerThread_;
    THolder<TInetStreamSocket> SendFinishSocket_;
};