RecordRetriever.hpp

Go to the documentation of this file.

#ifndef __RECORD_RETRIEVER_HPP__
#define __RECORD_RETRIEVER_HPP__

#include "gpudb/GPUdb.hpp"
#include "gpudb/Http.hpp"
#include "gpudb/Type.hpp"
#include "gpudb/utils/GPUdbMultiHeadIOUtils.h"


#include <atomic>
#include <map>
#include <string>
#include <vector>

#include <boost/noncopyable.hpp>
#include <boost/shared_ptr.hpp>


namespace gpudb
{

// Forward declaration
class GPUdb;


class RecordRetriever : private boost::noncopyable
{

public:

    RecordRetriever( const gpudb::GPUdb& db, const gpudb::Type& record_type,
                     const std::string& table_name,
                     const WorkerList& worker_list );

    RecordRetriever( const gpudb::GPUdb& db, const gpudb::Type& record_type,
                     const std::string& table_name );

    RecordRetriever( const gpudb::GPUdb& db, const gpudb::Type& record_type,
                     const std::string& table_name,
                     const WorkerList& worker_list,
                     const std::map<std::string, std::string>& retrieval_options );

    RecordRetriever( const gpudb::GPUdb& db, const gpudb::Type& record_type,
                     const std::string& table_name,
                     const std::map<std::string, std::string>& retrieval_options );


    ~RecordRetriever();

    const std::string& getTableName() const { return m_table_name; }

    const gpudb::GPUdb& getGPUdb() const { return m_db; }


    const std::map<std::string, std::string>& getOptions() const { return m_retrieval_options; }


    void setOptions( const std::map<std::string, std::string>& options );


    GetRecordsResponse<gpudb::GenericRecord> getRecordsByKey( const gpudb::GenericRecord& record,
                                                              const std::string& expression = "" );


private:

    typedef std::map<std::string, std::string>      str_to_str_map_t;
    typedef boost::shared_ptr<gpudb::WorkerQueue>   worker_queue_ptr_t;


    RecordRetriever();

    void construct( const gpudb::GPUdb& db,
                    const gpudb::Type& record_type,
                    const std::string& table_name,
                    const WorkerList& worker_list,
                    const std::map<std::string, std::string>& retrieval_options );

    const gpudb::GPUdb&              m_db;
    std::string                      m_table_name;
    gpudb::Type                      m_record_type;
    gpudb::RecordKeyBuilder*         m_shard_key_builder_ptr;
    std::vector<int32_t>             m_routing_table;
    std::vector<worker_queue_ptr_t>  m_worker_queues;
    str_to_str_map_t                 m_retrieval_options;

};  // end class RecordRetriever



} // namespace gpudb



#endif // __RECORD_RETRIEVER_HPP__