为ling::shared_mutex添加setResource

include/mutex/shared_mutex.h

@@ -10,7 +10,6 @@
 #include <shared_mutex>
 
 namespace ling {
-
     template<typename T>
     class ResourceReadOnly;
 
@@ -37,6 +36,11 @@ namespace ling {
 
         shared_mutex &operator=(const shared_mutex &) = delete;
 
+        void setResource(const T &&value) {
+            ResourceCompletely<T> lock_completely = lock();
+            this->resource = value;
+        }
+
 
         ResourceReadOnly<CT> lock_shared() {
             mutex.lock_shared();
@@ -63,11 +67,16 @@ namespace ling {
         }
     };
 
-    template<typename T, typename ...Args>
+    template<typename T, typename... Args>
     ling::shared_mutex<T> make_shared_mutex(Args &&... args) {
         return ling::shared_mutex<T>(std::move(T(std::forward<Args>(args)...)));
     }
 
+    template<typename T, typename CT, typename... Args>
+    ling::shared_mutex<T, CT> make_shared_mutex(Args &&... args) {
+        return ling::shared_mutex<T, CT>(std::move(T(std::forward<Args>(args)...)));
+    }
+
     /// 资源的读写代理
     template<typename T>
     class ResourceCompletely {
@@ -162,10 +171,7 @@ namespace ling {
         const T *operator->() const {
             return resource;
         }
-
-
     };
-
 } // ling
 
 #endif //DATASTRUCT_SHARED_MUTEX_H_04

main.cpp

@@ -43,28 +43,25 @@ public:
 
 TEST(资源管理, 1) {
     auto mutex = ling::shared_mutex<mutex_test>(mutex_test());
+    mutex.setResource(mutex_test());
 
-    auto ptr = std::make_shared<mutex_test>();
-    {
+    auto ptr = std::make_shared<mutex_test>(); {
         auto lock = mutex.lock();
         lock->test();
         lock->test_const();
-    }
-    {
+    } {
         auto shared = mutex.lock_shared();
         auto shared2 = mutex.lock_shared();
         shared->test_const();
         shared2->test_const();
-    }
-    {
+    } {
         auto lock = mutex.try_lock();
         ASSERT_TRUE(lock);
         auto lock_2 = mutex.try_lock();
         ASSERT_FALSE(lock_2);
         auto lock_3 = mutex.try_lock_shared();
         ASSERT_FALSE(lock_3);
-    }
-    {
+    } {
         auto lock = mutex.try_lock_shared();
         ASSERT_TRUE(lock);
         auto lock_2 = mutex.try_lock();
@@ -83,7 +80,6 @@ TEST(排序, 堆排序测试) {
     list.reserve(DATA_SIZE);
     std::uniform_int_distribution<> dis(1, DATA_SIZE * 10);
     for (int i = 0; i < DATA_SIZE; i++) {
-
         list.push_back(dis(gen));
     }
     StartStating(start);