| /* |
| * Copyright 2015 Google Inc. |
| * |
| * Licensed under the Apache License, Version 2.0 (the "License"); |
| * you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| |
| // Author: jmarantz@google.com (Joshua Marantz) |
| |
| #include "pagespeed/kernel/util/mem_lock.h" |
| |
| #include "pagespeed/kernel/base/function.h" |
| #include "pagespeed/kernel/base/timer.h" |
| #include "pagespeed/kernel/util/mem_lock_manager.h" |
| #include "pagespeed/kernel/util/mem_lock_state.h" |
| |
| namespace net_instaweb { |
| |
| MemLock::MemLock(int64 sequence, MemLockState* lock_state) |
| : lock_state_(lock_state), |
| sequence_(sequence) { |
| Clear(); |
| } |
| |
| MemLock::~MemLock() { |
| if (IsPending()) { |
| Deny(); |
| } else if (Held()) { |
| lock_state_->Unlock(); |
| } |
| lock_state_->RemoveLock(this); |
| } |
| |
| void MemLock::Clear() { |
| DCHECK(!lock_state_->IsHeldInOrderedSet(this)); |
| callback_ = NULL; |
| cancel_time_ms_ = 0; |
| steal_ms_ = 0; |
| wakeup_time_ms_ = kNotPending; |
| grant_time_ms_ = kNotHeld; |
| } |
| |
| // Return immediately. Wait wait_ms to take lock, invoke callback with lock |
| // held. On timeout, cancel callback. |
| void MemLock::LockTimedWait(int64 wait_ms, Function* callback) { |
| LockTimedWaitStealOld(wait_ms, kDoNotSteal, callback); |
| } |
| |
| void MemLock::LockTimedWaitStealOld(int64 wait_ms, int64 steal_ms, |
| Function* callback) { |
| if (Held() || IsPending()) { |
| LOG(DFATAL) << "Requesting lock " << name() << " when it's already " |
| << (IsPending() ? "pending" : "held"); |
| callback->CallCancel(); |
| } else if (lock_state_->GrabLock(this)) { |
| grant_time_ms_ = lock_state_->manager()->timer()->NowMs(); |
| callback->CallRun(); |
| } else { |
| DCHECK(!lock_state_->IsHeldInOrderedSet(this)); |
| CHECK(callback_ == NULL); |
| cancel_time_ms_ = lock_state_->manager()->timer()->NowMs() + wait_ms; |
| steal_ms_ = steal_ms; |
| callback_ = callback; |
| lock_state_->ScheduleLock(this); |
| } |
| } |
| |
| void MemLock::Wakeup() { |
| if (ShouldCancelOnWakeup()) { |
| Deny(); |
| } else { |
| CHECK(CanSteal()); |
| lock_state_->StealLock(this); |
| } |
| } |
| |
| GoogleString MemLock::name() const { |
| return lock_state_->name(); |
| } |
| |
| void MemLock::CalculateWakeupTime(int64 held_lock_grant_time_ms) { |
| DCHECK(!lock_state_->IsHeldInOrderedSet(this)); |
| if (!CanSteal() || (held_lock_grant_time_ms == kNotHeld)) { |
| wakeup_time_ms_ = cancel_time_ms_; |
| } else { |
| int64 steal_time_ms = steal_ms_ + held_lock_grant_time_ms; |
| wakeup_time_ms_ = std::min(cancel_time_ms_, steal_time_ms); |
| } |
| } |
| |
| void MemLock::Unlock() { |
| // Locks can be stolen from the holder without notifying the owner, |
| // so it is not considered an error to try to unlock a NamedLock that |
| // is not held. |
| if (Held()) { |
| lock_state_->Unlock(); |
| Clear(); |
| } |
| } |
| |
| void MemLock::Grant(int64 grant_time_ms) { |
| lock_state_->UnscheduleLock(this); |
| Function* callback = callback_; |
| CHECK(!Held()); |
| CHECK(IsPending()); |
| Clear(); |
| grant_time_ms_ = grant_time_ms; |
| CHECK(callback); |
| callback->CallRun(); |
| } |
| |
| void MemLock::Deny() { |
| lock_state_->UnscheduleLock(this); |
| Function* callback = callback_; |
| CHECK(IsPending()); |
| CHECK(!Held()); |
| Clear(); |
| CHECK(callback); |
| callback->CallCancel(); |
| } |
| |
| // Computes a stable ordering for multiple locks with the same |
| // time-based criteria based primarily on the name, and secondarily on |
| // the sequence number. |
| int MemLock::StableCompare(const MemLock* that) const { |
| int cmp = 0; |
| if (lock_state_ == that->lock_state_) { |
| cmp = MemLockState::Compare(sequence_, that->sequence_); |
| } else { |
| cmp = MemLockState::Compare(lock_state_->name(), that->lock_state_->name()); |
| } |
| |
| // Note that if we don't get a strict ordering here between two |
| // different locks we will lose one of them in the map. |
| DCHECK_NE(0, cmp); |
| return cmp; |
| } |
| |
| } // namespace net_instaweb |