database.h

/*
 *   This file is part of wwdedup.
 *
 *   wwdedup 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, either version 3 of the License, or
 *   (at your option) any later version.
 *
 *   wwdedup 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 Foobar.  If not, see <http://www.gnu.org/licenses/>.
*/

#ifndef DATABASE_H
#define DATABASE_H

#include "pre.h"

#include "SqlField.h"
#include "SqlTable.h"
#include "SqlField.h"
#include "SqlDatabase.h"
#include "SqlPreparedStmt.h"
#include "dedup_types.h"

#define REPORT_EXCEPTION(e) printf("Exception %s: %s\n\tin: %s\n", typeid(e).name(), e.what(), __PRETTY_FUNCTION__);

using namespace std;



class DataBase
{
private:
    sql::Database db;
    shared_ptr<sql::PreparedStmt> stGetByPath;
    shared_ptr<sql::PreparedStmt> stGetFileBlock;
    shared_ptr<sql::PreparedStmt> stDeleteFileBlock;
    shared_ptr<sql::PreparedStmt> stDecreaseUseCount;
    shared_ptr<sql::PreparedStmt> stSelectBlockHash;
    shared_ptr<sql::PreparedStmt> stIncreaseUseCount;
    shared_ptr<sql::PreparedStmt> stSetUseCount;
    shared_ptr<sql::PreparedStmt> stUpdateBlockHash;
    shared_ptr<sql::PreparedStmt> stInsertBlockHash;
    shared_ptr<sql::PreparedStmt> stReplaceFileBlock;
    shared_ptr<sql::PreparedStmt> stUpdateFileSize;
    //shared_ptr<sql::PreparedStmt>
    //shared_ptr<sql::PreparedStmt>
    int block_size;
    void prepareStatements();
public:
    DataBase(const std::string* db_url);
    void init();
    ~DataBase();

    boost::intrusive_ptr<file_info> getByPath(const boost::filesystem::path filename);
    int rename(const boost::filesystem::path oldPath, const boost::filesystem::path newPath);
    int create(boost::filesystem::path path, mode_t mode);
    int remove(boost::filesystem::path filename);
    //int truncate(boost::filesystem::path filename, off_t newSize);
    int ftruncate(boost::intrusive_ptr<file_info> fi, off_t newSize);
    int utime(const boost::filesystem::path filename, struct utimbuf *ubuf);
    bool dirEmpty(boost::intrusive_ptr<file_info> dir);
    std::vector<boost::intrusive_ptr<file_info> > readdir(boost::intrusive_ptr<file_info> directory);
    //shared_ptr<storage_block> getStorageBlock(boost::intrusive_ptr<file_info> finfo, off64_t fileBlockNum);
    //shared_ptr<storage_block> allocateStorageBlock(boost::intrusive_ptr<file_info> finfo, off64_t fileBlockNum, shared_ptr<string> hash);
    off64_t releaseStorageBlock(boost::intrusive_ptr<file_info> finfo, off64_t fileBlockNum);
    bool replaceStorageBlock(boost::intrusive_ptr<file_info> finfo, block_info *fblock, shared_ptr<string> hash);
    void loadFileBlock(boost::intrusive_ptr<file_info> finfo, block_info &block, int &use_count);
};

#endif // DATABASE_H