1 files changed, 103 insertions, 0 deletions
diff --git a/app/taskmanager.cpp b/app/taskmanager.cpp
new file mode 100644
index 0000000..1b913d0
--- /dev/null
+++ b/app/taskmanager.cpp
@@ -0,0 +1,103 @@
+#include <QJsonArray>
+#include <QJsonObject>
+#include <QtDebug>
+#include <QString>
+#include <unistd.h>
+#include <iostream>
+#include <QtCore>
+#include "taskmanager.h"
+
+TaskManager::TaskManager(QObject* parent) : QObject(parent) {
+	connect(&m_socket, SIGNAL(textMessageReceived(QString)), this, SLOT(onSocketTextReceived(QString)));
+}
+
+void TaskManager::open(const QUrl &bindingAddress) {
+	m_socket.open(bindingAddress);
+	timer = new QTimer();
+	connect(timer, SIGNAL(timeout()), this, SLOT(callService()));
+	timer->start(3000);
+}
+
+void TaskManager::callService() {
+	QJsonArray msg;
+	msg.append(2); // Call
+	msg.append(QString::number(m_nextCallId));
+	msg.append(QString("taskmanager/get_process_list"));
+	msg.append(QJsonValue());
+	m_nextCallId++;
+
+	QJsonDocument value;
+	value.setArray(msg);
+
+	m_socket.sendTextMessage(value.toJson(QJsonDocument::Compact));
+}
+
+void TaskManager::onSocketTextReceived(QString msg)
+{
+
+	QJsonDocument doc = QJsonDocument::fromJson(msg.toUtf8());
+	QJsonArray arr = doc.array();
+
+	switch(arr[0].toInt()) {
+	case 3: // RetOK
+	case 4: // RetErr
+		ProcessResponse(arr[0].toInt() == 3, arr[2]);
+		break;
+	}
+}
+
+void TaskManager::ProcessResponse(bool r, const QJsonValue& val)
+{
+	std::vector<ProcInfo> procs;
+		
+	if (r) {
+		QJsonObject ret_val = val.toObject();
+		QJsonObject response = ret_val["response"].toObject();
+		QJsonArray processes = response["processes"].toArray();
+
+		for(auto it = processes.constBegin(); it != processes.constEnd(); ++it)
+		{
+			ProcInfo Proc(it->toObject());
+			procs.push_back(Proc);
+		}
+	}
+
+	int flag;
+	if(m_procinfos.empty()){
+		for(auto it = procs.begin(); it != procs.end(); ++it){ // if m_procinfos is empty then this is first call
+			emit addProcess(it->cmd(), it->tid(), it->euid(), it->scpu(), it->ucpu(), it->resident_memory(), it->state()); 
+		}			
+	} else {
+		for(auto it = procs.begin(); it != procs.end(); ++it){						// loop through procs, it = procs element (ProcInfo obj)
+			flag = 0;
+			for(auto it2 = m_procinfos.begin(); it2 != m_procinfos.end(); ++it2){	// loop through m_procinfos, it2 m_procinfos element (ProcInfo obj)
+				// if(*it == *it2){  // if the same ID exists in both vectors
+				if(it->tid() == it2->tid()){
+					if(it->cmd() == it2->cmd()){ // if the name is still the same
+						if(!(it == it2)){ // if the same process has new data
+							emit updateProcess(it->cmd(), it->tid(), it->euid(), it->scpu(), it->ucpu(), it->resident_memory(), it->state());
+							m_procinfos.erase(it2);
+							flag = 1;
+							break;
+						}
+					} else { // process in m_procinfos has died and a new one has its ID
+						qDebug() << "The process ID has been reused for another process";
+						emit updateProcess(it->cmd(), it->tid(), it->euid(), it->scpu(), it->ucpu(), it->resident_memory(), it->state());
+						m_procinfos.erase(it2);
+						flag = 1;
+					}
+				}
+			}
+			if(flag == 0){ // if no ID was found in old vector; that means it's a new process
+				qDebug() << it->cmd() << " process has been added";
+				emit addProcess(it->cmd(), it->tid(), it->euid(), it->scpu(), it->ucpu(), it->resident_memory(), it->state()); 
+			}
+		}
+		for(auto it2 = m_procinfos.begin(); it2 != m_procinfos.end(); ++it2){ // remaining processes in m_procinfos are all dead
+			qDebug() << "Dead processes are removed";
+			qDebug() << it2->cmd();
+			emit removeProcess(it2->tid());
+		}
+	}
+	m_procinfos = procs;
+}