Update & BugFix Add GatewayTester

Signed-off-by: tomoaki <tomoaki@tomy-tech.com>

MQTTSNGateway/GatewayTester/src/LTaskManager.cpp

@@ -0,0 +1,177 @@
+/**************************************************************************************
+ * Copyright (c) 2016, Tomoaki Yamaguchi
+ *
+ * All rights reserved. This program and the accompanying materials
+ * are made available under the terms of the Eclipse Public License v1.0
+ * and Eclipse Distribution License v1.0 which accompany this distribution.
+ *
+ * The Eclipse Public License is available at
+ *    http://www.eclipse.org/legal/epl-v10.html
+ * and the Eclipse Distribution License is available at
+ *   http://www.eclipse.org/org/documents/edl-v10.php.
+ *
+ * Contributors:
+ *    Tomoaki Yamaguchi - initial API and implementation and/or initial documentation
+ **************************************************************************************/
+
+
+#include <stdio.h>
+#include <string.h>
+
+#include "LMqttsnClientApp.h"
+#include "LSubscribeManager.h"
+#include "LTimer.h"
+#include "LMqttsnClient.h"
+#include "LTaskManager.h"
+#include "LScreen.h"
+
+using namespace std;
+using namespace linuxAsyncClient;
+
+extern LMqttsnClient* theClient;
+extern LScreen* theScreen;
+extern bool theClientMode;
+/*=====================================
+           TaskManager
+ ======================================*/
+LTaskManager::LTaskManager(void){
+	_tasks = 0;
+	_tests = 0;
+	_index = 0;
+}
+
+LTaskManager::~LTaskManager(void){
+    
+}
+
+void LTaskManager::add(TaskList* task){
+    _tasks = task;
+}
+
+void LTaskManager::add(TestList* test){
+    _tests = test;
+}
+
+void LTaskManager::run(void){
+	int i = 0;
+	char c = 0;
+	bool cancelFlg = false;
+
+	if ( !theClientMode )
+	{
+			theClient->getGwProxy()->getMessage();
+
+			for (i = 0; _tests[i].testTask > 0; i++)
+			{
+				PROMPT("Execute \"%s\" ? ( y/n ) : ", _tests[i].testLabel);
+				while (true)
+				{
+					if (CHECKKEYIN(&c))
+					{
+						if ( toupper(c) == 'N' )
+						{
+
+							ASSERT("\033[0m\033[0;32m\n**** %s is canceled ****\033[0m\033[0;37m\n\n", _tests[i].testLabel);
+							theScreen->prompt("");
+							cancelFlg = true;
+							break;
+						}
+						else if ( toupper(c) == 'Y' )
+						{
+							ASSERT("\033[0m\033[0;32m\n\n**** %s start ****\033[0m\033[0;37m\n", _tests[i].testLabel);
+							theScreen->prompt("");
+							(_tests[i].testTask)();
+							cancelFlg = false;
+							break;
+						}
+					}
+					else
+					{
+						theClient->getGwProxy()->getMessage();
+					}
+				}
+
+				while ( true )
+				{
+					do
+					{
+						theClient->getGwProxy()->getMessage();
+					}
+					while(theClient->getPublishManager()->isMaxFlight() ||
+						   !theClient->getSubscribeManager()->isDone() ||
+						   !theClient->getRegisterManager()->isDone());
+
+					if (theClient->getPublishManager()->isDone())
+					{
+						break;
+					}
+				}
+				if ( !cancelFlg )
+				{
+					ASSERT("\033[0m\033[0;32m\n**** %s complete ****\033[0m\033[0;37m\n\n", _tests[i].testLabel);
+				}
+			}
+			ASSERT("\033[0m\033[0;32m\n\n#########  All tests complete!  ###########\033[0m\033[0;37m\n\n");
+	}
+	else
+	{
+		while (true)
+		{
+			theClient->getGwProxy()->getMessage();
+			for (_index = 0; _tasks[_index].callback > 0; _index++)
+			{
+				if ((_tasks[_index].prevTime + _tasks[_index].interval <= time(NULL)) &&
+					 _tasks[_index].count == 0)
+				{
+					_tasks[_index].prevTime = time(NULL);
+					(_tasks[_index].callback)();
+				}
+			}
+
+			do
+			{
+				theClient->getGwProxy()->getMessage();
+			}
+			while(theClient->getPublishManager()->isMaxFlight() ||
+				   !theClient->getSubscribeManager()->isDone() ||
+				   !theClient->getRegisterManager()->isDone());
+
+			if (theClient->getPublishManager()->isDone())
+			{
+				break;
+			}
+		}
+	}
+}
+
+uint8_t LTaskManager::getIndex(void){
+	return _index;
+}
+
+void LTaskManager::done(uint8_t index){
+	if (_tasks )
+	{
+		if (_tasks[index].count > 0)
+		{
+			_tasks[index].count--;
+		}
+	}
+	if (_tests )
+		{
+			if (_tests[index].count > 0)
+			{
+				_tests[index].count--;
+			}
+		}
+}
+
+void LTaskManager::suspend(uint8_t index){
+	if ( _tasks )
+	{
+		_tasks[index].count++;
+	}
+	if ( _tests )
+	{
+		_tests[index].count++;
+	}
+}