hyperion.ng/libsrc/jsonserver/JsonClientConnection.cpp

// project includes
#include "JsonClientConnection.h"
#include <api/JsonAPI.h>

// qt inc
#include <QTcpSocket>
#include <QHostAddress>

JsonClientConnection::JsonClientConnection(QTcpSocket *socket, const bool& localConnection)
	: QObject()
	, _socket(socket)
	, _receiveBuffer()
	, _log(Logger::getInstance("JSONCLIENTCONNECTION"))
{
	connect(_socket, &QTcpSocket::disconnected, this, &JsonClientConnection::disconnected);
	connect(_socket, &QTcpSocket::readyRead, this, &JsonClientConnection::readRequest);
	// create a new instance of JsonAPI
	_jsonAPI = new JsonAPI(socket->peerAddress().toString(), _log, localConnection, this);
	// get the callback messages from JsonAPI and send it to the client
	connect(_jsonAPI, &JsonAPI::callbackMessage, this , &JsonClientConnection::sendMessage);
	connect(_jsonAPI, &JsonAPI::forceClose, this , [&](){ _socket->close(); } );

	_jsonAPI->initialize();
}

void JsonClientConnection::readRequest()
{
	_receiveBuffer += _socket->readAll();
	// raw socket data, handling as usual
	int bytes = _receiveBuffer.indexOf('\n') + 1;
	while(bytes > 0)
	{
		// create message string
		QString message(QByteArray(_receiveBuffer.data(), bytes));

		// remove message data from buffer
		_receiveBuffer = _receiveBuffer.mid(bytes);

		// handle message
		_jsonAPI->handleMessage(message);

		// try too look up '\n' again
		bytes = _receiveBuffer.indexOf('\n') + 1;
	}
}

qint64 JsonClientConnection::sendMessage(QJsonObject message)
{
	QJsonDocument writer(message);
	QByteArray data = writer.toJson(QJsonDocument::Compact) + "\n";

	if (!_socket || (_socket->state() != QAbstractSocket::ConnectedState)) return 0;
	return _socket->write(data.data(), data.size());
}

void JsonClientConnection::disconnected(void)
{
	emit connectionClosed();
}