proxygen: wangle::ServerWorkerPool Class Reference

#include <ServerBootstrap-inl.h>

Inheritance diagram for wangle::ServerWorkerPool:

Public Member Functions
	ServerWorkerPool (std::shared_ptr< AcceptorFactory > acceptorFactory, folly::IOThreadPoolExecutor *exec, std::shared_ptr< std::vector< std::shared_ptr< folly::AsyncSocketBase >>> sockets, std::shared_ptr< ServerSocketFactory > socketFactory)

template<typename F >
void	forEachWorker (F &&f) const

void	threadStarted (folly::ThreadPoolExecutor::ThreadHandle *) override

void	threadStopped (folly::ThreadPoolExecutor::ThreadHandle *) override

void	threadPreviouslyStarted (folly::ThreadPoolExecutor::ThreadHandle *thread) override

void	threadNotYetStopped (folly::ThreadPoolExecutor::ThreadHandle *thread) override

Public Member Functions inherited from folly::ThreadPoolExecutor::Observer
virtual	~Observer ()=default

Private Types
using	WorkerMap = std::map< folly::ThreadPoolExecutor::ThreadHandle *, std::shared_ptr< Acceptor >>

using	Mutex = folly::SharedMutexReadPriority

Private Attributes
std::shared_ptr< WorkerMap >	workers_

std::shared_ptr< Mutex >	workersMutex_

std::shared_ptr< AcceptorFactory >	acceptorFactory_

folly::IOThreadPoolExecutor *	exec_ {nullptr}

std::shared_ptr< std::vector< std::shared_ptr< folly::AsyncSocketBase > > >	sockets_

std::shared_ptr< ServerSocketFactory >	socketFactory_

Detailed Description

Definition at line 273 of file ServerBootstrap-inl.h.

Member Typedef Documentation

using wangle::ServerWorkerPool::Mutex = folly::SharedMutexReadPriority

private

Definition at line 306 of file ServerBootstrap-inl.h.

using wangle::ServerWorkerPool::WorkerMap = std::map<folly::ThreadPoolExecutor::ThreadHandle*, std::shared_ptr<Acceptor>>

private

Definition at line 305 of file ServerBootstrap-inl.h.

Constructor & Destructor Documentation

wangle::ServerWorkerPool::ServerWorkerPool	(	std::shared_ptr< AcceptorFactory >	acceptorFactory,
		folly::IOThreadPoolExecutor *	exec,
		std::shared_ptr< std::vector< std::shared_ptr< folly::AsyncSocketBase >>>	sockets,
		std::shared_ptr< ServerSocketFactory >	socketFactory
	)

inlineexplicit

Definition at line 275 of file ServerBootstrap-inl.h.

References f.

       : workers_(std::make_shared<WorkerMap>())
       , workersMutex_(std::make_shared<Mutex>())
       , acceptorFactory_(acceptorFactory)
       , exec_(exec)
       , sockets_(sockets)
       , socketFactory_(socketFactory) {
     CHECK(exec);
   }

Member Function Documentation

template<typename F >

void wangle::ServerWorkerPool::forEachWorker ( F && f ) const

Definition at line 318 of file ServerBootstrap-inl.h.

References f.

                                                 {
   Mutex::ReadHolder holder(workersMutex_.get());
   for (const auto& kv : *workers_) {
     f(kv.second.get());
   }
 }

void wangle::ServerWorkerPool::threadNotYetStopped ( folly::ThreadPoolExecutor::ThreadHandle * thread )

inlineoverridevirtual

Reimplemented from folly::ThreadPoolExecutor::Observer.

Definition at line 298 of file ServerBootstrap-inl.h.

References map().

                                                             {
     threadStopped(thread);
   }

void wangle::ServerWorkerPool::threadPreviouslyStarted ( folly::ThreadPoolExecutor::ThreadHandle * thread )

inlineoverridevirtual

Reimplemented from folly::ThreadPoolExecutor::Observer.

Definition at line 294 of file ServerBootstrap-inl.h.

                                                             {
     threadStarted(thread);
   }

void wangle::ServerWorkerPool::threadStarted ( folly::ThreadPoolExecutor::ThreadHandle * h )

overridevirtual

Implements folly::ThreadPoolExecutor::Observer.

Definition at line 24 of file ServerBootstrap.cpp.

References acceptorFactory_, exec_, folly::IOThreadPoolExecutor::getEventBase(), h, folly::netops::socket(), socketFactory_, sockets_, workers_, and workersMutex_.

                                           {
   auto worker = acceptorFactory_->newAcceptor(exec_->getEventBase(h));
   {
     Mutex::WriteHolder holder(workersMutex_.get());
     workers_->insert({h, worker});
   }
 
   for(auto socket : *sockets_) {
     socket->getEventBase()->runImmediatelyOrRunInEventBaseThreadAndWait(
       [this, worker, socket](){
         socketFactory_->addAcceptCB(
           socket, worker.get(), worker->getEventBase());
     });
   }
 }

void wangle::ServerWorkerPool::threadStopped ( folly::ThreadPoolExecutor::ThreadHandle * h )

overridevirtual

Implements folly::ThreadPoolExecutor::Observer.

Definition at line 41 of file ServerBootstrap.cpp.

References folly::gen::move, folly::netops::socket(), socketFactory_, sockets_, workers_, and workersMutex_.

                                           {
   auto worker = [&]() -> std::shared_ptr<Acceptor> {
     Mutex::WriteHolder holder(workersMutex_.get());
     auto workerIt = workers_->find(h);
     if (workerIt == workers_->end()) {
       // The thread handle may not be present in the map if newAcceptor() throws
       // an exception. For example, some acceptors require TLS keys / certs to
       // start and will throw exceptions if those files do not exist.
       return nullptr;
     }
     auto w = std::move(workerIt->second);
     workers_->erase(workerIt);
     return w;
   }();
   if (!worker) {
     return;
   }
 
   for (auto socket : *sockets_) {
     socket->getEventBase()->runImmediatelyOrRunInEventBaseThreadAndWait(
       [&]() {
         socketFactory_->removeAcceptCB(
           socket, worker.get(), nullptr);
     });
   }
 
   auto evb = worker->getEventBase();
 
   evb->runImmediatelyOrRunInEventBaseThreadAndWait(
     [w = std::move(worker)]() mutable {
       w->dropAllConnections();
       w.reset();
     });
 }