vrfDatabaseConnection.h

Go to the documentation of this file.

/*******************************************************************************
** Copyright (c) 2024 MAK Technologies, Inc.
** All rights reserved.
*******************************************************************************/
#pragma once

#include <vrfMsgTransport/vrfMsgTransportDefines.h>
#include <vrfMsgTransport/vrfDatabaseEvent.h>
#include <atomic>
#include <thread>

class DtVrfDatabaseLoggerInterface;
class DtRwDatabaseLoggerConfig;

class DT_DLL_vrfMsgTransport DtVrfDatabaseConnection
{
public:

   virtual ~DtVrfDatabaseConnection();

   virtual bool initialize(const DtRwDatabaseLoggerConfig& config, 
      DtVrfDatabaseLoggerInterface* dbLogger);

   virtual void run();

protected:

   DtVrfDatabaseConnection();

   virtual bool processConfiguration(const DtRwDatabaseLoggerConfig& config);

   virtual bool connect() = 0;

   virtual bool createDatabase() = 0;

   virtual bool stop();

   virtual bool createTableForEvent(const DtVrfDatabaseEvent& event) = 0;

   virtual bool insertOrIgnoreEntriesForEvents(const DtVrfDatabaseEventVector& events) = 0;

   virtual bool insertOrUpdateEntriesForEvents(const DtVrfDatabaseEventVector& events) = 0;

   virtual bool updateEntryForEvent(const DtVrfDatabaseEvent& event) = 0;

   virtual std::string escapeString(std::string str)
   {
      return str;
   };

private:

   DtVrfDatabaseConnection(const DtVrfDatabaseConnection& other);

   DtVrfDatabaseConnection& operator=(const DtVrfDatabaseConnection& other);

protected:

   DtVrfDatabaseLoggerInterface* myDatabaseLogger;

   double myLogPeriod;
   std::unique_ptr<std::thread> myThread;

   std::vector<bool> myCreatedTables;

   std::atomic<bool> myTimeToStop;
   std::atomic<bool> myStopped;

   class LogToDatabaseTask;
   LogToDatabaseTask* myTask;

};

class DT_DLL_vrfMsgTransport DtVrfNoDatabaseConnection : public DtVrfDatabaseConnection
{
public:

   virtual ~DtVrfNoDatabaseConnection() override {};

   static DtVrfDatabaseConnection* create()
   {
      return new DtVrfNoDatabaseConnection();
   };

   static DtString type()
   {
      return "no-database-connection";
   };

protected:

   DtVrfNoDatabaseConnection() : DtVrfDatabaseConnection() {};

   virtual bool processConfiguration(const DtRwDatabaseLoggerConfig& config) override { return true; };

   virtual bool connect() override { return false; }

   virtual bool createDatabase() override { return false; };

   virtual bool createTableForEvent(const DtVrfDatabaseEvent& event) override { return false; };

   virtual bool insertOrIgnoreEntriesForEvents(const DtVrfDatabaseEventVector& events) override { return false; };

   virtual bool insertOrUpdateEntriesForEvents(const DtVrfDatabaseEventVector& events) override { return false; };

   virtual bool updateEntryForEvent(const DtVrfDatabaseEvent& event) override { return false; };

   
private:

   DtVrfNoDatabaseConnection(const DtVrfNoDatabaseConnection& other);

   DtVrfNoDatabaseConnection& operator=(const DtVrfNoDatabaseConnection& other);
};