Initial commit

Lts/Services/OcBaseDownload/pdf_download_socket.cpp

@@ -0,0 +1,102 @@
+/**
+  * @RDE_HEADER@
+  * $Id$
+  * $HeadURL$
+  * $Author$
+  * Created by Hannes Fritsche, NLE, 2019-04-30
+  */
+
+#include "pdf_download_socket.h"
+
+#include <nglsettings/nglsettings_connections.h>
+#include <nglsettings/nglsettings_settings.h>
+
+#include <QTimer>
+
+namespace OcBaseDownload
+{
+PdfDownloadSocket::PdfDownloadSocket(const QString &hostName, QObject *parent)
+  : QObject(parent)
+  , m_socket(new QTcpSocket(this))
+  , m_readTimeout(new QTimer(this))
+  , m_sizeToDownload(0)
+  , m_logger(getLogger("PdfDownloadSocket"))
+{
+  DEBUG_LOG(m_logger) << "PdfDownloadSocket::PdfDownloadSocket";
+  connect(m_socket, &QTcpSocket::connected, this, &PdfDownloadSocket::connected);
+  connect(m_socket, &QTcpSocket::readyRead, this, &PdfDownloadSocket::readyRead);
+  unsigned short port = NGL::Connections::port(NGL::Connections::eConnection::FLM_PDF_SERVER);
+  m_socket->connectToHost(hostName, port);
+  m_socket->waitForConnected();
+
+  DEBUG_LOG(m_logger) << "hostname " << hostName << " port " << port;
+
+  m_readTimeout->setInterval(10000);
+  m_readTimeout->setSingleShot(false);
+  connect(m_readTimeout, &QTimer::timeout, this, &PdfDownloadSocket::onReadTimeout);
+}
+
+PdfDownloadSocket::~PdfDownloadSocket()
+{
+  DEBUG_LOG(m_logger) << "~PdfDownloadSocket";
+  m_socket->close();
+}
+
+void PdfDownloadSocket::readyRead()
+{
+  DEBUG_LOG(m_logger) << "reading...";
+  m_readTimeout->stop();
+
+  m_downloadedData.append(m_socket->readAll());
+
+  DEBUG_LOG(m_logger) << m_downloadedData.size() << " of " << m_sizeToDownload << " bytes downloaded";
+
+  if (m_downloadedData.size() >= m_sizeToDownload)
+  {
+    m_sizeToDownload = 0;
+    DEBUG_LOG(m_logger) << "Download finished";
+    emit downloadFinished(m_downloadedData);
+  }
+  else
+    m_readTimeout->start();
+}
+
+void PdfDownloadSocket::connected()
+{
+  DEBUG_LOG(m_logger) << "connected...";
+}
+
+void PdfDownloadSocket::onReadTimeout()
+{
+  //check for filesize
+  m_readTimeout->stop();
+  DEBUG_LOG(m_logger) << "TimerEnd" << m_downloadedData.size() << " expected " << m_sizeToDownload;
+
+  if (m_downloadedData.size() < m_sizeToDownload)
+    WARNING_LOG(m_logger) << "not enough data: got " << m_downloadedData.size() << " expected " << m_sizeToDownload;
+
+  emit downloadFinished(m_downloadedData);
+}
+
+void PdfDownloadSocket::downloadFromHost(const int fileId, const qint64 fileSize)
+{
+  m_sizeToDownload = fileSize;
+  m_downloadedData = QByteArray();
+
+  QByteArray cmd;
+  cmd += QByteArray("PDF_DOWNLOAD ");
+  cmd += QByteArray::number(fileId);
+  cmd += "\n";
+  m_socket->readAll();
+
+  int writeHeaderSize = m_socket->write(cmd);
+  DEBUG_LOG(m_logger) << "writeHeaderSize = " << writeHeaderSize;
+  m_readTimeout->start();
+}
+
+bool PdfDownloadSocket::isConnected() const
+{
+  return m_socket->state() == QAbstractSocket::ConnectedState;
+}
+
+}