Wait group added

Author: vishnuvijayanklm

README.md

@@ -153,3 +153,13 @@ x << c; //Receving from channel
 
 c.close(); //Closing the channel
 ```
+
+### WaitGroup
+```cpp
+Core::WaitGroup wg; // Creating a wait group
+wg.add() // adding to waitgroup adding 1 to counter
+wg.add(10) // adding to waitgroup adding 10 to counter
+
+wg.done() // wait group is done and decrease counter by 1
+wg.wait() // wait for wait group to complete , ie counter becomes 0
+```

core/include/WaitGroup.h

@@ -0,0 +1,51 @@
+#ifndef WAITGROUP_H
+#define WAITGROUP_H
+
+using namespace std;
+namespace Core
+{
+	class WaitGroup
+	{
+		mutex mMtx;
+		mutex mCondMtx;
+		condition_variable mCondVar;
+		unsigned int mCounter; 
+	public:
+		WaitGroup()
+		{
+			mCounter = 0x00;
+		}
+		~WaitGroup()
+		{
+		}
+
+		void add(unsigned int delta = 1)
+		{
+			lock_guard<mutex> lock(this->mMtx);
+			mCounter += delta;
+		}
+
+		void done()
+		{
+			lock_guard<mutex> lock(this->mMtx);
+			mCounter--;
+			unique_lock<mutex> lockCondition(this->mCondMtx);
+			this->mCondVar.notify_all();
+		}
+
+		void wait()
+		{
+
+			unique_lock<mutex> lock(this->mCondMtx);
+			this->mCondVar.wait(lock, [&]{ return (!this->mCounter); });
+
+		}
+
+		unsigned int count()
+		{
+			return mCounter;
+		}
+	};
+}
+
+#endif

main.cpp

@@ -6,6 +6,7 @@
 #include <timer/include/TimerTask.h>
 #include <ipc/include/MessageQueue.h>
 #include <core/include/Channel.h>
+#include <core/include/WaitGroup.h>
 
 using namespace std;
 //unique_ptr<Core::Application> pApplication(new Core::Application());
@@ -90,18 +91,22 @@ void fn()
 
 int main()
 {
-
-	for (int i = 0; i < 1000000; i++)
+	Core::WaitGroup wg;
+	for (int i = 0; i < 1000; i++)
 	{
+		wg.add();
 		c << rand() % 100;
 		async([&]()
 			  {
 				  int x ;
 				  x << c;
 				  cout <<"Got from channel "<< x << " Channel size "<< c.size()<<endl;
+				  wg.done();
 			  });
-		usleep(1000);
+		//usleep(1000);
 	}
+	cout<<"Wait counter "<<wg.count()<<endl;
+	wg.wait();
 	/*
 	atexit(onExit);
 

util/include/Logger.h

@@ -9,6 +9,7 @@
 #include <inttypes.h>
 #include <dirent.h>
 #include <fstream>
+#include <memory>
 
 #define LOG_CR      1
 #define LOG_ER      2