commit
ee743613ce
@ -0,0 +1,6 @@ |
||||
build |
||||
dist |
||||
zuul.egg-info |
||||
MANIFEST |
||||
.tox |
||||
|
||||
@ -0,0 +1,4 @@ |
||||
[gerrit] |
||||
host=review.openstack.org |
||||
port=29418 |
||||
project=openstack-ci/zuul.git |
||||
@ -0,0 +1,202 @@ |
||||
|
||||
Apache License |
||||
Version 2.0, January 2004 |
||||
http://www.apache.org/licenses/ |
||||
|
||||
TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION |
||||
|
||||
1. Definitions. |
||||
|
||||
"License" shall mean the terms and conditions for use, reproduction, |
||||
and distribution as defined by Sections 1 through 9 of this document. |
||||
|
||||
"Licensor" shall mean the copyright owner or entity authorized by |
||||
the copyright owner that is granting the License. |
||||
|
||||
"Legal Entity" shall mean the union of the acting entity and all |
||||
other entities that control, are controlled by, or are under common |
||||
control with that entity. For the purposes of this definition, |
||||
"control" means (i) the power, direct or indirect, to cause the |
||||
direction or management of such entity, whether by contract or |
||||
otherwise, or (ii) ownership of fifty percent (50%) or more of the |
||||
outstanding shares, or (iii) beneficial ownership of such entity. |
||||
|
||||
"You" (or "Your") shall mean an individual or Legal Entity |
||||
exercising permissions granted by this License. |
||||
|
||||
"Source" form shall mean the preferred form for making modifications, |
||||
including but not limited to software source code, documentation |
||||
source, and configuration files. |
||||
|
||||
"Object" form shall mean any form resulting from mechanical |
||||
transformation or translation of a Source form, including but |
||||
not limited to compiled object code, generated documentation, |
||||
and conversions to other media types. |
||||
|
||||
"Work" shall mean the work of authorship, whether in Source or |
||||
Object form, made available under the License, as indicated by a |
||||
copyright notice that is included in or attached to the work |
||||
(an example is provided in the Appendix below). |
||||
|
||||
"Derivative Works" shall mean any work, whether in Source or Object |
||||
form, that is based on (or derived from) the Work and for which the |
||||
editorial revisions, annotations, elaborations, or other modifications |
||||
represent, as a whole, an original work of authorship. For the purposes |
||||
of this License, Derivative Works shall not include works that remain |
||||
separable from, or merely link (or bind by name) to the interfaces of, |
||||
the Work and Derivative Works thereof. |
||||
|
||||
"Contribution" shall mean any work of authorship, including |
||||
the original version of the Work and any modifications or additions |
||||
to that Work or Derivative Works thereof, that is intentionally |
||||
submitted to Licensor for inclusion in the Work by the copyright owner |
||||
or by an individual or Legal Entity authorized to submit on behalf of |
||||
the copyright owner. For the purposes of this definition, "submitted" |
||||
means any form of electronic, verbal, or written communication sent |
||||
to the Licensor or its representatives, including but not limited to |
||||
communication on electronic mailing lists, source code control systems, |
||||
and issue tracking systems that are managed by, or on behalf of, the |
||||
Licensor for the purpose of discussing and improving the Work, but |
||||
excluding communication that is conspicuously marked or otherwise |
||||
designated in writing by the copyright owner as "Not a Contribution." |
||||
|
||||
"Contributor" shall mean Licensor and any individual or Legal Entity |
||||
on behalf of whom a Contribution has been received by Licensor and |
||||
subsequently incorporated within the Work. |
||||
|
||||
2. Grant of Copyright License. Subject to the terms and conditions of |
||||
this License, each Contributor hereby grants to You a perpetual, |
||||
worldwide, non-exclusive, no-charge, royalty-free, irrevocable |
||||
copyright license to reproduce, prepare Derivative Works of, |
||||
publicly display, publicly perform, sublicense, and distribute the |
||||
Work and such Derivative Works in Source or Object form. |
||||
|
||||
3. Grant of Patent License. Subject to the terms and conditions of |
||||
this License, each Contributor hereby grants to You a perpetual, |
||||
worldwide, non-exclusive, no-charge, royalty-free, irrevocable |
||||
(except as stated in this section) patent license to make, have made, |
||||
use, offer to sell, sell, import, and otherwise transfer the Work, |
||||
where such license applies only to those patent claims licensable |
||||
by such Contributor that are necessarily infringed by their |
||||
Contribution(s) alone or by combination of their Contribution(s) |
||||
with the Work to which such Contribution(s) was submitted. If You |
||||
institute patent litigation against any entity (including a |
||||
cross-claim or counterclaim in a lawsuit) alleging that the Work |
||||
or a Contribution incorporated within the Work constitutes direct |
||||
or contributory patent infringement, then any patent licenses |
||||
granted to You under this License for that Work shall terminate |
||||
as of the date such litigation is filed. |
||||
|
||||
4. Redistribution. You may reproduce and distribute copies of the |
||||
Work or Derivative Works thereof in any medium, with or without |
||||
modifications, and in Source or Object form, provided that You |
||||
meet the following conditions: |
||||
|
||||
(a) You must give any other recipients of the Work or |
||||
Derivative Works a copy of this License; and |
||||
|
||||
(b) You must cause any modified files to carry prominent notices |
||||
stating that You changed the files; and |
||||
|
||||
(c) You must retain, in the Source form of any Derivative Works |
||||
that You distribute, all copyright, patent, trademark, and |
||||
attribution notices from the Source form of the Work, |
||||
excluding those notices that do not pertain to any part of |
||||
the Derivative Works; and |
||||
|
||||
(d) If the Work includes a "NOTICE" text file as part of its |
||||
distribution, then any Derivative Works that You distribute must |
||||
include a readable copy of the attribution notices contained |
||||
within such NOTICE file, excluding those notices that do not |
||||
pertain to any part of the Derivative Works, in at least one |
||||
of the following places: within a NOTICE text file distributed |
||||
as part of the Derivative Works; within the Source form or |
||||
documentation, if provided along with the Derivative Works; or, |
||||
within a display generated by the Derivative Works, if and |
||||
wherever such third-party notices normally appear. The contents |
||||
of the NOTICE file are for informational purposes only and |
||||
do not modify the License. You may add Your own attribution |
||||
notices within Derivative Works that You distribute, alongside |
||||
or as an addendum to the NOTICE text from the Work, provided |
||||
that such additional attribution notices cannot be construed |
||||
as modifying the License. |
||||
|
||||
You may add Your own copyright statement to Your modifications and |
||||
may provide additional or different license terms and conditions |
||||
for use, reproduction, or distribution of Your modifications, or |
||||
for any such Derivative Works as a whole, provided Your use, |
||||
reproduction, and distribution of the Work otherwise complies with |
||||
the conditions stated in this License. |
||||
|
||||
5. Submission of Contributions. Unless You explicitly state otherwise, |
||||
any Contribution intentionally submitted for inclusion in the Work |
||||
by You to the Licensor shall be under the terms and conditions of |
||||
this License, without any additional terms or conditions. |
||||
Notwithstanding the above, nothing herein shall supersede or modify |
||||
the terms of any separate license agreement you may have executed |
||||
with Licensor regarding such Contributions. |
||||
|
||||
6. Trademarks. This License does not grant permission to use the trade |
||||
names, trademarks, service marks, or product names of the Licensor, |
||||
except as required for reasonable and customary use in describing the |
||||
origin of the Work and reproducing the content of the NOTICE file. |
||||
|
||||
7. Disclaimer of Warranty. Unless required by applicable law or |
||||
agreed to in writing, Licensor provides the Work (and each |
||||
Contributor provides its Contributions) on an "AS IS" BASIS, |
||||
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or |
||||
implied, including, without limitation, any warranties or conditions |
||||
of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A |
||||
PARTICULAR PURPOSE. You are solely responsible for determining the |
||||
appropriateness of using or redistributing the Work and assume any |
||||
risks associated with Your exercise of permissions under this License. |
||||
|
||||
8. Limitation of Liability. In no event and under no legal theory, |
||||
whether in tort (including negligence), contract, or otherwise, |
||||
unless required by applicable law (such as deliberate and grossly |
||||
negligent acts) or agreed to in writing, shall any Contributor be |
||||
liable to You for damages, including any direct, indirect, special, |
||||
incidental, or consequential damages of any character arising as a |
||||
result of this License or out of the use or inability to use the |
||||
Work (including but not limited to damages for loss of goodwill, |
||||
work stoppage, computer failure or malfunction, or any and all |
||||
other commercial damages or losses), even if such Contributor |
||||
has been advised of the possibility of such damages. |
||||
|
||||
9. Accepting Warranty or Additional Liability. While redistributing |
||||
the Work or Derivative Works thereof, You may choose to offer, |
||||
and charge a fee for, acceptance of support, warranty, indemnity, |
||||
or other liability obligations and/or rights consistent with this |
||||
License. However, in accepting such obligations, You may act only |
||||
on Your own behalf and on Your sole responsibility, not on behalf |
||||
of any other Contributor, and only if You agree to indemnify, |
||||
defend, and hold each Contributor harmless for any liability |
||||
incurred by, or claims asserted against, such Contributor by reason |
||||
of your accepting any such warranty or additional liability. |
||||
|
||||
END OF TERMS AND CONDITIONS |
||||
|
||||
APPENDIX: How to apply the Apache License to your work. |
||||
|
||||
To apply the Apache License to your work, attach the following |
||||
boilerplate notice, with the fields enclosed by brackets "[]" |
||||
replaced with your own identifying information. (Don't include |
||||
the brackets!) The text should be enclosed in the appropriate |
||||
comment syntax for the file format. We also recommend that a |
||||
file or class name and description of purpose be included on the |
||||
same "printed page" as the copyright notice for easier |
||||
identification within third-party archives. |
||||
|
||||
Copyright [yyyy] [name of copyright owner] |
||||
|
||||
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. |
||||
@ -0,0 +1 @@ |
||||
This is a trunk gating system developed for the OpenStack Project. |
||||
@ -0,0 +1,31 @@ |
||||
#!/usr/bin/env python |
||||
# Copyright 2012 Hewlett-Packard Development Company, L.P. |
||||
# |
||||
# 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. |
||||
|
||||
from setuptools import find_packages |
||||
from setuptools.command.sdist import sdist |
||||
from setuptools import setup |
||||
import subprocess |
||||
|
||||
setup(name='zuul', |
||||
version='1.0', |
||||
description="Trunk gating system", |
||||
license='Apache License (2.0)', |
||||
author='Hewlett-Packard Development Company, L.P.', |
||||
author_email='openstack@lists.launchpad.net', |
||||
url='http://launchpad.net/zuul', |
||||
scripts=['zuul-server'], |
||||
include_package_data=True, |
||||
zip_safe=False, |
||||
) |
||||
@ -0,0 +1,53 @@ |
||||
import argparse |
||||
import ConfigParser |
||||
import os |
||||
|
||||
import zuul.scheduler |
||||
import zuul.launcher.jenkins |
||||
import zuul.trigger.gerrit |
||||
|
||||
import logging.config |
||||
|
||||
def parse_arguments(): |
||||
parser = argparse.ArgumentParser(description='Project gating system.') |
||||
parser.add_argument('-c', dest='config', |
||||
help='specify the config file') |
||||
return parser.parse_args() |
||||
|
||||
def read_config(args): |
||||
config=ConfigParser.ConfigParser() |
||||
if args.config: |
||||
locations = [args.config] |
||||
else: |
||||
locations = ['/etc/zuul/zuul.conf', |
||||
'~/zuul.conf'] |
||||
for fp in locations: |
||||
if os.path.exists(os.path.expanduser(fp)): |
||||
config.read(fp) |
||||
return config |
||||
raise Exception("Unable to locate config file in %s" % locations) |
||||
|
||||
def setup_logging(config): |
||||
if config.has_option('zuul', 'log_config'): |
||||
fp = os.path.expanduser(config.get('zuul', 'log_config')) |
||||
if not os.path.exists(fp): |
||||
raise Exception("Unable to read logging config file at %s" % fp) |
||||
logging.config.fileConfig(fp) |
||||
else: |
||||
logging.basicConfig(level=logging.DEBUG) |
||||
|
||||
def main(config): |
||||
sched = zuul.scheduler.Scheduler(config) |
||||
|
||||
jenkins = zuul.launcher.jenkins.Jenkins(config, sched) |
||||
gerrit = zuul.trigger.gerrit.Gerrit(config, sched) |
||||
|
||||
sched.setLauncher(jenkins) |
||||
sched.setTrigger(gerrit) |
||||
sched.run() |
||||
|
||||
if __name__ == '__main__': |
||||
args = parse_arguments() |
||||
config = read_config(args) |
||||
setup_logging(config) |
||||
main(config) |
||||
@ -0,0 +1,217 @@ |
||||
# Copyright 2012 Hewlett-Packard Development Company, L.P. |
||||
# |
||||
# 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. |
||||
|
||||
# So we can name this module "jenkins" and still load the "jenkins" |
||||
# system module |
||||
from __future__ import absolute_import |
||||
|
||||
import threading |
||||
from webob import Request |
||||
from paste import httpserver |
||||
from uuid import uuid1 |
||||
import jenkins |
||||
import json |
||||
import urllib2 # for extending jenkins lib |
||||
import logging |
||||
import pprint |
||||
|
||||
from zuul.model import Build |
||||
|
||||
class JenkinsCallback(threading.Thread): |
||||
log = logging.getLogger("zuul.JenkinsCallback") |
||||
|
||||
def __init__(self, jenkins): |
||||
threading.Thread.__init__(self) |
||||
self.jenkins = jenkins |
||||
|
||||
def run(self): |
||||
httpserver.serve(self.app, host='0.0.0.0', port='8080') |
||||
|
||||
def app(self, environ, start_response): |
||||
request = Request(environ) |
||||
if request.path == '/jenkins_endpoint': |
||||
self.jenkins_endpoint(request) |
||||
start_response('200 OK', [('content-type', 'text/html')]) |
||||
return ['Zuul good.'] |
||||
|
||||
def jenkins_endpoint(self, request): |
||||
data = json.loads(request.body) |
||||
if data: |
||||
self.log.debug("Received data from Jenkins: \n%s" % ( |
||||
pprint.pformat(data))) |
||||
build = data.get('build') |
||||
if build: |
||||
phase = build.get('phase') |
||||
status = build.get('status') |
||||
url = build.get('full_url') |
||||
number = build.get('number') |
||||
params = build.get('parameters') |
||||
if params: |
||||
uuid = params.get('UUID') |
||||
if (status and url and uuid and phase |
||||
and phase == 'COMPLETED'): |
||||
self.jenkins.onBuildCompleted(uuid, status, url, number) |
||||
if (phase and phase == 'STARTED'): |
||||
self.jenkins.onBuildStarted(uuid, url, number) |
||||
|
||||
|
||||
STOP_BUILD = 'job/%(name)s/%(number)s/stop' |
||||
CANCEL_QUEUE = 'queue/item/%(number)s/cancelQueue' |
||||
BUILD_INFO = 'job/%(name)s/%(number)s/api/json?depth=0' |
||||
|
||||
class ExtendedJenkins(jenkins.Jenkins): |
||||
def jenkins_open(self, req): |
||||
''' |
||||
Utility routine for opening an HTTP request to a Jenkins server. |
||||
''' |
||||
try: |
||||
if self.auth: |
||||
req.add_header('Authorization', self.auth) |
||||
return urllib2.urlopen(req).read() |
||||
except urllib2.HTTPError, e: |
||||
print e.msg |
||||
print e.fp.read() |
||||
raise |
||||
|
||||
def stop_build(self, name, number): |
||||
''' |
||||
Stop a running Jenkins build. |
||||
|
||||
@param name: Name of Jenkins job |
||||
@type name: str |
||||
@param number: Jenkins build number for the job |
||||
@type number: int |
||||
''' |
||||
self.jenkins_open(urllib2.Request(self.server + STOP_BUILD%locals())) |
||||
|
||||
def cancel_queue(self, number): |
||||
''' |
||||
Cancel a queued build. |
||||
|
||||
@param number: Jenkins queue number for the build |
||||
@type number: int |
||||
''' |
||||
# Jenkins returns a 302 from this URL, unless Referer is not set, |
||||
# then you get a 404. |
||||
self.jenkins_open(urllib2.Request(self.server + CANCEL_QUEUE%locals(), |
||||
headers={'Referer': self.server})) |
||||
|
||||
|
||||
def get_build_info(self, name, number): |
||||
''' |
||||
Get information for a build. |
||||
|
||||
@param name: Name of Jenkins job |
||||
@type name: str |
||||
@param number: Jenkins build number for the job |
||||
@type number: int |
||||
@return: dictionary |
||||
''' |
||||
return json.loads(self.jenkins_open(urllib2.Request(self.server + BUILD_INFO%locals()))) |
||||
|
||||
class Jenkins(object): |
||||
log = logging.getLogger("zuul.Jenkins") |
||||
|
||||
def __init__(self, config, sched): |
||||
self.sched = sched |
||||
self.builds = {} |
||||
server = config.get('jenkins', 'server') |
||||
user = config.get('jenkins', 'user') |
||||
apikey = config.get('jenkins', 'apikey') |
||||
self.jenkins = ExtendedJenkins(server, user, apikey) |
||||
self.callback_thread = JenkinsCallback(self) |
||||
self.callback_thread.start() |
||||
|
||||
def launch(self, job, change, dependent_changes = []): |
||||
self.log.info("Launch job %s for change %s with dependent changes %s" % ( |
||||
job, change, dependent_changes)) |
||||
uuid = str(uuid1()) |
||||
params = dict(UUID=uuid) |
||||
build = Build(job, uuid) |
||||
self.builds[uuid] = build |
||||
# We can get the started notification on another thread before this is done |
||||
# so we add the build even before we trigger the job on Jenkins. We should |
||||
# be careful to clean it up if it doesn't actually kick off. |
||||
try: |
||||
self.jenkins.build_job(job.name, parameters=params) |
||||
except: |
||||
self.log.exception("Exception launching build %s for job %s for change %s:" % ( |
||||
build, job, change)) |
||||
# Whoops. Remove that build we added. |
||||
del self.builds[uuid] |
||||
raise |
||||
return build |
||||
|
||||
def cancel(self, build): |
||||
self.log.info("Cancel build %s for job %s" % (build, build.job)) |
||||
if build.number: |
||||
self.log.debug("Build %s has already started" % build) |
||||
self.jenkins.stop_build(build.job.name, build.number) |
||||
self.log.debug("Canceled running build %s" % build) |
||||
return |
||||
else: |
||||
self.log.debug("Build %s has not started yet" % build) |
||||
|
||||
self.log.debug("Looking for build %s in queue" % build) |
||||
for item in self.jenkins.get_queue_info(): |
||||
if not item.has_key('actions'): |
||||
continue |
||||
for action in item['actions']: |
||||
if not action.has_key('parameters'): |
||||
continue |
||||
parameters = action['parameters'] |
||||
for param in parameters: |
||||
if (param['name'] == 'UUID' and build.uuid == param['value']): |
||||
self.log.debug("Found queue item %s for build %s" % ( |
||||
item['id'], build)) |
||||
try: |
||||
self.jenkins.cancel_queue(item['id']) |
||||
self.log.debug("Canceled queue item %s for build %s" % ( |
||||
item['id'], build)) |
||||
return |
||||
except: |
||||
self.log.exception("Exception canceling queue item %s for build %s" % ( |
||||
item['id'], build)) |
||||
|
||||
self.log.debug("Still unable to find build %s to cancel" % build) |
||||
if build.number: |
||||
self.log.debug("Build %s has just started" % build) |
||||
self.jenkins.stop_build(build.job.name, build.number) |
||||
self.log.debug("Canceled just running build %s" % build) |
||||
else: |
||||
self.log.error("Build %s has not started but was not found in queue" % build) |
||||
|
||||
def onBuildCompleted(self, uuid, status, url, number): |
||||
self.log.info("Build %s #%s complete, status %s" % ( |
||||
uuid, number, status)) |
||||
build = self.builds.get(uuid) |
||||
if build: |
||||
self.log.debug("Found build %s" % build) |
||||
del self.builds[uuid] |
||||
build.result = status |
||||
build.url = url |
||||
build.number = number |
||||
self.sched.onBuildCompleted(build) |
||||
else: |
||||
self.log.error("Unable to find build %s" % uuid) |
||||
|
||||
def onBuildStarted(self, uuid, url, number): |
||||
self.log.info("Build %s #%s started, url: %s" % (uuid, number, url)) |
||||
build = self.builds.get(uuid) |
||||
if build: |
||||
self.log.debug("Found build %s" % build) |
||||
build.url = url |
||||
build.number = number |
||||
else: |
||||
self.log.error("Unable to find build %s" % uuid) |
||||
@ -0,0 +1,171 @@ |
||||
# Copyright 2012 Hewlett-Packard Development Company, L.P. |
||||
# |
||||
# 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 threading |
||||
import select |
||||
import json |
||||
import time |
||||
import subprocess |
||||
import Queue |
||||
import paramiko |
||||
import logging |
||||
import pprint |
||||
|
||||
# TODO: switch this to paramiko? |
||||
|
||||
class GerritWatcher(threading.Thread): |
||||
log = logging.getLogger("gerrit.GerritWatcher") |
||||
|
||||
def __init__(self, gerrit, username, server, port=29418, keyfile=None): |
||||
threading.Thread.__init__(self) |
||||
self.username = username |
||||
self.keyfile = keyfile |
||||
self.server = server |
||||
self.port = port |
||||
self.proc = None |
||||
self.poll = select.poll() |
||||
self.gerrit = gerrit |
||||
|
||||
def _open(self): |
||||
self.log.debug("Opening ssh connection to %s" % self.server) |
||||
cmd = ['/usr/bin/ssh', '-p', str(self.port)] |
||||
if self.keyfile: |
||||
cmd += ['-i', self.keyfile] |
||||
cmd += ['-l', self.username, self.server, |
||||
'gerrit', 'stream-events'] |
||||
self.proc = subprocess.Popen(cmd, |
||||
bufsize=1, |
||||
stdin=None, |
||||
stdout=subprocess.PIPE, |
||||
stderr=None, |
||||
) |
||||
self.poll.register(self.proc.stdout) |
||||
|
||||
def _close(self): |
||||
self.log.debug("Closing ssh connection") |
||||
try: |
||||
self.poll.unregister(self.proc.stdout) |
||||
except: |
||||
pass |
||||
try: |
||||
self.proc.kill() |
||||
except: |
||||
pass |
||||
self.proc = None |
||||
|
||||
def _read(self): |
||||
l = self.proc.stdout.readline() |
||||
data = json.loads(l) |
||||
self.log.debug("Received data from Gerrit event stream: \n%s" % pprint.pformat(data)) |
||||
self.gerrit.addEvent(data) |
||||
|
||||
def _listen(self): |
||||
while True: |
||||
ret = self.poll.poll() |
||||
for (fd, event) in ret: |
||||
if fd == self.proc.stdout.fileno(): |
||||
if event == select.POLLIN: |
||||
self._read() |
||||
else: |
||||
raise Exception("event on ssh connection") |
||||
|
||||
def _run(self): |
||||
try: |
||||
if not self.proc: |
||||
self._open() |
||||
self._listen() |
||||
except: |
||||
self.log.exception("Exception on ssh event stream:") |
||||
self._close() |
||||
time.sleep(5) |
||||
|
||||
def run(self): |
||||
while True: |
||||
self._run() |
||||
|
||||
class Gerrit(object): |
||||
log = logging.getLogger("gerrit.Gerrit") |
||||
|
||||
def __init__(self, hostname, username, keyfile=None): |
||||
self.username = username |
||||
self.hostname = hostname |
||||
self.keyfile = keyfile |
||||
self.watcher_thread = None |
||||
self.event_queue = None |
||||
|
||||
def startWatching(self): |
||||
self.event_queue = Queue.Queue() |
||||
self.watcher_thread = GerritWatcher( |
||||
self, |
||||
self.username, |
||||
self.hostname, |
||||
keyfile=self.keyfile) |
||||
self.watcher_thread.start() |
||||
|
||||
def addEvent(self, data): |
||||
return self.event_queue.put(data) |
||||
|
||||
def getEvent(self): |
||||
return self.event_queue.get() |
||||
|
||||
def review(self, project, change, message, action={}): |
||||
cmd = 'gerrit review --project %s --message "%s"' % ( |
||||
project, message) |
||||
for k,v in action.items(): |
||||
if v == True: |
||||
cmd += ' --%s' % k |
||||
else: |
||||
cmd += ' --%s %s' % (k, v) |
||||
cmd += ' %s' % change |
||||
out, err = self._ssh(cmd) |
||||
return err |
||||
|
||||
def query(self, change): |
||||
cmd = 'gerrit query --format json %s"' % ( |
||||
change) |
||||
out, err = self._ssh(cmd) |
||||
if not out: |
||||
return False |
||||
lines = out.split('\n') |
||||
if not lines: |
||||
return False |
||||
data = json.loads(lines[0]) |
||||
if not data: |
||||
return False |
||||
self.log.debug("Received data from Gerrit query: \n%s" % ( |
||||
pprint.pformat(data))) |
||||
return data |
||||
|
||||
def _ssh(self, command): |
||||
client = paramiko.SSHClient() |
||||
client.load_system_host_keys() |
||||
client.set_missing_host_key_policy(paramiko.WarningPolicy()) |
||||
client.connect(self.hostname, |
||||
username=self.username, |
||||
port=29418) |
||||
|
||||
self.log.debug("SSH command:\n%s" % command) |
||||
stdin, stdout, stderr = client.exec_command(command) |
||||
|
||||
out = stdout.read() |
||||
self.log.debug("SSH received stdout:\n%s" % out) |
||||
|
||||
ret = stdout.channel.recv_exit_status() |
||||
self.log.debug("SSH exit status: %s" % ret) |
||||
|
||||
err = stderr.read() |
||||
self.log.debug("SSH received stderr:\n%s" % err) |
||||
if ret: |
||||
raise Exception("Gerrit error executing %s" % command) |
||||
return (out, err) |
||||
@ -0,0 +1,323 @@ |
||||
# Copyright 2012 Hewlett-Packard Development Company, L.P. |
||||
# |
||||
# 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 re |
||||
|
||||
class ChangeQueue(object): |
||||
def __init__(self, queue_name): |
||||
self.name = '' |
||||
self.queue_name = queue_name |
||||
self.projects = [] |
||||
self._jobs = set() |
||||
self.queue = [] |
||||
|
||||
def __str__(self): |
||||
return '<ChangeQueue %s: %s>' % (self.queue_name, self.name) |
||||
|
||||
def getJobs(self): |
||||
return self._jobs |
||||
|
||||
def addProject(self, project): |
||||
if project not in self.projects: |
||||
self.projects.append(project) |
||||
names = [x.name for x in self.projects] |
||||
names.sort() |
||||
self.name = ', '.join(names) |
||||
self._jobs |= set(project.getJobs(self.queue_name)) |
||||
|
||||
def enqueueChange(self, change): |
||||
if self.queue: |
||||
self.queue[-1].change_behind = change |
||||
change.change_ahead = self.queue[-1] |
||||
self.queue.append(change) |
||||
change.queue = self |
||||
|
||||
def dequeueChange(self, change): |
||||
if change in self.queue: |
||||
self.queue.remove(change) |
||||
|
||||
def mergeChangeQueue(self, other): |
||||
for project in other.projects: |
||||
self.addProject(project) |
||||
|
||||
class Job(object): |
||||
def __init__(self, name): |
||||
self.name = name |
||||
self.failure_message = None |
||||
self.success_message = None |
||||
self.event_filters = [] |
||||
|
||||
def __str__(self): |
||||
return self.name |
||||
|
||||
def __repr__(self): |
||||
return '<Job %s>' % (self.name) |
||||
|
||||
class Build(object): |
||||
def __init__(self, job, uuid): |
||||
self.job = job |
||||
self.uuid = uuid |
||||
self.status = None |
||||
self.url = None |
||||
self.number = None |
||||
|
||||
def __repr__(self): |
||||
return '<Build %s of %s>' % (self.uuid, self.job.name) |
||||
|
||||
class JobTree(object): |
||||
""" A JobTree represents an instance of one Job, and holds JobTrees |
||||
whose jobs should be run if that Job succeeds. A root node of a |
||||
JobTree will have no associated Job. """ |
||||
|
||||
def __init__(self, job): |
||||
self.job = job |
||||
self.job_trees = [] |
||||
|
||||
def addJob(self, job): |
||||
if job not in [x.job for x in self.job_trees]: |
||||
t = JobTree(job) |
||||
self.job_trees.append(t) |
||||
return t |
||||
|
||||
def getJobs(self): |
||||
jobs = [] |
||||
for x in self.job_trees: |
||||
jobs.append(x.job) |
||||
jobs.extend(x.getJobs()) |
||||
return jobs |
||||
|
||||
def getJobTreeForJob(self, job): |
||||
if self.job == job: |
||||
return self |
||||
for tree in self.job_trees: |
||||
ret = tree.getJobTreeForJob(job) |
||||
if ret: |
||||
return ret |
||||
return None |
||||
|
||||
class Project(object): |
||||
def __init__(self, name): |
||||
self.name = name |
||||
self.job_trees = {} # Queue -> JobTree |
||||
|
||||
def __str__(self): |
||||
return self.name |
||||
|
||||
def __repr__(self): |
||||
return '<Project %s>' % (self.name) |
||||
|
||||
def addQueue(self, name): |
||||
self.job_trees[name] = JobTree(None) |
||||
return self.job_trees[name] |
||||
|
||||
def hasQueue(self, name): |
||||
if self.job_trees.has_key(name): |
||||
return True |
||||
return False |
||||
|
||||
def getJobTreeForQueue(self, name): |
||||
return self.job_trees.get(name, None) |
||||
|
||||
def getJobs(self, queue_name): |
||||
tree = self.getJobTreeForQueue(queue_name) |
||||
if not tree: |
||||
return [] |
||||
return tree.getJobs() |
||||
|
||||
class Change(object): |
||||
def __init__(self, queue_name, project, number, patchset): |
||||
self.queue_name = queue_name |
||||
self.project = project |
||||
self.number = number |
||||
self.patchset = patchset |
||||
self.jobs = {} |
||||
self.job_urls = {} |
||||
self.change_ahead = None |
||||
self.change_behind = None |
||||
self.running_builds = [] |
||||
|
||||
def __str__(self): |
||||
return '<Change 0x%x %s,%s>' % (id(self), self.number, self.patchset) |
||||
|
||||
def formatStatus(self, indent=0): |
||||
indent_str = ' '*indent |
||||
ret = '' |
||||
ret += '%sProject %s change %s,%s\n' % (indent_str, |
||||
self.project.name, |
||||
self.number, |
||||
self.patchset) |
||||
for job in self.project.getJobs(self.queue_name): |
||||
result = self.jobs.get(job.name) |
||||
ret += '%s %s: %s\n' % (indent_str, job.name, result) |
||||
if self.change_ahead: |
||||
ret += '%sWaiting on:\n' % (indent_str) |
||||
ret += self.change_ahead.formatStatus(indent+2) |
||||
return ret |
||||
|
||||
def formatReport(self): |
||||
ret = '' |
||||
if self.didAllJobsSucceed(): |
||||
ret += 'Build successful\n\n' |
||||
else: |
||||
ret += 'Build failed\n\n' |
||||
|
||||
for job in self.project.getJobs(self.queue_name): |
||||
result = self.jobs.get(job.name) |
||||
url = self.job_urls.get(job.name, job.name) |
||||
ret += '- %s : %s\n' % (url, result) |
||||
return ret |
||||
|
||||
def resetAllBuilds(self): |
||||
self.jobs = {} |
||||
self.job_urls = {} |
||||
self.running_builds = [] |
||||
|
||||
def addBuild(self, build): |
||||
self.running_builds.append(build) |
||||
|
||||
def setResult(self, build): |
||||
self.running_builds.remove(build) |
||||
self.jobs[build.job.name] = build.result |
||||
self.job_urls[build.job.name] = build.url |
||||
if build.result != 'SUCCESS': |
||||
# Get a JobTree from a Job so we can find only its dependent jobs |
||||
root = self.project.getJobTreeForQueue(self.queue_name) |
||||
tree = root.getJobTreeForJob(build.job) |
||||
for job in tree.getJobs(): |
||||
self.jobs[job.name] = 'SKIPPED' |
||||
|
||||
def _findJobsToRun(self, job_trees): |
||||
torun = [] |
||||
for tree in job_trees: |
||||
job = tree.job |
||||
if job: |
||||
result = self.jobs.get(job.name, None) |
||||
else: |
||||
# This is a null job tree, run all of its jobs |
||||
result = 'SUCCESS' |
||||
if not result: |
||||
if job not in [b.job for b in self.running_builds]: |
||||
torun.append(job) |
||||
elif result == 'SUCCESS': |
||||
torun.extend(self._findJobsToRun(tree.job_trees)) |
||||
return torun |
||||
|
||||
def findJobsToRun(self): |
||||
tree = self.project.getJobTreeForQueue(self.queue_name) |
||||
return self._findJobsToRun(tree.job_trees) |
||||
|
||||
def areAllJobsComplete(self): |
||||
tree = self.project.getJobTreeForQueue(self.queue_name) |
||||
for job in tree.getJobs(): |
||||
if not self.jobs.has_key(job.name): |
||||
return False |
||||
return True |
||||
|
||||
def didAllJobsSucceed(self): |
||||
for result in self.jobs.values(): |
||||
if result != 'SUCCESS': |
||||
return False |
||||
return True |
||||
|
||||
def delete(self): |
||||
if self.change_behind: |
||||
self.change_behind.change_ahead = None |
||||
|
||||
class TriggerEvent(object): |
||||
def __init__(self): |
||||
self.data = None |
||||
self.type = None |
||||
self.project_name = None |
||||
self.change_number = None |
||||
self.patch_number = None |
||||
self.approvals = [] |
||||
self.branch = None |
||||
self.ref = None |
||||
|
||||
def __str__(self): |
||||
ret = '<TriggerEvent %s %s' % (self.type, self.project_name) |
||||
|
||||
if self.branch: |
||||
ret += " %s" % self.branch |
||||
if self.change_number: |
||||
ret += " %s,%s" % (self.change_number, self.patch_number) |
||||
if self.approvals: |
||||
ret += ' '+', '.join(['%s:%s' % (a['type'], a['value']) for a in self.approvals]) |
||||
ret += '>' |
||||
|
||||
return ret |
||||
|
||||
class EventFilter(object): |
||||
def __init__(self, types=[], branches=[], refs=[], approvals=[]): |
||||
self._types = types |
||||
self._branches = branches |
||||
self._refs = refs |
||||
self.types = [re.compile(x) for x in types] |
||||
self.branches = [re.compile(x) for x in branches] |
||||
self.refs = [re.compile(x) for x in refs] |
||||
self.approvals = approvals |
||||
|
||||
def __str__(self): |
||||
ret = '<EventFilter' |
||||
|
||||
if self._types: |
||||
ret += ' types: %s' % ', '.join(self._types) |
||||
if self._branches: |
||||
ret += ' branches: %s' % ', '.join(self._branches) |
||||
if self._refs: |
||||
ret += ' refs: %s' % ', '.join(self._refs) |
||||
if self.approvals: |
||||
ret += ' approvals: %s' % ', '.join(['%s:%s' % a for a in self.approvals.items()]) |
||||
ret += '>' |
||||
|
||||
return ret |
||||
|
||||
def matches(self, event): |
||||
def normalizeCategory(name): |
||||
name = name.lower() |
||||
return re.sub(' ', '-', name) |
||||
|
||||
# event types are ORed |
||||
matches_type = False |
||||
for etype in self.types: |
||||
if etype.match(event.type): |
||||
matches_type = True |
||||
if self.types and not matches_type: |
||||
return False |
||||
|
||||
# branches are ORed |
||||
matches_branch = False |
||||
for branch in self.branches: |
||||
if branch.match(event.branch): |
||||
matches_branch = True |
||||
if self.branches and not matches_branch: |
||||
return False |
||||
|
||||
# refs are ORed |
||||
matches_ref = False |
||||
for ref in self.refs: |
||||
if ref.match(event.ref): |
||||
matches_ref = True |
||||
if self.refs and not matches_ref: |
||||
return False |
||||
|
||||
# approvals are ANDed |
||||
for category, value in self.approvals.items(): |
||||
matches_approval = False |
||||
for eapproval in event.approvals: |
||||
if (normalizeCategory(eapproval['description']) == category and |
||||
int(eapproval['value']) == int(value)): |
||||
matches_approval = True |
||||
if not matches_approval: return False |
||||
return True |
||||
@ -0,0 +1,436 @@ |
||||
# Copyright 2012 Hewlett-Packard Development Company, L.P. |
||||
# |
||||
# 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 os |
||||
import Queue |
||||
import threading |
||||
import logging |
||||
import yaml |
||||
|
||||
|
||||
from model import Job, Change, Project, ChangeQueue, EventFilter |
||||
|
||||
class Scheduler(object): |
||||
log = logging.getLogger("zuul.Scheduler") |
||||
|
||||
def __init__(self, config): |
||||
self.wake_event = threading.Event() |
||||
self.queue_managers = {} |
||||
self.jobs = {} |
||||
self.projects = {} |
||||
self.launcher = None |
||||
self.trigger = None |
||||
|
||||
self.trigger_event_queue = Queue.Queue() |
||||
self.result_event_queue = Queue.Queue() |
||||
|
||||
self._parseConfig(config.get('zuul', 'layout_config')) |
||||
|
||||
def _parseConfig(self, fp): |
||||
def toList(item): |
||||
if not item: return [] |
||||
if type(item) == type([]): |
||||
return item |
||||
return [item] |
||||
|
||||
if fp: |
||||
fp = os.path.expanduser(fp) |
||||
if not os.path.exists(fp): |
||||
raise Exception("Unable to read layout config file at %s" % fp) |
||||
fp = open(fp) |
||||
data = yaml.load(fp) |
||||
import pprint |
||||
pprint.pprint(data) |
||||
|
||||
for config_queue in data['queue']: |
||||
manager = globals()[config_queue['manager']](self, config_queue['name']) |
||||
self.queue_managers[config_queue['name']] = manager |
||||
manager.success_action = config_queue.get('success') |
||||
manager.failure_action = config_queue.get('failure') |
||||
for trigger in toList(config_queue['trigger']): |
||||
approvals = {} |
||||
for approval_dict in toList(trigger.get('approval')): |
||||
for k, v in approval_dict.items(): |
||||
approvals[k]=v |
||||
f = EventFilter(types=toList(trigger['event']), |
||||
branches=toList(trigger.get('branch')), |
||||
refs=toList(trigger.get('ref')), |
||||
approvals=approvals) |
||||
manager.event_filters.append(f) |
||||
|
||||
for config_job in data['jobs']: |
||||
job = self.getJob(config_job['name']) |
||||
job.failure_message = config_job.get('failure-message', None) |
||||
job.success_message = config_job.get('success-message', None) |
||||
silent = config_job.get('silent', None) |
||||
if silent: |
||||
job.silent = True |
||||
branches = toList(config_job.get('branch')) |
||||
if branches: |
||||
f = EventFilter(branches=branches) |
||||
job.event_filters.append(f) |
||||
|
||||
def add_jobs(job_tree, config_jobs): |
||||
for job in config_jobs: |
||||
if isinstance(job, list): |
||||
for x in job: |
||||
add_jobs(job_tree, x) |
||||
if isinstance(job, dict): |
||||
for parent, children in job.items(): |
||||
parent_tree = job_tree.addJob(self.getJob(parent)) |
||||
add_jobs(parent_tree, children) |
||||
if isinstance(job, str): |
||||
job_tree.addJob(self.getJob(job)) |
||||
|
||||
for config_project in data['projects']: |
||||
project = Project(config_project['name']) |
||||
self.projects[config_project['name']] = project |
||||
for qname in self.queue_managers.keys(): |
||||
if config_project.has_key(qname): |
||||
print project, qname |
||||
job_tree = project.addQueue(qname) |
||||
config_jobs = config_project[qname] |
||||
add_jobs(job_tree, config_jobs) |
||||
|
||||
|
||||
# TODO(jeblair): check that we don't end up with jobs like |
||||
# "foo - bar" because a ':' is missing in the yaml for a dependent job |
||||
for manager in self.queue_managers.values(): |
||||
manager._postConfig() |
||||
|
||||
def getJob(self, name): |
||||
if self.jobs.has_key(name): |
||||
return self.jobs[name] |
||||
job = Job(name) |
||||
self.jobs[name] = job |
||||
return job |
||||
|
||||
def setLauncher(self, launcher): |
||||
self.launcher = launcher |
||||
|
||||
def setTrigger(self, trigger): |
||||
self.trigger = trigger |
||||
|
||||
def addEvent(self, event): |
||||
self.log.debug("Adding trigger event: %s" % event) |
||||
self.trigger_event_queue.put(event) |
||||
self.wake_event.set() |
||||
|
||||
def onBuildCompleted(self, build): |
||||
self.log.debug("Adding result event for build: %s" % build) |
||||
self.result_event_queue.put(build) |
||||
self.wake_event.set() |
||||
|
||||
def run(self): |
||||
while True: |
||||
self.log.debug("Run handler sleeping") |
||||
self.wake_event.wait() |
||||
self.wake_event.clear() |
||||
self.log.debug("Run handler awake") |
||||
try: |
||||
if not self.trigger_event_queue.empty(): |
||||
self.process_event_queue() |
||||
if not self.result_event_queue.empty(): |
||||
self.process_result_queue() |
||||
except: |
||||
self.log.exception("Exception in run handler:") |
||||
|
||||
def process_event_queue(self): |
||||
self.log.debug("Fetching trigger event") |
||||
event = self.trigger_event_queue.get() |
||||
self.log.debug("Processing trigger event %s" % event) |
||||
project = self.projects.get(event.project_name) |
||||
if not project: |
||||
self.log.warning("Project %s not found" % event.project_name) |
||||
return |
||||
|
||||
for manager in self.queue_managers.values(): |
||||
if not manager.eventMatches(event): |
||||
self.log.debug("Event %s ignored by %s" % (event, manager)) |
||||
continue |
||||
if event.change_number: |
||||
change = Change(manager.name, project, |
||||
event.change_number, event.patch_number) |
||||
self.log.info("Adding %s, %s to to %s" % ( |
||||
project, change, manager)) |
||||
manager.addChange(change) |
||||
|
||||
def process_result_queue(self): |
||||
self.log.debug("Fetching result event") |
||||
build = self.result_event_queue.get() |
||||
self.log.debug("Processing result event %s" % build) |
||||
for manager in self.queue_managers.values(): |
||||
if manager.onBuildCompleted(build): |
||||
return |
||||
|
||||
class BaseQueueManager(object): |
||||
log = logging.getLogger("zuul.BaseQueueManager") |
||||
|
||||
def __init__(self, sched, name): |
||||
self.sched = sched |
||||
self.name = name |
||||
self.building_jobs = {} |
||||
self.event_filters = [] |
||||
self.success_action = {} |
||||
self.failure_action = {} |
||||
|
||||
def __str__(self): |
||||
return "<%s %s>" % (self.__class__.__name__, self.name) |
||||
|
||||
def _postConfig(self): |
||||
self.log.info("Configured Queue Manager %s" % self.name) |
||||
self.log.info(" Events:") |
||||
for e in self.event_filters: |
||||
self.log.info(" %s" % e) |
||||
self.log.info(" Projects:") |
||||
def log_jobs(tree, indent=0): |
||||
istr = ' '+' '*indent |
||||
if tree.job: |
||||
efilters = '' |
||||
for e in tree.job.event_filters: |
||||
efilters += str(e) |
||||
if efilters: |
||||
efilters = ' '+efilters |
||||
self.log.info("%s%s%s" % (istr, repr(tree.job), efilters)) |
||||
for x in tree.job_trees: |
||||
log_jobs(x, indent+2) |
||||
for p in self.sched.projects.values(): |
||||
if p.hasQueue(self.name): |
||||
self.log.info(" %s" % p) |
||||
log_jobs(p.getJobTreeForQueue(self.name)) |
||||
if self.success_action: |
||||
self.log.info(" On success:") |
||||
self.log.info(" %s" % self.success_action) |
||||
if self.failure_action: |
||||
self.log.info(" On failure:") |
||||
self.log.info(" %s" % self.failure_action) |
||||
|
||||
|
||||
def eventMatches(self, event): |
||||
for ef in self.event_filters: |
||||
print ef |
||||
if ef.matches(event): |
||||
return True |
||||
return False |
||||
|
||||
|
||||
def addChange(self, change): |
||||
self.log.debug("Adding change %s" % change) |
||||
self.launchJobs(change) |
||||
|
||||
def launchJobs(self, change): |
||||
self.log.debug("Launching jobs for change %s" % change) |
||||
for job in change.findJobsToRun(): |
||||
self.log.debug("Found job %s for change %s" % (job, change)) |
||||
try: |
||||
build = self.sched.launcher.launch(job, change) |
||||
self.building_jobs[build] = change |
||||
self.log.debug("Adding build %s of job %s to change %s" % ( |
||||
build, job, change)) |
||||
change.addBuild(build) |
||||
except: |
||||
self.log.exception("Exception while launching job %s for change %s:" % ( |
||||
job, change)) |
||||
|
||||
def onBuildCompleted(self, build): |
||||
self.log.debug("Build %s completed" % build) |
||||
if not self.building_jobs.has_key(build): |
||||
self.log.warning("Build %s not found (may have been canceled)" % ( |
||||
build)) |
||||
# Or triggered externally, or triggered before zuul started, |
||||
# or restarted |
||||
return False |
||||
change = self.building_jobs[build] |
||||
self.log.debug("Found change %s which triggered completed build %s" % ( |
||||
change, build)) |
||||
|
||||
del self.building_jobs[build] |
||||
|
||||
change.setResult(build) |
||||
self.log.info("Change %s status is now:\n %s" % ( |
||||
change, change.formatStatus())) |
||||
|
||||
if change.areAllJobsComplete(): |
||||
self.log.debug("All jobs for change %s are complete" % change) |
||||
self.possiblyReportChange(change) |
||||
else: |
||||
# There may be jobs that depended on jobs that are now complete |
||||
self.log.debug("All jobs for change %s are not yet complete" % ( |
||||
change)) |
||||
self.launchJobs(change) |
||||
return True |
||||
|
||||
def possiblyReportChange(self, change): |
||||
self.log.debug("Possibly reporting change %s" % change) |
||||
self.reportChange(change) |
||||
|
||||
def reportChange(self, change): |
||||
self.log.debug("Reporting change %s" % change) |
||||
ret = None |
||||
if change.didAllJobsSucceed(): |
||||
action = self.success_action |
||||
else: |
||||
action = self.failure_action |
||||
try: |
||||
self.log.info("Reporting change %s, action: %s" % ( |
||||
change, action)) |
||||
ret = self.sched.trigger.report(change, change.formatReport(), |
||||
action) |
||||
if ret: |
||||
self.log.error("Reporting change %s received: %s" % ( |
||||
change, ret)) |
||||
print ret |
||||
except: |
||||
self.log.exception("Exception while reporting:") |
||||
return ret |
||||
|
||||
class IndependentQueueManager(BaseQueueManager): |
||||
log = logging.getLogger("zuul.IndependentQueueManager") |
||||
pass |
||||
|
||||
class DependentQueueManager(BaseQueueManager): |
||||
log = logging.getLogger("zuul.DependentQueueManager") |
||||
|
||||
def __init__(self, *args, **kwargs): |
||||
super(DependentQueueManager, self).__init__(*args, **kwargs) |
||||
self.change_queues = [] |
||||
|
||||
def _postConfig(self): |
||||
super(DependentQueueManager, self)._postConfig() |
||||
self.buildChangeQueues() |
||||
|
||||
def buildChangeQueues(self): |
||||
self.log.debug("Building shared change queues") |
||||
change_queues = [] |
||||
|
||||
for project in self.sched.projects.values(): |
||||
if project.hasQueue(self.name): |
||||
change_queue = ChangeQueue(self.name) |
||||
change_queue.addProject(project) |
||||
change_queues.append(change_queue) |
||||
self.log.debug("Created queue: %s" % change_queue) |
||||
|
||||
self.log.debug("Combining shared queues") |
||||
new_change_queues = [] |
||||
for a in change_queues: |
||||
merged_a = False |
||||
for b in new_change_queues: |
||||
if not a.getJobs().isdisjoint(b.getJobs()): |
||||
self.log.debug("Merging queue %s into %s" % (a, b)) |
||||
b.mergeChangeQueue(a) |
||||
merged_a = True |
||||
break # this breaks out of 'for b' and continues 'for a' |
||||
if not merged_a: |
||||
self.log.debug("Keeping queue %s" % (a)) |
||||
new_change_queues.append(a) |
||||
|
||||
self.change_queues = new_change_queues |
||||
self.log.info(" Shared change queues:") |
||||
for x in self.change_queues: |
||||
self.log.info(" %s" % x) |
||||
|
||||
def getQueue(self, project): |
||||
for queue in self.change_queues: |
||||
if project in queue.projects: |
||||
return queue |
||||
self.log.error("Unable to find change queue for project %s" % project) |
||||
|
||||
def addChange(self, change): |
||||
self.log.debug("Adding change %s" % change) |
||||
change_queue = self.getQueue(change.project) |
||||
self.log.debug("Adding change %s to queue %s" % (change, change_queue)) |
||||
change_queue.enqueueChange(change) |
||||
super(DependentQueueManager, self).addChange(change) |
||||
|
||||
def _getDependentChanges(self, change): |
||||
changes = [] |
||||
while change.change_ahead: |
||||
changes.append(change.change_ahead) |
||||
change = change.change_ahead |
||||
self.log.info("Change %s depends on changes %s" % (change, changes)) |
||||
return changes |
||||
|
||||
def launchJobs(self, change): |
||||
self.log.debug("Launching jobs for change %s" % change) |
||||
dependent_changes = self._getDependentChanges(change) |
||||
for job in change.findJobsToRun(): |
||||
self.log.debug("Found job %s for change %s" % (job, change)) |
||||
try: |
||||
build = self.sched.launcher.launch(job, change, |
||||
dependent_changes) |
||||
self.building_jobs[build] = change |
||||
self.log.debug("Adding build %s of job %s to change %s" % ( |
||||
build, job, change)) |
||||
change.addBuild(build) |
||||
except: |
||||
self.log.exception("Exception while launching job %s for change %s:" % ( |
||||
job, change)) |
||||
if change.change_behind: |
||||
self.log.debug("Launching jobs for change %s, behind change %s" % ( |
||||
change.change_behind, change)) |
||||
self.launchJobs(change.change_behind) |
||||
|
||||
def cancelJobs(self, change): |
||||
self.log.debug("Cancel jobs for change %s" % change) |
||||
to_remove = [] |
||||
change.resetAllBuilds() |
||||
for build, build_change in self.building_jobs.items(): |
||||
if build_change == change: |
||||
self.log.debug("Found build %s for change %s to cancel" % ( |
||||
build, change)) |
||||
try: |
||||
self.sched.launcher.cancel(build) |
||||
except: |
||||
self.log.exception("Exception while canceling build %s for change %s" % ( |
||||
build, change)) |
||||
to_remove.append(build) |
||||
for build in to_remove: |
||||
self.log.debug("Removing build %s from running builds" % build) |
||||
del self.building_jobs[build] |
||||
if change.change_behind: |
||||
self.log.debug("Canceling jobs for change %s, behind change %s" % ( |
||||
change.change_behind, change)) |
||||
self.cancelJobs(change.change_behind) |
||||
|
||||
def possiblyReportChange(self, change): |
||||
self.log.debug("Possibly reporting change %s" % change) |
||||
if not change.change_ahead: |
||||
self.log.debug("Change %s is at the front of the queue, reporting" % ( |
||||
change)) |
||||
ret = self.reportChange(change) |
||||
self.log.debug("Removing reported change %s from queue" % change) |
||||
change.delete() |
||||
change.queue.dequeueChange(change) |
||||
merged = (not ret) |
||||
if merged: |
||||
merged = self.sched.trigger.isMerged(change) |
||||
succeeded = change.didAllJobsSucceed() |
||||
self.log.info("Reported change %s status: all-succeeded: %s, merged: %s" % ( |
||||
change, succeeded, merged)) |
||||
|
||||
if not (succeeded and merged): |
||||
self.log.debug("Reported change %s failed tests or failed to merge" % ( |
||||
change)) |
||||
# The merge or test failed, re-run all jobs behind this one |
||||
if change.change_behind: |
||||
self.log.info("Canceling/relaunching jobs for change %s behind failed change %s" % ( |
||||
change.change_behind, change)) |
||||
self.cancelJobs(change.change_behind) |
||||
self.launchJobs(change.change_behind) |
||||
# If the change behind this is ready, notify |
||||
if (change.change_behind and |
||||
change.change_behind.areAllJobsComplete()): |
||||
self.log.info("Change %s behind change %s is ready, possibly reporting" % ( |
||||
change.change_behind, change)) |
||||
self.possiblyReportChange(change.change_behind) |
||||
@ -0,0 +1,92 @@ |
||||
# Copyright 2012 Hewlett-Packard Development Company, L.P. |
||||
# |
||||
# 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 threading |
||||
import logging |
||||
from zuul.lib import gerrit |
||||
from zuul.model import TriggerEvent |
||||
|
||||
class GerritEventConnector(threading.Thread): |
||||
"""Move events from Gerrit to the scheduler.""" |
||||
|
||||
log = logging.getLogger("zuul.GerritEventConnector") |
||||
|
||||
def __init__(self, gerrit, sched): |
||||
super(GerritEventConnector, self).__init__() |
||||
self.gerrit = gerrit |
||||
self.sched = sched |
||||
|
||||
|
||||
def _handleEvent(self): |
||||
data = self.gerrit.getEvent() |
||||
event = TriggerEvent() |
||||
event.type = data.get('type') |
||||
change = data.get('change') |
||||
if change: |
||||
event.project_name = change.get('project') |
||||
event.branch = change.get('branch') |
||||
event.change_number = change.get('number') |
||||
patchset = data.get('patchSet') |
||||
if patchset: |
||||
event.patch_number = patchset.get('number') |
||||
event.approvals = data.get('approvals') |
||||
self.sched.addEvent(event) |
||||
|
||||
|
||||
def run(self): |
||||
while True: |
||||
try: |
||||
self._handleEvent() |
||||
except: |
||||
self.log.exception("Exception moving Gerrit event:") |
||||
|
||||
|
||||
class Gerrit(object): |
||||
log = logging.getLogger("zuul.Gerrit") |
||||
|
||||
def __init__(self, config, sched): |
||||
self.sched = sched |
||||
server = config.get('gerrit', 'server') |
||||
user = config.get('gerrit', 'user') |
||||
if config.has_option('gerrit', 'sshkey'): |
||||
sshkey = config.get('gerrit', 'sshkey') |
||||
else: |
||||
sshkey = None |
||||
self.gerrit = gerrit.Gerrit(server, user, sshkey) |
||||
self.gerrit.startWatching() |
||||
self.gerrit_connector = GerritEventConnector( |
||||
self.gerrit, sched) |
||||
self.gerrit_connector.start() |
||||
|
||||
|
||||
def report(self, change, message, action): |
||||
self.log.debug("Report change %s, action %s, message: %s" % |
||||
(change, action, message)) |
||||
changeid = '%s,%s' % (change.number, change.patchset) |
||||
return self.gerrit.review(change.project.name, changeid, |
||||
message, action) |
||||
|
||||
|
||||
def isMerged(self, change): |
||||
self.log.debug("Checking if change %s is merged", change) |
||||
data = self.gerrit.query(change.number) |
||||
if not data: |
||||
return False |
||||
status = data.get('status') |
||||
if not status: |
||||
return False |
||||
self.log.debug("Change %s status: %s" % (change, status)) |
||||
if status == 'MERGED' or status == 'SUBMITTED': |
||||
return True |
||||
|
||||
Loading…
Reference in new issue