Use std::unique_ptr to store DHTMessages instead of std::shared_ptr

1 files changed, 4 insertions, 6 deletions

diff --git a/src/DHTMessageEntry.h b/src/DHTMessageEntry.h
index 2698b5ae..c2794e74 100644
--- a/src/DHTMessageEntry.h
+++ b/src/DHTMessageEntry.h
@@ -47,15 +47,13 @@ class DHTMessage;
 class DHTMessageCallback;
 
 struct DHTMessageEntry {
-  std::shared_ptr<DHTMessage> message;
+  std::unique_ptr<DHTMessage> message;
   time_t timeout;
-  std::shared_ptr<DHTMessageCallback> callback;
+  std::unique_ptr<DHTMessageCallback> callback;
 
-  DHTMessageEntry(const std::shared_ptr<DHTMessage>& message,
+  DHTMessageEntry(std::unique_ptr<DHTMessage> message,
                   time_t timeout,
-                  const std::shared_ptr<DHTMessageCallback>& callback);
-
-  ~DHTMessageEntry();
+                  std::unique_ptr<DHTMessageCallback> callback);
 };
 
 } // namespace aria2