mirror of
https://gitlab.com/niansa/libcrosscoro.git
synced 2025-03-06 20:53:32 +01:00
80fea9c49a
* io_scheduler uses thread pool to schedule work fixes #41 * use task_container in bench tcp server test * adjust benchmark for github actions CI * fix io_scheduler tests cross thread memory boundaries * more memory barriers * sprinkle some shutdowns in there * update readme
65 lines
2.2 KiB
C++
65 lines
2.2 KiB
C++
#pragma once
|
|
|
|
#include "coro/io_scheduler.hpp"
|
|
#include "coro/net/ip_address.hpp"
|
|
#include "coro/net/socket.hpp"
|
|
#include "coro/net/tcp_client.hpp"
|
|
#include "coro/task.hpp"
|
|
|
|
#include <fcntl.h>
|
|
#include <functional>
|
|
#include <sys/socket.h>
|
|
|
|
namespace coro::net
|
|
{
|
|
class tcp_server
|
|
{
|
|
public:
|
|
struct options
|
|
{
|
|
/// The ip address for the tcp server to bind and listen on.
|
|
net::ip_address address{net::ip_address::from_string("0.0.0.0")};
|
|
/// The port for the tcp server to bind and listen on.
|
|
uint16_t port{8080};
|
|
/// The kernel backlog of connections to buffer.
|
|
int32_t backlog{128};
|
|
};
|
|
|
|
tcp_server(
|
|
io_scheduler& scheduler,
|
|
options opts = options{.address = net::ip_address::from_string("0.0.0.0"), .port = 8080, .backlog = 128});
|
|
|
|
tcp_server(const tcp_server&) = delete;
|
|
tcp_server(tcp_server&&) = delete;
|
|
auto operator=(const tcp_server&) -> tcp_server& = delete;
|
|
auto operator=(tcp_server&&) -> tcp_server& = delete;
|
|
~tcp_server() = default;
|
|
|
|
/**
|
|
* Polls for new incoming tcp connections.
|
|
* @param timeout How long to wait for a new connection before timing out, zero waits indefinitely.
|
|
* @return The result of the poll, 'event' means the poll was successful and there is at least 1
|
|
* connection ready to be accepted.
|
|
*/
|
|
auto poll(std::chrono::milliseconds timeout = std::chrono::milliseconds{0}) -> coro::task<coro::poll_status>
|
|
{
|
|
return m_io_scheduler.poll(m_accept_socket, coro::poll_op::read, timeout);
|
|
}
|
|
|
|
/**
|
|
* Accepts an incoming tcp client connection. On failure the tcp clients socket will be set to
|
|
* and invalid state, use the socket.is_value() to verify the client was correctly accepted.
|
|
* @return The newly connected tcp client connection.
|
|
*/
|
|
auto accept() -> coro::net::tcp_client;
|
|
|
|
private:
|
|
/// The io scheduler for awaiting new connections.
|
|
io_scheduler& m_io_scheduler;
|
|
/// The bind and listen options for this server.
|
|
options m_options;
|
|
/// The socket for accepting new tcp connections on.
|
|
net::socket m_accept_socket{-1};
|
|
};
|
|
|
|
} // namespace coro::net
|