summaryrefslogtreecommitdiff
path: root/morphlib/buildworker.py
diff options
context:
space:
mode:
authorJannis Pohlmann <jannis.pohlmann@codethink.co.uk>2012-01-23 17:52:37 +0000
committerJannis Pohlmann <jannis.pohlmann@codethink.co.uk>2012-01-23 18:26:51 +0000
commitd8f4dbdfe07df8cbb576e32f653c86190f07c392 (patch)
treef3844aece4a8ff488504b040d6b773102367042c /morphlib/buildworker.py
parent9d59ed4d78747902cd141f743a6aeabd9e531dc8 (diff)
downloadmorph-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.py99
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)