Haizea

# -------------------------------------------------------------------------- #

# Copyright 2006-2010, University of Chicago                                 #

# Copyright 2008-2010, Distributed Systems Architecture Group, Universidad   #

# Complutense de Madrid (dsa-research.org)                                   #

#                                                                            #

# 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.                                             #

# -------------------------------------------------------------------------- #

import haizea.common.stats as stats

from haizea.core.leases import LeaseWorkload, LeaseAnnotation, LeaseAnnotations, Timestamp, Lease,\

    Capacity, Duration, UnmanagedSoftwareEnvironment,\

    DiskImageSoftwareEnvironment, Site

from haizea.common.utils import round_datetime_delta

from mx.DateTime import DateTimeDelta, TimeDelta, Parser

import ConfigParser 

try:

    import xml.etree.ElementTree as ET

except ImportError:

    # Compatibility with Python <=2.4

    import elementtree.ElementTree as ET 

class FileGenerator(object):

    GENERAL_SEC = "general"

    TYPE_OPT = "type"

    DISTRIBUTION_OPT = "distribution"

    DISCRETE_OPT = "discrete"

    MIN_OPT = "min"

    MAX_OPT = "max"

    VALUES_OPT = "values"

    MEAN_OPT = "mu"

    STDEV_OPT = "sigma"

    ALPHA_OPT = "alpha"

    SCALE_OPT = "scale"

    INVERT_OPT = "invert"

    SEED_OPT = "seed"

    START_DELAY_SEC = "start-delay"

    START_ABSOLUTE = "absolute"

    START_DURATION = "multiple-of-duration"

    DEADLINE_STRETCH_SEC = "deadline-stretch"

    DEADLINE_DURATION = "multiple-of-duration"

    DEADLINE_SLOWDOWN = "original-slowdown"

    DEADLINE_ABSOLUTE = "absolute"

    RATE_SEC = "user-rate"

    NODES_SEC = "nodes"

    RESOURCES_OPT = "resources"

    DURATION_SEC = "duration"

    SOFTWARE_SEC = "software"

    def __init__(self, outfile, conffile):

        self.outfile = outfile

        self.conffile = conffile

        conffile = open(conffile, "r")

        self.config = ConfigParser.ConfigParser()

        self.config.readfp(conffile)

        self.startdelay_dist = self._get_dist(FileGenerator.START_DELAY_SEC)

        self.deadlinestretch_dist = self._get_dist(FileGenerator.DEADLINE_STRETCH_SEC)

        self.rate_dist = self._get_dist(FileGenerator.RATE_SEC)

        self.duration_dist = self._get_dist(FileGenerator.DURATION_SEC)

        self.numnodes_dist = self._get_dist(FileGenerator.NODES_SEC)

        self.software_dist = self._get_dist(FileGenerator.SOFTWARE_SEC)

        self.start_type = self.config.get(FileGenerator.START_DELAY_SEC, FileGenerator.TYPE_OPT)

        self.deadline_type = self.config.get(FileGenerator.DEADLINE_STRETCH_SEC, FileGenerator.TYPE_OPT)

    def _get_start(self, type, lease = None):

        if self.startdelay_dist == None:

            return None, None

        else:

            delta = self.startdelay_dist.get()

            if type == FileGenerator.START_ABSOLUTE:

                start = round_datetime_delta(TimeDelta(seconds=delta))

            elif type == FileGenerator.START_DURATION:

                start = round_datetime_delta(delta * lease.duration.requested)

            return start, delta

    def _get_numnodes(self, lease = None):

        if self.numnodes_dist == None:

            return None

        else:

            numnodes = int(self.numnodes_dist.get())

            return numnodes      

    def _get_duration(self, lease = None):

        if self.duration_dist == None:

            return None

        else:

            numnodes = int(self.duration_dist.get())

            return numnodes              

    def _get_deadline(self, type, lease, start):

        if self.deadlinestretch_dist == None:

            return None, None

        else:

            if type in (FileGenerator.DEADLINE_DURATION, FileGenerator.DEADLINE_SLOWDOWN):

                if type == FileGenerator.DEADLINE_DURATION:

                    tau = self.deadlinestretch_dist.get()

                elif type == FileGenerator.DEADLINE_SLOWDOWN:

                    runtime = float(lease.extras["SWF_runtime"])

                    waittime = float(lease.extras["SWF_waittime"])

                    if runtime < 10: runtime = 10

                    slowdown = (waittime + runtime) / runtime

                    min = self.deadlinestretch_dist.min

                    max = self.deadlinestretch_dist.max

                    tau = self.deadlinestretch_dist.get()

                    tau = (slowdown - 1)*((tau-min) / (max-min))

                deadline = round_datetime_delta(start + (1 + tau)*lease.duration.requested)                

            elif type == FileGenerator.DEADLINE_ABSOLUTE:

                wait = self.deadlinestretch_dist.get()

                deadline = round_datetime_delta(start + TimeDelta(seconds=wait) + lease.duration.requested)  

                tau = ((deadline - start) / lease.duration.requested) - 1                    

            return deadline, tau

    def _get_software(self, lease = None):

        if self.software_dist == None:

            return UnmanagedSoftwareEnvironment()

        else:

            software = self.software_dist.get()

            image_id, image_size = software.split("|")

            return DiskImageSoftwareEnvironment(image_id, int(image_size))

    def _get_rate(self, lease):

        if self.rate_dist == None:

            return None

        else:

            if lease.user_id == -1:

                return self.rate_dist.get()

            else:

                if self.user_rates.has_key(lease.user_id):

                    return self.user_rates[lease.user_id]

                else:

                    rate = self.rate_dist.get()

                    self.user_rates[lease.user_id] = rate

                    return rate

    def _get_attributes(self):

        attributes = {}

        attrs = self.config.get(FileGenerator.GENERAL_SEC, FileGenerator.ATTRIBUTES_OPT)

        attrs = attrs.split(",")

        for attr in attrs:

            (k,v) = attr.split("=")

            attributes[k] = v

        return attributes

    def _get_dist(self, section):

        if self.config.has_section(section):

            return self.__create_distribution_from_section(section)

        else:

            return None

    def __create_distribution_from_section(self, section):

        dist_type = self.config.get(section, FileGenerator.DISTRIBUTION_OPT)

        if self.config.has_option(section, FileGenerator.DISCRETE_OPT):

            discrete = self.config.getboolean(section, FileGenerator.DISCRETE_OPT)

        else:

            discrete = False

        if self.config.has_option(section, FileGenerator.MIN_OPT) and self.config.has_option(section, FileGenerator.MAX_OPT):

            min = self.config.get(section, FileGenerator.MIN_OPT)

            max = self.config.get(section, FileGenerator.MAX_OPT)

            if min == "unbounded":

                min = float("inf")

            else:

                min = float(min)

            if max == "unbounded":

                max = float("inf")

            else:

                max = float(max)

        else:

            min = max = None

        if discrete:

            if dist_type == "uniform":

                if min != None:

                    values = range(int(min), int(max) + 1)

                else:

                    values = self.config.get(section, FileGenerator.VALUES_OPT).split()

                dist = stats.DiscreteUniformDistribution(values)

        else:

            if dist_type == "uniform":

                dist = stats.UniformDistribution(min, max)

            elif dist_type == "normal":

                mu = self.config.getfloat(section, FileGenerator.MEAN_OPT)

                sigma = self.config.getfloat(section, FileGenerator.STDEV_OPT)

                dist = stats.BoundedNormalDistribution(min,max,mu,sigma)

            elif dist_type == "bounded-pareto" or dist_type == "truncated-pareto":

                alpha = self.config.getfloat(section, FileGenerator.ALPHA_OPT)

                if self.config.has_option(section, FileGenerator.INVERT_OPT):

                    invert = self.config.getboolean(section, FileGenerator.INVERT_OPT)

                else:

                    invert = False

                if dist_type == "bounded-pareto":

                    dist = stats.BoundedParetoDistribution(min,max,alpha,invert)

                else:

                    scale = self.config.getfloat(section, FileGenerator.SCALE_OPT)

                    dist = stats.TruncatedParetoDistribution(min,max,scale,alpha,invert)

        if self.config.has_option(section, FileGenerator.SEED_OPT):

            seed = self.config.getint(section, FileGenerator.SEED_OPT)

            dist.seed(seed)

        return dist

