sapphire/src/common/Database/DbWorkerPool.h

#ifndef SAPPHIRE_DBWORKERPOOL_H
#define SAPPHIRE_DBWORKERPOOL_H

#include <array>
#include <string>
#include <vector>
#include <ResultSet.h>
#include "Util/LockedWaitQueue.h"
#include "DbConnection.h"

namespace Core::Db
{

  template< typename T >
  class LockedWaitQueue;

  class Operation;

  class PreparedStatement;

  struct ConnectionInfo;

  template< class T >
  class DbWorkerPool
  {
  private:
    enum InternalIndex
    {
      IDX_ASYNC,
      IDX_SYNCH,
      IDX_SIZE
    };

  public:
    DbWorkerPool();

    ~DbWorkerPool();

    void setConnectionInfo( const ConnectionInfo& info, uint8_t asyncThreads, uint8_t synchThreads );

    uint32_t open();

    void close();

    bool prepareStatements();

    inline ConnectionInfo getConnectionInfo() const
    {
      return m_connectionInfo;
    }

    // Async execution
    void execute( const std::string& sql );

    void execute( std::shared_ptr< PreparedStatement > stmt );

    // Sync execution
    void directExecute( const std::string& sql );

    void directExecute( std::shared_ptr< PreparedStatement > stmt );

    std::shared_ptr< Mysql::ResultSet >
    query( const std::string& sql, std::shared_ptr< T > connection = nullptr );

    std::shared_ptr< Mysql::PreparedResultSet > query( std::shared_ptr< PreparedStatement > stmt );

    using PreparedStatementIndex = typename T::Statements;

    std::shared_ptr< PreparedStatement > getPreparedStatement( PreparedStatementIndex index );

    void escapeString( std::string& str );

    void keepAlive();

  private:
    uint32_t openConnections( InternalIndex type, uint8_t numConnections );

    unsigned long escapeString( char* to, const char* from, unsigned long length );

    void enqueue( std::shared_ptr< Operation > op );

    std::shared_ptr< T > getFreeConnection();

    const std::string& getDatabaseName() const;

    std::unique_ptr< Core::LockedWaitQueue< std::shared_ptr< Operation > > > m_queue;
    std::array< std::vector< std::shared_ptr< T > >, IDX_SIZE > m_connections;
    ConnectionInfo m_connectionInfo;
    uint8_t m_asyncThreads;
    uint8_t m_synchThreads;
  };

}

#endif //SAPPHIRE_DBWORKERPOOL_H