Initial commit

Lts/Proxies/TrainingAreaBase/Plugins/DebugJsonDir/directory_cleaner.cpp

@@ -0,0 +1,163 @@
+/**
+  * Helper class to remove outdated telegram files in a directory
+  * $Id$
+  * $HeadURL$
+  * $Author$
+  * Created by Christian Ehrlicher, SLSE, 2016-09-26
+  */
+
+#include "directory_cleaner.h"
+#include <NGLLogger/NGLLoggerBootstrap.h>
+#include <TrainingAreaBase/training_area_telegrams.h>
+#include <QDateTime>
+#include <QDir>
+#include <QDirIterator>
+#include <QTimer>
+#include <QThread>
+#include <QJsonDocument>
+#include <QJsonObject>
+#include <ctime>
+
+namespace TetraRadio
+{
+
+//
+// DirectoryCleaner
+//
+DirectoryCleaner::DirectoryCleaner(const QString &directory, unsigned int maxAgeSec, QObject *parent)
+  : QObject(parent)
+  , m_cleanerThread(new QThread(this))
+  , m_worker(new DirectoryCleanerWorker(directory, maxAgeSec))
+  , m_logger(NGL::NGLLoggerBootstrap::getNGLRootLogger().getNGLLogger("TAB.Plugin.DirectoryCleaner"))
+{
+  QDir().mkpath(directory);
+  doInitialCleanup(directory);
+  m_worker->moveToThread(m_cleanerThread);
+  m_cleanerThread->start();
+  QTimer::singleShot(0, m_worker, &DirectoryCleanerWorker::onTimeout);
+}
+
+DirectoryCleaner::~DirectoryCleaner()
+{
+  m_cleanerThread->quit();
+  m_cleanerThread->wait();
+  delete m_worker;
+}
+
+void DirectoryCleaner::doInitialCleanup(const QString &directory)
+{
+  QDir d(directory);
+  if (!d.exists())
+  {
+    WARNING_LOG(m_logger) << "Directory " << directory
+                          << " does not exist - can not cleanup directory";
+    return;
+  }
+  QDirIterator it(d.absolutePath(), QDir::Files | QDir::NoDotAndDotDot);
+  while (it.hasNext())
+  {
+    it.next();
+    QFile f(it.filePath());
+    if (!f.open(QIODevice::ReadOnly))
+      continue;
+    const QByteArray ba = f.readAll();
+    const QJsonObject obj = QJsonDocument::fromJson(ba).object();
+    const int teleId = obj[QStringLiteral("typeId")].toInt();
+    if (teleShouldBeRemoved(teleId))
+      QFile::remove(it.filePath());
+  }
+}
+
+bool DirectoryCleaner::teleShouldBeRemoved(int teleId)
+{
+  switch (teleId)
+  {
+  case TrainingAreaBase::TetraIcd::WIRKFELD_EINZELGESCHOSS:
+  case TrainingAreaBase::TetraIcd::WIRKFELD_FLAECHENDECKENDES_FEUER:
+  case TrainingAreaBase::TetraIcd::WIRKFELD_FLAECHENDECKENDES_FEUER_STOP:
+  case TrainingAreaBase::TetraIcd::WIRKFELD_FLAECHENDECKENDES_FEUER_FORTSETZEN:
+  case TrainingAreaBase::TetraIcd::WIRKFELD_MINENSPERRE:
+  case TrainingAreaBase::TetraIcd::WIRKFELD_STATUS_MINENSPERRE:
+  case TrainingAreaBase::TetraIcd::WIRKFELD_AENDERUNG_BEZEICHNER:
+  case TrainingAreaBase::TetraIcd::REAKTIVIERUNG_DUS_MINE:
+  case TrainingAreaBase::TetraIcd::WIRKFELD_A_KAMPFMITTEL:
+  case TrainingAreaBase::TetraIcd::WIRKFELD_C_KAMPFMITTEL:
+  case TrainingAreaBase::TetraIcd::DEAKTIVIERUNG_WIRKFELD:
+  case TrainingAreaBase::TetraIcd::DEAKTIVIERUNG_EINZELMINE:
+  case TrainingAreaBase::TetraIcd::FUNKTIONSRAUM:
+  case TrainingAreaBase::TetraIcd::DEAKTIVIEREN_FUNKTIONSRAUM:
+  case TrainingAreaBase::TetraIcd::BETRIEBSZUSTANDSMELDUNG:
+    return true;
+  default:
+    break;
+  }
+  return false;
+}
+
+
+//
+// DirectoryCleanerWorker
+//
+DirectoryCleanerWorker::DirectoryCleanerWorker(const QString &directory, unsigned int maxAgeSec)
+  : m_directory(directory)
+  , m_maxAgeSec(maxAgeSec)
+  , m_bWarnedOnce(false)
+  , m_logger(NGL::NGLLoggerBootstrap::getNGLRootLogger().getNGLLogger("TRA.Plugin.DirectoryCleaner"))
+{}
+
+void DirectoryCleanerWorker::onTimeout()
+{
+  doCleanup();
+  // search every 5sec
+  QTimer::singleShot(5000, Qt::VeryCoarseTimer, this, &DirectoryCleanerWorker::onTimeout);
+}
+
+void DirectoryCleanerWorker::doCleanup()
+{
+  QDir d(m_directory);
+  if (!d.exists())
+  {
+    if (!m_bWarnedOnce)
+      WARNING_LOG(m_logger) << "Directory " << m_directory
+                            << " does not exist - can not cleanup directory";
+    m_bWarnedOnce = true;
+    return;
+  }
+  m_bWarnedOnce = false;
+  const qint64 tRemoveTimeMSec = (time(nullptr) - m_maxAgeSec) * 1000;
+  QDirIterator it(d.absolutePath(), QDir::Files | QDir::NoDotAndDotDot);
+  while (it.hasNext())
+  {
+    const QString filepath(it.next());
+    const QString filename(it.fileName());
+    bool bOk = false;
+    qint64 timestampUSec = filename.toLongLong(&bOk);
+    TRACE_LOG(m_logger) << "File: " << filename
+                        << " timestamp: " << timestampUSec
+                        << " remove time: " << tRemoveTimeMSec * 1000;
+    if (!bOk)
+    {
+      WARNING_LOG(m_logger) << "Found file with name '" << filepath
+                            << " which is not a valid xdr filename"
+                            << " - removing this file when mtime older than "
+                            << m_maxAgeSec << "seconds";
+      qint64 modTime = it.fileInfo().lastModified().toMSecsSinceEpoch();
+      if (modTime < tRemoveTimeMSec)
+        QFile::remove(filepath);
+    }
+    else
+    {
+      const qint64 timestampMSec = timestampUSec / 1000; // usec -> msec
+      if (timestampMSec < tRemoveTimeMSec)
+      {
+        if (!QFile::remove(filepath))
+          WARNING_LOG(m_logger) << "Could not remove " << filepath;
+        else
+          TRACE_LOG(m_logger) << "Removed " << filepath;
+      }
+    }
+
+  }
+}
+
+} // namespace TetraRadio