LocalDB.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 __LocalDB_hpp_
#define __LocalDB_hpp_
#include <map>
#include <list>
#include <boost/thread/shared_mutex.hpp>

#define LIMIT_SERVER_USER_NOT_FOUND 2
#define LIMIT_SERVER_ACCESS_GRANT 3
#define LIMIT_SERVER_ACCESS_DENY 4

#include "types.hpp"
#include "hashtable.hpp"

class DataDB;

typedef std::list<std::pair<t_uid, t_traffic> > listL;
typedef HashMap<t_uid, DataDB>                                  mapL;
typedef std::pair<t_uid, DataDB>                                pairL;
typedef std::pair<t_uid, t_traffic>                     pairLL;
class DataDB{
        public :
        
        t_traffic               traffic; 
        t_traffic               specyfic_traffic;
        bool                    always_allow;
        bool                    ban;
        unsigned char   zone_index;
        
        long long               last_up;                
        listL::iterator ptr;   
        DataDB();
        DataDB(t_traffic _traffic);
        DataDB(const DataDB& ref)
        {
                traffic = ref.traffic;
                specyfic_traffic = ref.specyfic_traffic;
                last_up = ref.last_up;
                ptr = ref.ptr;
        }
        DataDB& operator=(const DataDB& ref)
        {
                if(&ref == this) return *this;
                
                traffic = ref.traffic;
                specyfic_traffic = ref.specyfic_traffic;
                last_up = ref.last_up;
                ptr = ref.ptr;
                return *this;
        }
        t_traffic add(t_traffic_int size);
};
class LocalDB {
        long long pos; //position to iterate through hashtable
        public:
        mapL                    *DB;
        listL                    update;
    boost::mutex         list_mutex;
        long long                last_up;

        LocalDB();
        ~LocalDB();
        /*void add_new_user(const t_uid& uid);
        void add_new_user(const t_uid& uid, t_traffic traffic);
        void add_new_user(const t_uid& uid, t_traffic traffic, t_traffic limit);*/
        void add_new_user(const t_uid& uid, t_traffic traffic = 0, t_traffic limit = 0, bool always_allow = false);
        void add_new_user(const t_uid& uid, t_traffic traffic, t_traffic limit, bool always_allow, bool ban);
        void add_new_user_no_update(const t_uid& uid, t_traffic traffic, t_traffic limit, bool always_allow, bool ban);
        void add_full_traffic_to_user(const t_uid& uid, const t_traffic& diff);
        int add_traffic_to_user(const t_uid& context, const t_uid& uid, t_traffic_int diff);
        t_traffic get_traffic_by_uid(const t_uid& uid);
        int get_access_by_uid(const t_uid& context, const t_uid& uid, t_traffic limit);
        int update_base();
        int lock_base();
        int unlock_base();
        void set_specyfic_data(const t_uid& uid, const config_entry& config);
        void reset_specyfic_data(const t_uid& uid);
        void set_data(const t_uid& uid, t_traffic traffic);
        int next(std::pair<t_uid, DataDB> &next);
        size_t size(){return DB->size(); }
        
        void request_data(const t_uid& context, const t_uid& uid);
        void drop_outdated(int ttl);
};
#endif

---