68 lines
1.2 KiB
C++
68 lines
1.2 KiB
C++
#pragma once
|
|
|
|
#include <vector>
|
|
#include <memory>
|
|
#include <thread>
|
|
#include <atomic>
|
|
#include <cassert>
|
|
|
|
#include "io/network/server.hpp"
|
|
#include "bolt/v1/bolt.hpp"
|
|
|
|
namespace bolt
|
|
{
|
|
|
|
template <class Worker>
|
|
class Server : public io::Server<Server<Worker>>
|
|
{
|
|
public:
|
|
Server(io::Socket&& socket)
|
|
: io::Server<Server<Worker>>(std::forward<io::Socket>(socket)) {}
|
|
|
|
void start(size_t n)
|
|
{
|
|
workers.reserve(n);
|
|
|
|
for(size_t i = 0; i < n; ++i)
|
|
{
|
|
workers.push_back(std::make_shared<Worker>(bolt));
|
|
workers.back()->start(alive);
|
|
}
|
|
|
|
while(alive)
|
|
{
|
|
this->wait_and_process_events();
|
|
}
|
|
}
|
|
|
|
void shutdown()
|
|
{
|
|
alive.store(false);
|
|
|
|
for(auto& worker : workers)
|
|
worker->thread.join();
|
|
}
|
|
|
|
void on_connect()
|
|
{
|
|
assert(idx < workers.size());
|
|
|
|
if(UNLIKELY(!workers[idx]->accept(this->socket)))
|
|
return;
|
|
|
|
idx = idx == workers.size() - 1 ? 0 : idx + 1;
|
|
}
|
|
|
|
void on_wait_timeout() {}
|
|
|
|
private:
|
|
Bolt bolt;
|
|
|
|
std::vector<typename Worker::sptr> workers;
|
|
std::atomic<bool> alive {true};
|
|
|
|
int idx {0};
|
|
};
|
|
|
|
}
|