source: old/trunk/vtest/cluster.py @ 2598

#!/usr/bin/python
# -*- coding: utf-8 -*-
# Copyright © 2009 by Karl Ramm
#
# All rights reserved.
#
# Permission to use, copy, modify, and distribute this software and
# its documentation for any purpose and without fee is hereby granted,
# provided that the above copyright notice appear in all copies and
# that both that copyright notice and this permission notice appear in
# supporting documentation, and that the name of Karl Ramm not be used
# in advertising or publicity pertaining to distribution of the
# software without specific, written prior permission.
#
# KARL RAMM DISCLAIMS ALL WARRANTIES WITH REGARD TO THIS SOFTWARE,
# INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS, IN
# NO EVENT SHALL KARL RAMM BE LIABLE FOR ANY SPECIAL, INDIRECT OR
# CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS
# OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT,
# NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION
# WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.

import vm
from time import sleep, time

class host(object):
    def __init__(self, machine):
        self.machine = machine
    def setup(self):
        pass
    def teardown(self):
        pass

class cluster(object):
    def __init__(self, count=None, spec=None, machine=vm.sshtestmachine):
        if spec is None:
            spec = count*[host]
        self.machines = {}
        self._all = []
        for proto in spec:
            m = machine()
            p = proto(m)
            t = (m, p)
            self._all.append(t)
            self.machines.setdefault(proto, []).append(t)
    def setup(self):
        while any(not m.ready() for (m, p) in self._all):
            sleep(7)
        for (m, p) in self._all:
            p.setup()
    def teardown(self):
        for (m, p) in self._all:
            p.teardown()
            m.shutdown()
        start = time()
        while True:
            still = [m for (m, p) in self._all if m.alive()]
            if not still:
                break
            sleep(7)
            if time() > (start + 300):
                for m in still:
                    m.shoot()