mariadb-columnstore-engine/primitives/blockcache/blockrequestprocessor.h

/* Copyright (C) 2014 InfiniDB, Inc.

   This program is free software; you can redistribute it and/or
   modify it under the terms of the GNU General Public License
   as published by the Free Software Foundation; version 2 of
   the License.

   This program is distributed in the hope that it will be useful,
   but WITHOUT ANY WARRANTY; without even the implied warranty of
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
   GNU General Public License for more details.

   You should have received a copy of the GNU General Public License
   along with this program; if not, write to the Free Software
   Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
   MA 02110-1301, USA. */

#ifndef BLOCKREQUESTPROCESSOR_H
#define BLOCKREQUESTPROCESSOR_H

/***************************************************************************
 *
 *   $Id: blockrequestprocessor.h 2055 2013-02-08 19:09:09Z pleblanc $
 *
 *   jrodriguez@calpont.com   *
 *                                                                         *
 ***************************************************************************/

#include <boost/thread.hpp>

#include "blocksize.h"
#include "fileblockrequestqueue.h"
#include "filebuffermgr.h"
#include "iomanager.h"
#include <stdint.h>

/**
	@author Jason Rodriguez <jrodriguez@calpont.com>
*/

namespace dbbc {

typedef std::list<FileBuffer> FileBufferList_t;

/**
 * @brief class to control the populating of the Disk Block Buffer Cache and manage Block requests.
 **/
 
class BlockRequestProcessor {

public:
	
	/**
	 * @brief default ctor
	 **/
    BlockRequestProcessor(uint32_t numBlcks, int thrCount, int blocksPerRead, uint32_t deleteBlocks=0,
		uint32_t blckSz=BLOCK_SIZE);

	/**
	 * @brief default dtor
	 **/
	virtual ~BlockRequestProcessor();

	/**
	 * @brief send a request for disk blocks to the IO manager
	 **/
	int sendRequest(fileRequest& blk) {
		return fBRPRequestQueue.push(blk); }

	/**
	 * @brief verify that the lbid@ver disk block is in the block cache. Send request if it is not
	 **/
	int check(BRM::LBID_t lbid, const BRM::QueryContext &ver, BRM::VER_t txn, bool flg, int compType, bool& wasBlockInCache);

	/**
	 * @brief verify the LBIDRange of disk blocks is in the block cache. Send request if it is not
	 **/
	int check(const BRM::InlineLBIDRange& range, const BRM::QueryContext &ver, const BRM::VER_t txn, const int compType,
		uint32_t& lbidCount);

	/**
	 * @brief retrieve the lbid@ver disk block from the block cache
	 **/
	inline FileBuffer* getBlockPtr(const BRM::LBID_t lbid, const BRM::VER_t ver, bool flg) {
		return fbMgr.findPtr(HashObject_t(lbid, ver, flg)); }

	inline const int read(const BRM::LBID_t& lbid, const BRM::VER_t& ver, FileBuffer& fb) {
		return (fbMgr.find(HashObject_t(lbid, ver, 0), fb) ? 1 : 0); }

	/**
	 * @brief retrieve the lbid@ver disk block from the block cache
	 **/
	inline const int read(const BRM::LBID_t& lbid, const BRM::VER_t &ver, void* bufferPtr) {
		return (fbMgr.find(HashObject_t(lbid, ver, 0), bufferPtr) ? 1 : 0); }
	
	const int getBlock(const BRM::LBID_t& lbid, const BRM::QueryContext &ver, BRM::VER_t txn, int compType,
		void* bufferPtr, bool flg, bool &wasCached, bool *wasVersioned, bool insertIntoCache,
		bool readFromCache);

	int getCachedBlocks(const BRM::LBID_t *lbids, const BRM::VER_t *vers, uint8_t **ptrs,
		bool *wasCached, uint32_t count);

	inline bool exists(BRM::LBID_t lbid, BRM::VER_t ver) {
		return fbMgr.exists(HashObject_t(lbid, ver, 0)); }

	/**
	 * @brief 
	 **/
	void flushCache() {
		fbMgr.flushCache(); }

	/**
	 * @brief 
	 **/
	void flushOne(BRM::LBID_t lbid, BRM::VER_t ver) {
		fbMgr.flushOne(lbid, ver); }

	void flushMany(const LbidAtVer* laVptr, uint32_t cnt) {
		fbMgr.flushMany(laVptr, cnt); }
		
	void flushManyAllversion(const BRM::LBID_t* laVptr, uint32_t cnt) {
		fbMgr.flushManyAllversion(laVptr, cnt); }

	void flushOIDs(const uint32_t *oids, uint32_t count) {
		fbMgr.flushOIDs(oids, count); }

	void flushPartition(const std::vector<BRM::OID_t> &oids, const std::set<BRM::LogicalPartition> &partitions) {
		fbMgr.flushPartition(oids, partitions); }

	void setReportingFrequency(const uint32_t d) {
		fbMgr.setReportingFrequency(d); }

	uint32_t ReportingFrequency() const {return fbMgr.ReportingFrequency();}

	std::ostream& formatLRUList(std::ostream& os) const {
		return fbMgr.formatLRUList(os); }
	
private:
	
	FileBufferMgr fbMgr;
	fileBlockRequestQueue fBRPRequestQueue;
	ioManager fIOMgr;
	boost::mutex check_mutex;

	/**
	 * helper function for public check functions
	 **/
	int check(fileRequest& rqstBlk);

	/**
	 * send stop requests for IOmanager and request Q
	 **/
	void stop();

	std::ofstream fLogFile;
	bool fTrace;

	BRM::DBRM fdbrm;

	// do not implement
	BlockRequestProcessor(const BlockRequestProcessor& brp);
	BlockRequestProcessor& operator=(const BlockRequestProcessor& brp);
	
};

}
#endif
// vim:ts=4 sw=4: