diff options
author | Jannis Pohlmann <jannis.pohlmann@codethink.co.uk> | 2012-01-23 17:52:37 +0000 |
---|---|---|
committer | Jannis Pohlmann <jannis.pohlmann@codethink.co.uk> | 2012-01-23 18:26:51 +0000 |
commit | d8f4dbdfe07df8cbb576e32f653c86190f07c392 (patch) | |
tree | f3844aece4a8ff488504b040d6b773102367042c /morphlib/buildworker.py | |
parent | 9d59ed4d78747902cd141f743a6aeabd9e531dc8 (diff) | |
download | morph-d8f4dbdfe07df8cbb576e32f653c86190f07c392.tar.gz |
Add controller, worker classes and a new "build-distributed" command.
This commit introduces four new classes:
BuildController:
* takes an app instance and a tempdir
* allows to add BuildWorker objects
* provides a build() method that takes a set of blobs and
a build order that is then built by assigning work to the
build workers as needed
* the build() method takes care of polling the workers for
their state, moving them between busy and idle states
reliably, collect and print their output in a non-confusing
order, and makes sure to wait for all workers to finish
before processing the next group in the build order.
* at this point, when waiting for one or more workers to
become idle to assign them another blob to build, the
controller always picks the worker that has been idling
for the longest period of time. this can be changed later.
BuildWorker:
* base class for all worker classes
* takes a name and an app instance
* has a idle_since datetime property
* provides a build() method that takes a Blob object and builds
it in whatever way the subclasses implement it
* provides a check_complete(timeout) method that checks whether
the worker has finished building the blob yet or not
LocalBuildWorker:
* worker class for local builds that don't go through SSH
* it uses morphlib.execute.Execute to run morph in a child process
in build()
* at the moment, this class executes "./morph" instead of "morph"
as it assumes the user to run morph from its source tree. obviously,
this will have to be fixed later.
RemoteBuildWorker:
* doesn't implement anything yet, will be used for distributing
work to other machines running morph via SSH
Notes:
* At the moment, there is a degree of undesired redundancy when
building a stratum in a worker, as this will cause the worker to
rebuild all its dependencies. This will have to be fixed as it is
avoidable and wastes a lot of time and processing power.
Diffstat (limited to 'morphlib/buildworker.py')
-rw-r--r-- | morphlib/buildworker.py | 99 |
1 files changed, 99 insertions, 0 deletions
diff --git a/morphlib/buildworker.py b/morphlib/buildworker.py new file mode 100644 index 00000000..ad951466 --- /dev/null +++ b/morphlib/buildworker.py @@ -0,0 +1,99 @@ +# Copyright (C) 2012 Codethink Limited +# +# This program is free software; you can redistribute it and/or modify +# it under the terms of the GNU General Public License as published by +# the Free Software Foundation; version 2 of the License. +# +# This program is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU General Public License for more details. +# +# You should have received a copy of the GNU General Public License along +# with this program; if not, write to the Free Software Foundation, Inc., +# 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. + + +import datetime +from multiprocessing import Manager, Process + +import morphlib + + +class BuildWorker(object): + + def __init__(self, name, app): + self.name = name + self.settings = app.settings + self.real_msg = app.msg + self.indent = 2 + self.idle_since = datetime.datetime.now() + + def __str__(self): + return self.name + + def indent_more(self): + self.indent += 1 + + def indent_less(self): + self.indent -= 1 + + def msg(self, text): + spaces = ' ' * self.indent + self.real_msg('%s%s' % (spaces, text)) + + def build(self, blob): + raise NotImplementedError + + def check_complete(self, timeout): + raise NotImplementedError + + +class LocalBuildWorker(BuildWorker): + + def __init__(self, name, app): + BuildWorker.__init__(self, name, app) + self.manager = Manager() + self.reset() + + def reset(self): + self.process = None + self.blob = None + self._output = self.manager.list() + + def run(self, repo, ref, filename, output): + ex = morphlib.execute.Execute('.', self.msg) + stdout = ex.runv(['./morph', '--verbose', '--keep-path', + 'build', repo, ref, filename]) + output.append(stdout) + + def build(self, blob): + self.reset() + self.blob = blob + args = (blob.morph.treeish.original_repo, + blob.morph.treeish.ref, + blob.morph.filename, + self._output) + self.process = Process(group=None, target=self.run, args=args) + self.process.start() + + def check_complete(self, timeout): + if self.process: + self.process.join(timeout) + if self.process.is_alive(): + return False + else: + self.idle_since = datetime.datetime.now() + return True + else: + return True + + @property + def output(self): + return self._output[0] + + +class RemoteBuildWorker(BuildWorker): + + def __init__(self, app): + BuildWorker.__init__(self, app) |