Synnefo

# Copyright 2011 GRNET S.A. All rights reserved.

#

# Redistribution and use in source and binary forms, with or

# without modification, are permitted provided that the following

# conditions are met:

#

#   1. Redistributions of source code must retain the above

#      copyright notice, this list of conditions and the following

#      disclaimer.

#

#   2. Redistributions in binary form must reproduce the above

#      copyright notice, this list of conditions and the following

#      disclaimer in the documentation and/or other materials

#      provided with the distribution.

#

# THIS SOFTWARE IS PROVIDED BY GRNET S.A. ``AS IS'' AND ANY EXPRESS

# OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED

# WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR

# PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL GRNET S.A OR

# CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,

# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT

# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF

# USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED

# AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT

# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN

# ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE

# POSSIBILITY OF SUCH DAMAGE.

#

# The views and conclusions contained in the software and

# documentation are those of the authors and should not be

# interpreted as representing official policies, either expressed

# or implied, of GRNET S.A.

from __future__ import division

from synnefo.api.util import backend_public_networks

def allocate(backends, vm):

    if len(backends) == 1:

        return backends[0]

    # Filter those that can not host the vm

    capable_backends = [backend for backend in backends

                        if vm_fits_in_backend(backend, vm)]

    # Since we are conservatively updating backend resources on each

    # allocation, a backend may actually be able to host a vm (despite

    # the state of the backend in db)

    if not capable_backends:

        capable_backends = backends

    # Compute the scores for each backend

    backend_scores = [(backend, backend_score(backend, vm))

                      for backend in capable_backends]

    # Pick out the best

    result = min(backend_scores, key=lambda (b, b_score): b_score)

    backend = result[0]

    return backend

def vm_fits_in_backend(backend, vm):

    return backend.dfree > vm['disk'] and backend.mfree > vm['ram'] and\

           has_free_ip(backend)

def backend_score(backend, flavor):

    mratio = 1 - (backend.mfree / backend.mtotal)

    dratio = 1 - (backend.dfree / backend.dtotal)

    cratio = (backend.pinst_cnt + 1) / (backend.ctotal * 4)

    return 0.7 * (mratio + dratio) * 0.3 * cratio

def has_free_ip(backend):

    """Find if Backend has any free public IP."""

    for network in backend_public_networks(backend):

        if not network.pool.is_full():

            return True

    return False