config_store.hpp

Go to the documentation of this file.

/*
 *  Copyright 2009 DreamLab Onet.pl Sp. z o.o.
 *  Licensed under the Apache License, Version 2.0 (the "License");
 *  you may not use this file except in compliance with the License.
 *  You may obtain a copy of the License at
 *
 *  http://www.apache.org/licenses/LICENSE-2.0
 *
 *   Unless required by applicable law or agreed to in writing, software
 *   distributed under the License is distributed on an "AS IS" BASIS,
 *   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *   See the License for the specific language governing permissions and
 *   limitations under the License.
 *
 *   Authors:
 *   Michal Szklarzewicz <michal(x)szklarzewicz(y)gmail(x)com>
 *   Grzegorz Jablonski  <jablonski(x)grzegorz(y)gmail(x)com>
 *   (use dot instead of x, and @ instead of y)
 */

#ifndef __CONFIG_STORE_HPP__
#define __CONFIG_STORE_HPP__
#include "types.hpp"
#include "LocalDB.hpp"
#include "hashtable.hpp"
#include "dst_config.hpp"
#include <db_cxx.h>
#include <string.h>

struct dest_config;
struct t_uid_const
{
        uint32_t        length; 
        char            data[255];  
    t_uid_const(const char * dest){
                length = strlen(dest);
                memcpy(data, dest, length);
        }
        t_uid_const(const t_uid &ref){
        length = ref.length;
        memcpy(data, ref.data, length);
    }
};

class configStore {
        t_uid dest_;
        dest_config* conf_;
        LocalDB &localDB_;
        Db db_;
        std::string dbFileName_;
        u_int32_t cFlags_;
        //close nicely berkeley db
        void close();
        char* size_;
    int size_len_;
        char* traffic_;
    int traffic_len_;

public:
        configStore(dest_config* conf, LocalDB &ref, t_uid dest, std::string &path, std::string &dbName);
        ~configStore() {
                close();
        }
        inline Db &getDb() {
                return db_;
        }
        void flush() { 
                db_.sync(0);
        }
        void save();
        void save_diff();
        void init();
        void save_size();
    long long  read_size();
    void remove_size();
    void save_transfer();
    void remove_ransfer();
    void read_transfer();
    std::pair<bool, t_traffic> get_traffic_by_uid(const t_uid &ref);
};

#endif

---