BatchJobSet.h

 
#ifndef WRENCH_BATCHJOBSET_H
#define WRENCH_BATCHJOBSET_H
 
#include "wrench/services/compute/batch/BatchJob.h"
 
namespace wrench {
 
    /***********************/
    /***********************/
 
    class BatchJobSet {
 
    public:
 
        BatchJobSet() = default;
 
        std::set<std::shared_ptr<BatchJob>> jobs;
        unsigned long num_nodes_utilized = 0;
 
        BatchJobSet& operator += (const BatchJobSet& right)
        {
            for (const auto &j : right.jobs) {
                add(j);
            }
            return *this;
        }
 
        BatchJobSet& operator -= (const BatchJobSet& right)
        {
            for (const auto &j : right.jobs) {
                remove(j);
            }
            return *this;
        }
 
        void inline add(std::shared_ptr<BatchJob> job) {
            if (this->jobs.find(job) == this->jobs.end()) {
                num_nodes_utilized += job->getRequestedNumNodes();
                this->jobs.insert(job);
            }
        }
 
        void inline remove(std::shared_ptr<BatchJob> job) {
            if  (this->jobs.find(job) != this->jobs.end()) {
                num_nodes_utilized -= job->getRequestedNumNodes();
                this->jobs.erase(job);
            }
        }
 
    };
 
    /***********************/
    /***********************/
 
}
 
#endif //WRENCH_BATCHJOBSET_H