ContractBasedOptimizer.h

Go to the documentation of this file.

#ifndef MEDUSA_CONTRACTBASEDOPT_H
#define MEDUSA_CONTRACTBASEDOPT_H

#include "NHOptimizer.h"

MEDUSA_NAMESPACE_BEGIN


// --------------------------------------------------
class Contract { 
 public: 
    Contract(string partner = "", double min_price = 0.0, double max_price = 0.0) 
        : m_partner(partner), m_min_price(min_price), 
        m_current_price(min_price), m_max_price(max_price), m_last_offer_rejected(false), m_broken(false)  {} 
    string m_partner; 
    double m_min_price; 
    double m_current_price;
    double m_max_price;
    bool m_last_offer_rejected;
    bool m_broken; // Indicates if there is something wrong with talking to this partner
    bool operator<(const Contract& c) const { 

        // Broken/unresponsive partners go last
        if ( c.m_broken != m_broken) {
            return  c.m_broken; 
        }

        // Before last go all partners that last rejected our offer
        if ( c.m_last_offer_rejected != m_last_offer_rejected) {
            return  c.m_last_offer_rejected; 
        }

        // Finally we sort on price
        return ( m_current_price < c.m_current_price);

        // If prices are the same, this contract is lower if didn't reject while partner did
        //if ( c.m_current_price == m_current_price) {            
        //  return (!m_last_offer_rejected) && c.m_last_offer_rejected;
        //} else {
        //   return ( m_current_price < c.m_current_price);
        //}
    } 

    NMSTL_SIMPLY_SERIALIZABLE(Contract, << m_partner << m_min_price << m_max_price);

    string as_string() const { 
        stringstream out;
        out.precision(3);
        out << "{Contract with " << m_partner << "[" << m_min_price << ":" 
            << m_current_price << ":" << m_max_price << "]";
        
        return out.str();
    }
    
};
NMSTL_TO_STRING(Contract);   


// --------------------------------------------------
class ContractBasedOptimizer : public NHOptimizer {

 public:
    ContractBasedOptimizer(string id, string configFileName=DEFAULT_CONFIG_FILE);
    ~ContractBasedOptimizer() {}
    
 protected:
    void in_thread_init();
    void handle_load_offer(AsyncRPC< ptr<LoadOffer> > completion, ptr<LoadOffer> offer);

 private:

    class OverloadedException {
    public:
        string m_message;
        OverloadedException(string message): m_message(message) { }
    };

    //static const double EPSILON_COST = 0.002;
    static const double EPSILON_COST = 0.1;

    // Cached version of latest stats
    typedef vector< Stats > StatsList;
    vector<Stats> m_operatorStats;    
    double m_total_cpu; // Total used in percent
    double m_total_bw; // Total used in MBytes
    double m_bw_utilization; // Fraction total used (ratio)
    double m_cpu_utilization; // Fraction total used (ratio)
    double m_load_level;
    double m_marginal_cost;


    typedef list< Contract > Contracts;
    Contracts m_contracts; 

    void init_contracts();

    void update_stats();
    void update_stats_2(RPC<StatsList> result);

    double compute_avg_marginal_cost(double cpuUsage, double deltaCPU, double bwUsage, double deltaBW);    

    // Giving load
    bool m_offerPending;

    void reset_partners();
    void send_out_load_offers();
    void pick_boxes(StatsList& src, StatsList& dst, double threshold);
    void handle_load_offer_response(string partner, ptr<LoadOffer> original_offer, 
                                    RPC< ptr<LoadOffer> > counter_offer);
    void done_moving_load(RPC<void> stat);
    vector<Name> extract_names(vector<Stats> boxes);

    // Taking load
    bool m_movementInProgress;
    typedef list< ptr<LoadOffer> > LoadOffers;
    typedef map < ptr<LoadOffer>, AsyncRPC< ptr<LoadOffer> > > LoadOfferCompletions;
    LoadOffers m_load_offers;
    LoadOfferCompletions m_load_offer_completions;
    void examine_load_offers();
    void pick_new_boxes(StatsList& src, StatsList& dst, double threshold,
                        double& acceptedDeltaCPU, double& acceptedDeltaBW);


};

MEDUSA_NAMESPACE_END

#endif

---