folly/io/async/ScopedEventBaseThread.cpp

   1 /*
   2  * Copyright 2017 Facebook, Inc.
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at
   7  *
   8  *   http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16
  17 #include <folly/io/async/ScopedEventBaseThread.h>
  18
  19 #include <thread>
  20
  21 #include <folly/Function.h>
  22 #include <folly/Range.h>
  23 #include <folly/ThreadName.h>
  24 #include <folly/io/async/EventBaseManager.h>
  25
  26 using namespace std;
  27
  28 namespace folly {
  29
  30 static void run(
  31     EventBaseManager* ebm,
  32     EventBase* eb,
  33     folly::Baton<>* stop,
  34     const StringPiece& name) {
  35   if (name.size()) {
  36     folly::setThreadName(name);
  37   }
  38
  39   ebm->setEventBase(eb, false);
  40   eb->loopForever();
  41
  42   // must destruct in io thread for on-destruction callbacks
  43   EventBase::StackFunctionLoopCallback cb([=] { ebm->clearEventBase(); });
  44   eb->runOnDestruction(&cb);
  45   // wait until terminateLoopSoon() is complete
  46   stop->wait();
  47   eb->~EventBase();
  48 }
  49
  50 ScopedEventBaseThread::ScopedEventBaseThread()
  51     : ScopedEventBaseThread(nullptr, "") {}
  52
  53 ScopedEventBaseThread::ScopedEventBaseThread(const StringPiece& name)
  54     : ScopedEventBaseThread(nullptr, name) {}
  55
  56 ScopedEventBaseThread::ScopedEventBaseThread(EventBaseManager* ebm)
  57     : ScopedEventBaseThread(ebm, "") {}
  58
  59 ScopedEventBaseThread::ScopedEventBaseThread(
  60     EventBaseManager* ebm,
  61     const StringPiece& name)
  62     : ebm_(ebm ? ebm : EventBaseManager::get()) {
  63   new (&eb_) EventBase();
  64   th_ = thread(run, ebm_, &eb_, &stop_, name);
  65   eb_.waitUntilRunning();
  66 }
  67
  68 ScopedEventBaseThread::~ScopedEventBaseThread() {
  69   eb_.terminateLoopSoon();
  70   stop_.post();
  71   th_.join();
  72 }
  73
  74 }