folly/experimental/wangle/concurrent/IOExecutor.cpp

   1 /*
   2  * Copyright 2014 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 #include <folly/experimental/wangle/concurrent/IOExecutor.h>
  17
  18 #include <folly/experimental/Singleton.h>
  19 #include <folly/experimental/wangle/concurrent/GlobalExecutor.h>
  20
  21 using folly::Singleton;
  22 using folly::wangle::IOExecutor;
  23
  24 namespace {
  25
  26 Singleton<std::atomic<IOExecutor*>> globalIOExecutorSingleton(
  27     "GlobalIOExecutor",
  28     [](){
  29       return new std::atomic<IOExecutor*>(nullptr);
  30     });
  31
  32 }
  33
  34 namespace folly { namespace wangle {
  35
  36 IOExecutor::~IOExecutor() {
  37   auto thisCopy = this;
  38   try {
  39     getSingleton()->compare_exchange_strong(thisCopy, nullptr);
  40   } catch (const std::runtime_error& e) {
  41     // The global IOExecutor singleton was already destructed so doesn't need to
  42     // be restored. Ignore.
  43   }
  44 }
  45
  46 std::atomic<IOExecutor*>* IOExecutor::getSingleton() {
  47   return Singleton<std::atomic<IOExecutor*>>::get("GlobalIOExecutor");
  48 }
  49
  50 }} // folly::wangle