experimental/fibers/BoostContextCompatibility.h \
experimental/fibers/EventBaseLoopController.h \
experimental/fibers/EventBaseLoopController-inl.h \
+ experimental/fibers/ExecutionObserver.h \
experimental/fibers/Fiber.h \
experimental/fibers/Fiber-inl.h \
experimental/fibers/FiberManager.h \
--- /dev/null
+/*
+ * Copyright 2015 Facebook, Inc.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+#pragma once
+
+namespace folly { namespace fibers {
+
+/**
+ * Observes the execution of a task.
+ */
+class ExecutionObserver {
+ public:
+ virtual ~ExecutionObserver() { }
+
+ /**
+ * Called when a task is about to start executing.
+ */
+ virtual void starting() noexcept = 0;
+
+ /**
+ * Called just after a task stops executing.
+ */
+ virtual void stopped() noexcept = 0;
+};
+
+}} // namespace folly::fibers
assert(fiber->state_ == Fiber::NOT_STARTED ||
fiber->state_ == Fiber::READY_TO_RUN);
currentFiber_ = fiber;
+ if (observer_) {
+ observer_->starting();
+ }
while (fiber->state_ == Fiber::NOT_STARTED ||
fiber->state_ == Fiber::READY_TO_RUN) {
if (fiber->state_ == Fiber::AWAITING) {
awaitFunc_(*fiber);
awaitFunc_ = nullptr;
+ if (observer_) {
+ observer_->stopped();
+ }
currentFiber_ = nullptr;
} else if (fiber->state_ == Fiber::INVALID) {
assert(fibersActive_ > 0);
fiber->finallyFunc_ = nullptr;
}
// Make sure LocalData is not accessible from its destructor
+ if (observer_) {
+ observer_->stopped();
+ }
currentFiber_ = nullptr;
fiber->localData_.reset();
--fibersAllocated_;
}
} else if (fiber->state_ == Fiber::YIELDED) {
+ if (observer_) {
+ observer_->stopped();
+ }
currentFiber_ = nullptr;
fiber->state_ = Fiber::READY_TO_RUN;
yieldedFibers_.push_back(*fiber);
}
}
+void FiberManager::setObserver(ExecutionObserver* observer) {
+ observer_ = observer;
+}
+
}}
#include <folly/futures/Try.h>
#include <folly/experimental/fibers/BoostContextCompatibility.h>
+#include <folly/experimental/fibers/ExecutionObserver.h>
#include <folly/experimental/fibers/Fiber.h>
#include <folly/experimental/fibers/traits.h>
*/
void yield();
+ /**
+ * Setup fibers execution observation/instrumentation. Fiber locals are
+ * available to observer.
+ *
+ * @param observer Fiber's execution observer.
+ */
+ void setObserver(ExecutionObserver* observer);
+
static FiberManager& getFiberManager();
static FiberManager* getFiberManagerUnsafe();
*/
std::function<void()> immediateFunc_;
+ /**
+ * Fiber's execution observer.
+ */
+ ExecutionObserver* observer_{nullptr};
+
ExceptionCallback exceptionCallback_; /**< task exception callback */
folly::AtomicLinkedList<Fiber, &Fiber::nextRemoteReady_> remoteReadyQueue_;