blob: be903b5855ae39e6c4525121a9889750aa044631 [file] [log] [blame]
// Copyright 2017 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_SCHEDULER_PLATFORM_NATIVE_WORKER_POOL_WIN_H
#define BASE_TASK_SCHEDULER_PLATFORM_NATIVE_WORKER_POOL_WIN_H
#include <windows.h>
#include "base/base_export.h"
#include "base/logging.h"
#include "base/synchronization/atomic_flag.h"
#include "base/task_scheduler/priority_queue.h"
#include "base/task_scheduler/scheduler_worker_pool.h"
namespace base {
namespace internal {
// A SchedulerWorkerPool implementation backed by the Windows Thread Pool API.
//
// Windows Thread Pool API official documentation:
// https://msdn.microsoft.com/en-us/library/windows/desktop/ms686766(v=vs.85).aspx
//
// Blog posts on the Windows Thread Pool API:
// https://msdn.microsoft.com/magazine/hh335066.aspx
// https://msdn.microsoft.com/magazine/hh394144.aspx
// https://msdn.microsoft.com/magazine/hh456398.aspx
// https://msdn.microsoft.com/magazine/hh547107.aspx
// https://msdn.microsoft.com/magazine/hh580731.aspx
class BASE_EXPORT PlatformNativeWorkerPoolWin : public SchedulerWorkerPool {
public:
PlatformNativeWorkerPoolWin(TrackedRef<TaskTracker> task_tracker,
DelayedTaskManager* delayed_task_manager);
// Destroying a PlatformNativeWorkerPoolWin is not allowed in
// production; it is always leaked. In tests, it can only be destroyed after
// JoinForTesting() has returned.
~PlatformNativeWorkerPoolWin() override;
// Starts the worker pool and allows tasks to begin running.
void Start();
// SchedulerWorkerPool:
void JoinForTesting() override;
private:
// Callback that gets run by |pool_|. It runs a task off the next sequence on
// the |priority_queue_|.
static void CALLBACK RunNextSequence(PTP_CALLBACK_INSTANCE,
void* scheduler_worker_pool_windows_impl,
PTP_WORK);
// SchedulerWorkerPool:
void OnCanScheduleSequence(scoped_refptr<Sequence> sequence) override;
// Returns the top Sequence off the |priority_queue_|. Returns nullptr
// if the |priority_queue_| is empty.
scoped_refptr<Sequence> GetWork();
// Thread pool object that |work_| gets executed on.
PTP_POOL pool_ = nullptr;
// Callback environment. |pool_| is associated with |environment_| so that
// work objects using this environment run on |pool_|.
TP_CALLBACK_ENVIRON environment_ = {};
// Work object that executes RunNextSequence. It has a pointer to the current
// |PlatformNativeWorkerPoolWin| and a pointer to |environment_| bound to
// it.
PTP_WORK work_ = nullptr;
// PriorityQueue from which all threads of this worker pool get work.
PriorityQueue priority_queue_;
// Indicates whether the pool has been started yet. This is only accessed
// under |priority_queue_|'s lock.
bool started_ = false;
#if DCHECK_IS_ON()
// Set once JoinForTesting() has returned.
AtomicFlag join_for_testing_returned_;
#endif
DISALLOW_COPY_AND_ASSIGN(PlatformNativeWorkerPoolWin);
};
} // namespace internal
} // namespace base
#endif // BASE_TASK_SCHEDULER_PLATFORM_NATIVE_WORKER_POOL_WIN_H