38 lines
1.2 KiB
C++
38 lines
1.2 KiB
C++
|
// Copyright (c) 2013, Facebook, Inc. All rights reserved.
|
||
|
|
// This source code is licensed under the BSD-style license found in the
|
||
|
|
// LICENSE file in the root directory of this source tree. An additional grant
|
||
|
|
// of patent rights can be found in the PATENTS file in the same directory.
|
||
|
|
|
||
|
|
#include "db/write_controller.h"
|
||
|
|
|
||
|
|
#include <cassert>
|
||
|
|
|
||
|
|
namespace rocksdb {
|
||
|
|
|
||
|
|
std::unique_ptr<WriteControllerToken> WriteController::GetStopToken() {
|
||
|
|
++total_stopped_;
|
||
|
|
return std::unique_ptr<WriteControllerToken>(new StopWriteToken(this));
|
||
|
|
}
|
||
|
|
|
||
|
|
std::unique_ptr<WriteControllerToken> WriteController::GetDelayToken(
|
||
|
|
uint64_t delay_us) {
|
||
|
|
total_delay_us_ += delay_us;
|
||
|
|
return std::unique_ptr<WriteControllerToken>(
|
||
|
|
new DelayWriteToken(this, delay_us));
|
||
|
|
}
|
||
|
|
|
||
|
|
bool WriteController::IsStopped() const { return total_stopped_ > 0; }
|
||
|
|
uint64_t WriteController::GetDelay() const { return total_delay_us_; }
|
||
|
|
|
||
|
|
StopWriteToken::~StopWriteToken() {
|
||
|
|
assert(controller_->total_stopped_ >= 1);
|
||
|
|
--controller_->total_stopped_;
|
||
|
|
}
|
||
|
|
|
||
|
|
DelayWriteToken::~DelayWriteToken() {
|
||
|
|
assert(controller_->total_delay_us_ >= delay_us_);
|
||
|
|
controller_->total_delay_us_ -= delay_us_;
|
||
|
|
}
|
||
|
|
|
||
|
|
} // namespace rocksdb
|