5b53540166
- c4b7a4e517c9404932c45f2f9f92eb7dc694e45d Internal change by Abseil Team <absl-team@google.com> - 76c78ed9385f65d881511645446e0bb8ababf6ec Add missing ABSL_PREDICT_FALSE to one of FixedArray::at()... by Abseil Team <absl-team@google.com> - 1204fb1c46f007dd9dfb7d9abf3e96c58835d193 Internal change. by Greg Falcon <gfalcon@google.com> - f1f47c98a026bc5e425ae83ff4a2eb391bbd3d9b Add internal-only functionality to examine the stack, to ... by Derek Mauro <dmauro@google.com> - 30d63097cd268d912f917526f6511005580465c4 fix typo by Abseil Team <absl-team@google.com> - 942d7efa6cf54cd248ca57dcaf3c245188b52a76 Remove unnecessary semicolons from comment examples. by Abseil Team <absl-team@google.com> - 7db0669cf23a06d934d3ed8c76aee4e4e23b7e04 Remove malloc_hook and malloc_extension from our internal... by Greg Falcon <gfalcon@google.com> - 0190f1063d101b1ded355019df2e1d325931f6c7 Make the maximum length of a string view equal difference... by Abseil Team <absl-team@google.com> - c8ae37cbce29449b02115a0ebd435ddc3d7ab062 Add namespace qualification. by Shaindel Schwartz <shaindel@google.com> - ff70afe2e6e3dd39f51ce9829e3e1f18231bf4d7 Fix internal/direct_mmap.h for non-linux builds. by Greg Falcon <gfalcon@google.com> GitOrigin-RevId: ed0ba496fe01eb8edfa86beade8a37768e7c12ef Change-Id: I7595ee3480d1d6724fd3797c15ba9d9be0d17e62
99 lines
3.5 KiB
C++
99 lines
3.5 KiB
C++
// Copyright 2017 The Abseil Authors.
|
|
//
|
|
// 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.
|
|
|
|
// This file is a no-op if the required LowLevelAlloc support is missing.
|
|
#include "absl/base/internal/low_level_alloc.h"
|
|
#ifndef ABSL_LOW_LEVEL_ALLOC_MISSING
|
|
|
|
#include "absl/synchronization/internal/per_thread_sem.h"
|
|
|
|
#include <atomic>
|
|
|
|
#include "absl/base/attributes.h"
|
|
#include "absl/base/internal/thread_identity.h"
|
|
#include "absl/synchronization/internal/waiter.h"
|
|
|
|
namespace absl {
|
|
namespace synchronization_internal {
|
|
|
|
void PerThreadSem::SetThreadBlockedCounter(std::atomic<int> *counter) {
|
|
base_internal::ThreadIdentity *identity;
|
|
identity = GetOrCreateCurrentThreadIdentity();
|
|
identity->blocked_count_ptr = counter;
|
|
}
|
|
|
|
std::atomic<int> *PerThreadSem::GetThreadBlockedCounter() {
|
|
base_internal::ThreadIdentity *identity;
|
|
identity = GetOrCreateCurrentThreadIdentity();
|
|
return identity->blocked_count_ptr;
|
|
}
|
|
|
|
void PerThreadSem::Init(base_internal::ThreadIdentity *identity) {
|
|
Waiter::GetWaiter(identity)->Init();
|
|
identity->ticker.store(0, std::memory_order_relaxed);
|
|
identity->wait_start.store(0, std::memory_order_relaxed);
|
|
identity->is_idle.store(false, std::memory_order_relaxed);
|
|
}
|
|
|
|
void PerThreadSem::Tick(base_internal::ThreadIdentity *identity) {
|
|
const int ticker =
|
|
identity->ticker.fetch_add(1, std::memory_order_relaxed) + 1;
|
|
const int wait_start = identity->wait_start.load(std::memory_order_relaxed);
|
|
const bool is_idle = identity->is_idle.load(std::memory_order_relaxed);
|
|
if (wait_start && (ticker - wait_start > Waiter::kIdlePeriods) && !is_idle) {
|
|
// Wakeup the waiting thread since it is time for it to become idle.
|
|
Waiter::GetWaiter(identity)->Poke();
|
|
}
|
|
}
|
|
|
|
} // namespace synchronization_internal
|
|
} // namespace absl
|
|
|
|
extern "C" {
|
|
|
|
ABSL_ATTRIBUTE_WEAK void AbslInternalPerThreadSemPost(
|
|
absl::base_internal::ThreadIdentity *identity) {
|
|
absl::synchronization_internal::Waiter::GetWaiter(identity)->Post();
|
|
}
|
|
|
|
ABSL_ATTRIBUTE_WEAK bool AbslInternalPerThreadSemWait(
|
|
absl::synchronization_internal::KernelTimeout t) {
|
|
bool timeout = false;
|
|
absl::base_internal::ThreadIdentity *identity;
|
|
identity = absl::synchronization_internal::GetOrCreateCurrentThreadIdentity();
|
|
|
|
// Ensure wait_start != 0.
|
|
int ticker = identity->ticker.load(std::memory_order_relaxed);
|
|
identity->wait_start.store(ticker ? ticker : 1, std::memory_order_relaxed);
|
|
identity->is_idle.store(false, std::memory_order_relaxed);
|
|
|
|
if (identity->blocked_count_ptr != nullptr) {
|
|
// Increment count of threads blocked in a given thread pool.
|
|
identity->blocked_count_ptr->fetch_add(1, std::memory_order_relaxed);
|
|
}
|
|
|
|
timeout =
|
|
!absl::synchronization_internal::Waiter::GetWaiter(identity)->Wait(t);
|
|
|
|
if (identity->blocked_count_ptr != nullptr) {
|
|
identity->blocked_count_ptr->fetch_sub(1, std::memory_order_relaxed);
|
|
}
|
|
identity->is_idle.store(false, std::memory_order_relaxed);
|
|
identity->wait_start.store(0, std::memory_order_relaxed);
|
|
return !timeout;
|
|
}
|
|
|
|
} // extern "C"
|
|
|
|
#endif // ABSL_LOW_LEVEL_ALLOC_MISSING
|