[gprpp] absl::Notification polyfill (#31008)
* [gprpp] absl::Notification polyfill * grpc_core:: * Automated change: Fix sanity tests * fix * Automated change: Fix sanity tests Co-authored-by: ctiller <ctiller@users.noreply.github.com>pull/30996/head
parent
868b7d82a7
commit
d9ac89a441
25 changed files with 298 additions and 45 deletions
@ -0,0 +1,67 @@ |
||||
// Copyright 2022 The gRPC 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.
|
||||
|
||||
#ifndef GRPC_CORE_LIB_GPRPP_NOTIFICATION_H |
||||
#define GRPC_CORE_LIB_GPRPP_NOTIFICATION_H |
||||
|
||||
#include <grpc/support/port_platform.h> |
||||
|
||||
#include "absl/time/clock.h" |
||||
#include "absl/time/time.h" |
||||
|
||||
#include "src/core/lib/gprpp/sync.h" |
||||
|
||||
namespace grpc_core { |
||||
|
||||
// Polyfill for absl::Notification until we can use that type.
|
||||
class Notification { |
||||
public: |
||||
void Notify() { |
||||
MutexLock lock(&mu_); |
||||
notified_ = true; |
||||
cv_.SignalAll(); |
||||
} |
||||
|
||||
void WaitForNotification() { |
||||
MutexLock lock(&mu_); |
||||
while (!notified_) { |
||||
cv_.Wait(&mu_); |
||||
} |
||||
} |
||||
|
||||
bool WaitForNotificationWithTimeout(absl::Duration timeout) { |
||||
auto now = absl::Now(); |
||||
auto deadline = now + timeout; |
||||
MutexLock lock(&mu_); |
||||
while (!notified_ && now < deadline) { |
||||
cv_.WaitWithTimeout(&mu_, deadline - now); |
||||
now = absl::Now(); |
||||
} |
||||
return notified_; |
||||
} |
||||
|
||||
bool HasBeenNotified() { |
||||
MutexLock lock(&mu_); |
||||
return notified_; |
||||
} |
||||
|
||||
private: |
||||
Mutex mu_; |
||||
CondVar cv_; |
||||
bool notified_ = false; |
||||
}; |
||||
|
||||
} // namespace grpc_core
|
||||
|
||||
#endif // GRPC_CORE_LIB_GPRPP_NOTIFICATION_H
|
@ -0,0 +1,88 @@ |
||||
// Copyright 2022 The gRPC 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.
|
||||
|
||||
#include "src/core/lib/gprpp/notification.h" |
||||
|
||||
#include <stdlib.h> |
||||
|
||||
#include <memory> |
||||
#include <thread> |
||||
|
||||
#include <gtest/gtest.h> |
||||
|
||||
namespace grpc_core { |
||||
namespace testing { |
||||
namespace { |
||||
|
||||
TEST(Notification, Works) { |
||||
Notification n; |
||||
EXPECT_FALSE(n.HasBeenNotified()); |
||||
n.Notify(); |
||||
EXPECT_TRUE(n.HasBeenNotified()); |
||||
n.WaitForNotification(); |
||||
EXPECT_TRUE(n.HasBeenNotified()); |
||||
} |
||||
|
||||
TEST(Notification, Waits) { |
||||
Notification n; |
||||
auto start = absl::Now(); |
||||
std::thread t([&n] { |
||||
absl::SleepFor(absl::Seconds(5)); |
||||
n.Notify(); |
||||
}); |
||||
n.WaitForNotification(); |
||||
auto end = absl::Now(); |
||||
EXPECT_GE(end - start, absl::Seconds(5)); |
||||
t.join(); |
||||
} |
||||
|
||||
TEST(Notification, WaitsWithTimeout) { |
||||
Notification n; |
||||
auto start = absl::Now(); |
||||
std::thread t([&n] { |
||||
absl::SleepFor(absl::Seconds(5)); |
||||
n.Notify(); |
||||
}); |
||||
EXPECT_TRUE(n.WaitForNotificationWithTimeout(absl::Seconds(10))); |
||||
auto end = absl::Now(); |
||||
EXPECT_GE(end - start, absl::Seconds(5)); |
||||
EXPECT_LE(end - start, absl::Seconds(10)); |
||||
t.join(); |
||||
} |
||||
|
||||
TEST(Notification, WaitWithTimeoutCanFinishEarly) { |
||||
Notification n; |
||||
auto start = absl::Now(); |
||||
std::thread t([&n] { |
||||
absl::SleepFor(absl::Seconds(5)); |
||||
n.Notify(); |
||||
}); |
||||
EXPECT_FALSE(n.WaitForNotificationWithTimeout(absl::Seconds(1))); |
||||
auto end = absl::Now(); |
||||
EXPECT_GE(end - start, absl::Seconds(1)); |
||||
EXPECT_LE(end - start, absl::Seconds(5)); |
||||
n.WaitForNotification(); |
||||
end = absl::Now(); |
||||
EXPECT_GE(end - start, absl::Seconds(5)); |
||||
t.join(); |
||||
} |
||||
|
||||
} // namespace
|
||||
} // namespace testing
|
||||
} // namespace grpc_core
|
||||
|
||||
int main(int argc, char** argv) { |
||||
::testing::InitGoogleTest(&argc, argv); |
||||
return RUN_ALL_TESTS(); |
||||
} |
Loading…
Reference in new issue