Abseil Common Libraries (C++) (grcp 依赖)
https://abseil.io/
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
88 lines
2.5 KiB
88 lines
2.5 KiB
// 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 |
|
// |
|
// https://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. |
|
|
|
#include "absl/base/internal/sysinfo.h" |
|
|
|
#ifndef _WIN32 |
|
#include <sys/types.h> |
|
#include <unistd.h> |
|
#endif |
|
|
|
#include <thread> // NOLINT(build/c++11) |
|
#include <unordered_set> |
|
#include <vector> |
|
|
|
#include "gtest/gtest.h" |
|
#include "absl/synchronization/barrier.h" |
|
#include "absl/synchronization/mutex.h" |
|
|
|
namespace absl { |
|
ABSL_NAMESPACE_BEGIN |
|
namespace base_internal { |
|
namespace { |
|
|
|
TEST(SysinfoTest, NumCPUs) { |
|
EXPECT_NE(NumCPUs(), 0) |
|
<< "NumCPUs() should not have the default value of 0"; |
|
} |
|
|
|
TEST(SysinfoTest, GetTID) { |
|
EXPECT_EQ(GetTID(), GetTID()); // Basic compile and equality test. |
|
#ifdef __native_client__ |
|
// Native Client has a race condition bug that leads to memory |
|
// exaustion when repeatedly creating and joining threads. |
|
// https://bugs.chromium.org/p/nativeclient/issues/detail?id=1027 |
|
return; |
|
#endif |
|
// Test that TIDs are unique to each thread. |
|
// Uses a few loops to exercise implementations that reallocate IDs. |
|
for (int i = 0; i < 10; ++i) { |
|
constexpr int kNumThreads = 10; |
|
Barrier all_threads_done(kNumThreads); |
|
std::vector<std::thread> threads; |
|
|
|
Mutex mutex; |
|
std::unordered_set<pid_t> tids; |
|
|
|
for (int j = 0; j < kNumThreads; ++j) { |
|
threads.push_back(std::thread([&]() { |
|
pid_t id = GetTID(); |
|
{ |
|
MutexLock lock(&mutex); |
|
ASSERT_TRUE(tids.find(id) == tids.end()); |
|
tids.insert(id); |
|
} |
|
// We can't simply join the threads here. The threads need to |
|
// be alive otherwise the TID might have been reallocated to |
|
// another live thread. |
|
all_threads_done.Block(); |
|
})); |
|
} |
|
for (auto& thread : threads) { |
|
thread.join(); |
|
} |
|
} |
|
} |
|
|
|
#ifdef __linux__ |
|
TEST(SysinfoTest, LinuxGetTID) { |
|
// On Linux, for the main thread, GetTID()==getpid() is guaranteed by the API. |
|
EXPECT_EQ(GetTID(), getpid()); |
|
} |
|
#endif |
|
|
|
} // namespace |
|
} // namespace base_internal |
|
ABSL_NAMESPACE_END |
|
} // namespace absl
|
|
|