naiveproxy/base/task/task_scheduler/task_tracker_posix.h
2018-12-09 21:59:24 -05:00

75 lines
2.3 KiB
C++

// Copyright 2016 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef BASE_TASK_TASK_SCHEDULER_TASK_TRACKER_POSIX_H_
#define BASE_TASK_TASK_SCHEDULER_TASK_TRACKER_POSIX_H_
#include <memory>
#include "base/base_export.h"
#include "base/logging.h"
#include "base/macros.h"
#include "base/task/task_scheduler/task_tracker.h"
#include "base/threading/platform_thread.h"
namespace base {
class MessageLoopForIO;
namespace internal {
struct Task;
// A TaskTracker that instantiates a FileDescriptorWatcher in the scope in which
// a task runs. Used on all POSIX platforms except NaCl SFI.
// set_watch_file_descriptor_message_loop() must be called before the
// TaskTracker can run tasks.
class BASE_EXPORT TaskTrackerPosix : public TaskTracker {
public:
TaskTrackerPosix(StringPiece name);
~TaskTrackerPosix() override;
// Sets the MessageLoopForIO with which to setup FileDescriptorWatcher in the
// scope in which tasks run. Must be called before starting to run tasks.
// External synchronization is required between a call to this and a call to
// RunTask().
void set_watch_file_descriptor_message_loop(
MessageLoopForIO* watch_file_descriptor_message_loop) {
watch_file_descriptor_message_loop_ = watch_file_descriptor_message_loop;
}
#if DCHECK_IS_ON()
// TODO(robliao): http://crbug.com/698140. This addresses service thread tasks
// that could run after the task scheduler has shut down. Anything from the
// service thread is exempted from the task scheduler shutdown DCHECKs.
void set_service_thread_handle(
const PlatformThreadHandle& service_thread_handle) {
DCHECK(!service_thread_handle.is_null());
service_thread_handle_ = service_thread_handle;
}
#endif
protected:
// TaskTracker:
void RunOrSkipTask(Task task, Sequence* sequence, bool can_run_task) override;
private:
#if DCHECK_IS_ON()
bool IsPostingBlockShutdownTaskAfterShutdownAllowed() override;
#endif
MessageLoopForIO* watch_file_descriptor_message_loop_ = nullptr;
#if DCHECK_IS_ON()
PlatformThreadHandle service_thread_handle_;
#endif
DISALLOW_COPY_AND_ASSIGN(TaskTrackerPosix);
};
} // namespace internal
} // namespace base
#endif // BASE_TASK_TASK_SCHEDULER_TASK_TRACKER_POSIX_H_