naiveproxy/base/task/sequence_manager/graceful_queue_shutdown_helper.h

// 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_SEQUENCE_MANAGER_GRACEFUL_QUEUE_SHUTDOWN_HELPER_H_
#define BASE_TASK_SEQUENCE_MANAGER_GRACEFUL_QUEUE_SHUTDOWN_HELPER_H_

#include <memory>
#include <vector>

#include "base/macros.h"
#include "base/memory/ref_counted.h"
#include "base/synchronization/lock.h"

namespace base {
namespace sequence_manager {
namespace internal {

class TaskQueueImpl;

// Thread-safe helper to shutdown queues from any thread.
class GracefulQueueShutdownHelper
    : public RefCountedThreadSafe<GracefulQueueShutdownHelper> {
 public:
  GracefulQueueShutdownHelper();

  void GracefullyShutdownTaskQueue(
      std::unique_ptr<internal::TaskQueueImpl> queue);

  void OnSequenceManagerDeleted();

  std::vector<std::unique_ptr<internal::TaskQueueImpl>> TakeQueues();

 private:
  // This class is ref-counted so it controls its own lifetime.
  ~GracefulQueueShutdownHelper();
  friend class RefCountedThreadSafe<GracefulQueueShutdownHelper>;

  Lock lock_;
  bool sequence_manager_deleted_;
  std::vector<std::unique_ptr<internal::TaskQueueImpl>> queues_;

  DISALLOW_COPY_AND_ASSIGN(GracefulQueueShutdownHelper);
};

}  // namespace internal
}  // namespace sequence_manager
}  // namespace base

#endif  // BASE_TASK_SEQUENCE_MANAGER_GRACEFUL_QUEUE_SHUTDOWN_HELPER_H_