bdb_writer.cpp

Go to the documentation of this file.

/*************************************************************************
 * Copyright (C) 2011 by Saleh Dindar and the Swarm-NG Development Team  *
 *                                                                       *
 * 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; either version 3 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.,                                       *
 * 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
 ************************************************************************/

#include "../common.hpp"

#include "../types/config.hpp"
#include "../plugin.hpp"

#include "log.hpp"
#include "writer.h"

#include "bdb_database.hpp"

namespace swarm { namespace log {

using namespace gpulog;

class bdb_writer : public writer
{
        // To do this, first we have to include the BDB in the CMake files
        // in process, we should make a gpulog::ilogstream out of the data
        // and read all the gpulog records and put them in the database
        // now the problem is what are we going to use for the key. Maybe we
        // can use a very simple key and later on we can do secondary databases
        // to make the indexes based on time and system id.

    DbEnv* env;
    bdb_database db;
public:
        bdb_writer(const config& cfg)
        :env(bdb_database::createDefaultEnv())
        ,db(env)
    {
                std::string fileName = cfg.require("log_output_db",std::string());
                db.createEmpty(fileName);
        }

        virtual void process(const char *log_data, size_t length) {
                ilogstream stream(log_data,length);
                while(logrecord lr = stream.next()){
            db.put(lr);
                }
                db.flush();
        }

        ~bdb_writer(){
                db.close();
        env->close(0);
        }
};

writer_plugin_initializer< bdb_writer >
        bdb_writer_plugin("bdb", "This is the Berkeley DB writer");

} }