class LWFGenerator(FileGenerator):

    SITE_OPT = "site"    

    NUMLEASES_SEC = "numleases"    

    NUMLEASES_TYPE_UTILIZATION = "utilization"    

    NUMLEASES_UTILIZATION_OPT = "utilization"    

    NUMLEASES_LAST_REQUEST_OPT = "last-request"

    NUMLEASES_TYPE_INTERVAL = "interval"    

    NUMLEASES_OPT = "numleases"    

    def __init__(self, outfile, conffile):

        FileGenerator.__init__(self, outfile, conffile)

        self.numleases_type = self.config.get(LWFGenerator.NUMLEASES_SEC, LWFGenerator.TYPE_OPT)

        self.interval_dist = self._get_dist(LWFGenerator.NUMLEASES_SEC)

    def _get_interval(self):

        if self.interval_dist == None:

            return None

        else:

            interval = int(self.interval_dist.get())

            return interval    

    def __gen_lease(self):

        submit_time = None

        user_id = None

        res = self.config.get(LWFGenerator.NODES_SEC, LWFGenerator.RESOURCES_OPT)

        res = Capacity.from_resources_string(res)        

        numnodes = self._get_numnodes(None)

        requested_resources = dict([(i+1,res) for i in xrange(numnodes)])

        start, delta = self._get_start(self.start_type, None)

        start = Timestamp(TimeDelta(seconds=start))

        duration = self._get_duration()

        duration = Duration(TimeDelta(seconds=duration))

        deadline = None

        preemptible = False

        software = self._get_software()

        l = Lease.create_new(submit_time, user_id, requested_resources, 

                             start, duration, deadline, preemptible, software)

        return l

    def generate(self):

        lwf = ET.Element("lease-workload")

        lwf.set("name", self.outfile)

        description = ET.SubElement(lwf, "description")

        description.text = "Created with haizea-generate"

        site = self.config.get(LWFGenerator.GENERAL_SEC, LWFGenerator.SITE_OPT)

        if site.startswith("file:"):

            sitefile = site.split(":")

            site = Site.from_xml_file(sitefile[1])

        else:

            site = Site.from_resources_string(site)

        lwf.append(site.to_xml())

        time = TimeDelta(seconds=0)

        requests = ET.SubElement(lwf, "lease-requests")   

        if self.numleases_type == LWFGenerator.NUMLEASES_TYPE_INTERVAL:

            leases = []            

            numleases = self.config.getint(LWFGenerator.NUMLEASES_SEC, LWFGenerator.NUMLEASES_OPT)

            for i in xrange(numleases):

                leases.append(self.__gen_lease())

            for l in leases:

                interval = TimeDelta(seconds=self._get_interval())

                time += interval

                l.start.requested += time

                lease_request = ET.SubElement(requests, "lease-request")

                lease_request.set("arrival", str(time))            

                lease_request.append(l.to_xml())

        elif self.numleases_type == LWFGenerator.NUMLEASES_TYPE_UTILIZATION:

            utilization = self.config.getfloat(LWFGenerator.NUMLEASES_SEC, LWFGenerator.NUMLEASES_UTILIZATION_OPT)

            last_request = self.config.get(LWFGenerator.NUMLEASES_SEC, LWFGenerator.NUMLEASES_LAST_REQUEST_OPT)

            last_request = Parser.DateTimeDeltaFromString(last_request)

            max_utilization = 0

            for res in site.nodes.get_all_nodes().values():

                for i in range(1,res.get_ninstances("CPU") + 1):

                    max_utilization += res.get_quantity_instance("CPU", i) * last_request.seconds

            target_utilization = int(max_utilization * utilization)

            accum_utilization = 0

            leases = []            

            while accum_utilization < target_utilization:

                lease = self.__gen_lease()

                leases.append(lease)

                duration = lease.duration.requested.seconds

                lease_utilization = 0

                for res in lease.requested_resources.values():

                    for i in range(1,res.get_ninstances("CPU") + 1):

                        lease_utilization += res.get_quantity_instance("CPU", i) * duration                

                accum_utilization += lease_utilization

            time = TimeDelta(seconds=0)            

            avg_interval = int(last_request.seconds / len(leases))

            for l in leases:

                interval = avg_interval + TimeDelta(seconds=self._get_interval())

                time = max(time + interval, TimeDelta(seconds=0))

                l.start.requested += time

                lease_request = ET.SubElement(requests, "lease-request")

                lease_request.set("arrival", str(time))            

                lease_request.append(l.to_xml())                            

        tree = ET.ElementTree(lwf)

        outfile = open(self.outfile, "w")

        tree.write(outfile)

        outfile.close()

class LWFAnnotationGenerator(FileGenerator):

    def __init__(self, lwffile, outfile, conffile):

        FileGenerator.__init__(self, outfile, conffile)

        self.lwffile = lwffile

    def generate(self):

        lease_workload = LeaseWorkload.from_xml_file(self.lwffile)

        leases = lease_workload.get_leases()

        annotations = {}

        for lease in leases:

            lease_id = lease.id

            extra = {}

            start, delta = self._get_start(self.start_type, lease)

            if start != None:

                start = Timestamp(start)

                extra["simul_start_delta"] = "%.2f" % delta

            deadline, tau = self._get_deadline(self.deadline_type, lease, start.requested)

            if deadline != None:

                extra["simul_deadline_tau"] = "%.2f" % tau

            software = self._get_software(lease)

            rate = self._get_rate(lease)

            if rate != None:

                extra["simul_userrate"] = "%.2f" % rate

            annotation = LeaseAnnotation(lease_id, start, deadline, software, extra)

            annotations[lease_id] = annotation

        attributes = self._get_attributes()

        annotations = LeaseAnnotations(annotations, attributes)

        tree = ET.ElementTree(annotations.to_xml())

        outfile = open(self.outfile, "w")

        tree.write(outfile)

        outfile.close()