Support read and write from blocking and non-blocking pipes and sockets

[folly.git] / folly / LockTraitsBoost.h

diff --git a/folly/LockTraitsBoost.h b/folly/LockTraitsBoost.h
index dd9a17b6b74e69194a1992990b24eccdef2aeb6f..0d86d2356eb4bafba1230b6a5cb46dfd4e2af291 100644 (file)
--- a/folly/LockTraitsBoost.h
+++ b/folly/LockTraitsBoost.h
@@ -22,36 +22,44 @@
  */
 #pragma once
 
+#include <boost/thread.hpp>
 #include <folly/LockTraits.h>
 
 #if FOLLY_LOCK_TRAITS_HAVE_TIMED_MUTEXES
 
-#include <boost/thread.hpp>
 
 namespace folly {
 
+namespace detail {
+/// Convert a std::chrono::duration argument to boost::chrono::duration
+template <class Rep, std::intmax_t Num, std::intmax_t Denom>
+boost::chrono::duration<Rep, boost::ratio<Num, Denom>> toBoostDuration(
+    const std::chrono::duration<Rep, std::ratio<Num, Denom>>& d) {
+  return boost::chrono::duration<Rep, boost::ratio<Num, Denom>>(d.count());
+}
+}
+
 /**
  * LockTraits specialization for boost::shared_mutex
  */
 template <>
 struct LockTraits<boost::shared_mutex>
-    : public folly::detail::LockTraitsSharedBase<boost::shared_mutex> {
+    : public LockTraitsBase<boost::shared_mutex> {
   static constexpr bool is_shared = true;
   static constexpr bool is_timed = true;
 
+  template <class Rep, class Period>
   static bool try_lock_for(
       boost::shared_mutex& mutex,
-      std::chrono::milliseconds timeout) {
-    // Convert the std::chrono argument to boost::chrono
-    return mutex.try_lock_for(boost::chrono::milliseconds(timeout.count()));
+      const std::chrono::duration<Rep, Period>& timeout) {
+    return mutex.try_lock_for(detail::toBoostDuration(timeout));
   }
 
+  template <class Rep, class Period>
   static bool try_lock_shared_for(
       boost::shared_mutex& mutex,
-      std::chrono::milliseconds timeout) {
-    // Convert the std::chrono argument to boost::chrono
-    return mutex.try_lock_shared_for(
-        boost::chrono::milliseconds(timeout.count()));
+      const std::chrono::duration<Rep, Period>& timeout) {
+    return mutex.try_lock_shared_for(detail::toBoostDuration(timeout));
   }
 };
 
@@ -60,15 +68,15 @@ struct LockTraits<boost::shared_mutex>
  */
 template <>
 struct LockTraits<boost::timed_mutex>
-    : public folly::detail::LockTraitsUniqueBase<boost::timed_mutex> {
+    : public LockTraitsBase<boost::timed_mutex> {
   static constexpr bool is_shared = false;
   static constexpr bool is_timed = true;
 
+  template <class Rep, class Period>
   static bool try_lock_for(
       boost::timed_mutex& mutex,
-      std::chrono::milliseconds timeout) {
-    // Convert the std::chrono argument to boost::chrono
-    return mutex.try_lock_for(boost::chrono::milliseconds(timeout.count()));
+      const std::chrono::duration<Rep, Period>& timeout) {
+    return mutex.try_lock_for(detail::toBoostDuration(timeout));
   }
 };
 
@@ -77,15 +85,15 @@ struct LockTraits<boost::timed_mutex>
  */
 template <>
 struct LockTraits<boost::recursive_timed_mutex>
-    : public folly::detail::LockTraitsUniqueBase<boost::recursive_timed_mutex> {
+    : public LockTraitsBase<boost::recursive_timed_mutex> {
   static constexpr bool is_shared = false;
   static constexpr bool is_timed = true;
 
+  template <class Rep, class Period>
   static bool try_lock_for(
       boost::recursive_timed_mutex& mutex,
-      std::chrono::milliseconds timeout) {
-    // Convert the std::chrono argument to boost::chrono
-    return mutex.try_lock_for(boost::chrono::milliseconds(timeout.count()));
+      const std::chrono::duration<Rep, Period>& timeout) {
+    return mutex.try_lock_for(detail::toBoostDuration(timeout));
   }
 };
 } // folly