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 <cstdio>
#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;
        
        long long last_up;
        listL::iterator ptr;
        DataDB();
        DataDB(t_traffic _traffic);
        DataDB(const DataDB& ref) {
                traffic = ref.traffic;
                specyfic_traffic = ref.specyfic_traffic;
                this->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
        //t_traffic limit;
        public:
                mapL *DB;
                listL update;
            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_no_update(const t_uid& uid, t_traffic traffic, t_traffic limit, bool always_allow, bool ban);

        void add_traffic_to_user(const t_uid& uid, t_traffic_int diff);
        void add_full_traffic_to_user(const t_uid& uid, const t_traffic& diff);
        bool get_traffic_by_uid(const t_uid& uid, t_traffic& ret);
    t_traffic get_data_by_uid(const t_uid& uid);
        int get_access_by_uid(const t_uid& uid, t_traffic limit);
        int update_base();
        int lock_base();
        int unlock_base();
        void set_specyfic_data(const t_uid& uid, t_traffic limit, bool always_allow);
        void set_data(const t_uid& uid, t_traffic traffic/*, t_traffic limit*/);
        int next(std::pair<t_uid, DataDB> &next);
        size_t size() { return DB->size(); }

        void drop_outdated(int ttl);
};
#endif

---