folly/wangle/QueuedImmediateExecutor.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
  17 #include "QueuedImmediateExecutor.h"
  18 #include "folly/ThreadLocal.h"
  19 #include <queue>
  20
  21 namespace folly { namespace wangle {
  22
  23 void QueuedImmediateExecutor::add(Action&& callback)
  24 {
  25   thread_local std::queue<Action> q;
  26
  27   if (q.empty()) {
  28     q.push(std::move(callback));
  29     while (!q.empty()) {
  30       q.front()();
  31       q.pop();
  32     }
  33   } else {
  34     q.push(callback);
  35   }
  36 }
  37
  38 }} // namespace