feat(ThreadSafeQueue): add move semantics

EdgeFortress · Sep 21, 2019 · 1ed41ad · 1ed41ad
1 parent 8fb0d99
commit 1ed41ad
Showing 1 changed file with 10 additions and 3 deletions.
diff --git a/SharedLibrary/Sources/Shared/ThreadSafeQueue.hpp b/SharedLibrary/Sources/Shared/ThreadSafeQueue.hpp
@@ -11,24 +11,31 @@ namespace uf {
 		std::queue<T> queue;
 		std::mutex mutex;
 	public:
-		void Push(T t);
+		void Push(const T &t);
+		void Push(T &&t);
 		T Pop();
 		bool Empty();
 		uint32_t GetSize();
 	};
 
 	template<class T>
-	void ThreadSafeQueue<T>::Push(T t) {
+	void ThreadSafeQueue<T>::Push(const T &t) {
 		std::scoped_lock lock(mutex);
 		queue.push(t);
 	}
 
+	template<class T>
+	void ThreadSafeQueue<T>::Push(T &&t) {
+		std::scoped_lock lock(mutex);
+		queue.emplace(std::forward<T>(t));
+	}
+
 	template<class T>
 	T ThreadSafeQueue<T>::Pop() {
 		std::scoped_lock lock(mutex);
 		if (queue.empty())
 			return nullptr;
-		T t = queue.front();
+		T t = std::forward<T>(queue.front());
 		queue.pop();
 		return t;
 	}