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 {
namespace 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