Packages

rocksdb

3.1.2
3.1.2 3.1.1 3.1.0 3.0.0 2.6.2 2.6.1 retired 2.6.0 retired 2.5.0 2.4.1 2.4.0 2.3.0 2.2.0 2.1.0 2.0.0 1.9.0 1.8.0 1.7.0 1.6.0 1.5.1 1.5.0 1.4.0 1.3.2 1.3.1 1.3.0 1.2.0 1.1.1 1.1.0 1.0.0 0.26.2 0.26.1 0.26.0 0.25.0 0.24.0 0.23.3 0.23.2 0.23.1 0.23.0 0.22.0 0.21.0 0.20.1 0.20.0 0.19.0 0.18.0 0.17.0 0.16.0 0.15.0 0.14.0 0.13.1 0.13.0 0.12.0 0.11.0 0.10.0 0.9.1 0.9.0 0.8.2 0.8.1 0.8.0 0.7.1 0.7.0 0.6.4 0.6.3 0.6.2 0.6.1 0.6.0

RocksDB for Erlang

HexDocs

Current section

Files

Jump to
Readme
64 Versions
0 Dependencies
Dependants
Files
Activity
rocksdb deps rocksdb util concurrent_task_limiter_impl.h
Raw

deps/rocksdb/util/concurrent_task_limiter_impl.h 

//  Copyright (c) 2011-present, Facebook, Inc.  All rights reserved.

//  This source code is licensed under both the GPLv2 (found in the

//  COPYING file in the root directory) and Apache 2.0 License

//  (found in the LICENSE.Apache file in the root directory).

//

// Copyright (c) 2011 The LevelDB Authors. All rights reserved.

// Use of this source code is governed by a BSD-style license that can be

// found in the LICENSE file. See the AUTHORS file for names of contributors.



#pragma once

#include <atomic>

#include <memory>



#include "rocksdb/concurrent_task_limiter.h"

#include "rocksdb/env.h"



namespace ROCKSDB_NAMESPACE {



class TaskLimiterToken;



class ConcurrentTaskLimiterImpl : public ConcurrentTaskLimiter {

 public:

  explicit ConcurrentTaskLimiterImpl(const std::string& name,

                                     int32_t max_outstanding_task);

  // No copying allowed

  ConcurrentTaskLimiterImpl(const ConcurrentTaskLimiterImpl&) = delete;

  ConcurrentTaskLimiterImpl& operator=(const ConcurrentTaskLimiterImpl&) =

      delete;



  virtual ~ConcurrentTaskLimiterImpl();



  const std::string& GetName() const override;



  void SetMaxOutstandingTask(int32_t limit) override;



  void ResetMaxOutstandingTask() override;



  int32_t GetOutstandingTask() const override;



  // Request token for adding a new task.

  // If force == true, it requests a token bypassing throttle.

  // Returns nullptr if it got throttled.

  virtual std::unique_ptr<TaskLimiterToken> GetToken(bool force);



 private:

  friend class TaskLimiterToken;



  std::string name_;

  std::atomic<int32_t> max_outstanding_tasks_;

  std::atomic<int32_t> outstanding_tasks_;

};



class TaskLimiterToken {

 public:

  explicit TaskLimiterToken(ConcurrentTaskLimiterImpl* limiter)

      : limiter_(limiter) {}

  ~TaskLimiterToken();



 private:

  ConcurrentTaskLimiterImpl* limiter_;



  // no copying allowed

  TaskLimiterToken(const TaskLimiterToken&) = delete;

  void operator=(const TaskLimiterToken&) = delete;

};



}  // namespace ROCKSDB_NAMESPACE