/* -------------------------------------------------------------------------- */
/* Copyright 2002-2022, OpenNebula Project, OpenNebula Systems                */
/*                                                                            */
/* 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.                                             */
/* -------------------------------------------------------------------------- */

#ifndef RANK_SCHEDULER_H_
#define RANK_SCHEDULER_H_

#include "Scheduler.h"
#include "SchedulerTemplate.h"
#include "RankPolicy.h"
#include "UserPriorityPolicy.h"


class RankScheduler : public Scheduler
{
public:

    RankScheduler():Scheduler(),rp_host(0),rp_ds(0),rp_nics(0), rp_vm(0){};

    ~RankScheduler()
    {
        delete rp_host;
        delete rp_ds;
        delete rp_nics;

        delete rp_vm;
    };

    void register_policies(const SchedulerTemplate& conf)
    {
        rp_host = new RankHostPolicy(hpool, conf.get_policy(), 1.0);

        add_host_policy(rp_host);

        rp_ds = new RankDatastorePolicy(dspool, conf.get_ds_policy(), 1.0);

        add_ds_policy(rp_ds);

        rp_vm = new UserPriorityPolicy(vmpool, 1.0);

        add_vm_policy(rp_vm);

        rp_nics = new RankNetworkPolicy(vnetpool, conf.get_nics_policy(), 1.0);

        add_nic_policy(rp_nics);
    };

private:
    RankPolicy * rp_host;
    RankPolicy * rp_ds;
    RankPolicy * rp_nics;

    UserPriorityPolicy * rp_vm;
};

#endif // RANK_SCHEDULER_H_