1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
|
# This file is part of Buildbot. Buildbot 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.
#
# 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.
#
# Copyright Buildbot Team Members
from twisted.internet import defer
from twisted.python import log
from buildbot import util, interfaces, config
from buildbot.status.results import SUCCESS, WARNINGS
from buildbot.schedulers import base
class Dependent(base.BaseScheduler):
compare_attrs = base.BaseScheduler.compare_attrs + ('upstream_name',)
def __init__(self, name, upstream, builderNames, properties={}, **kwargs):
base.BaseScheduler.__init__(self, name, builderNames, properties,
**kwargs)
if not interfaces.IScheduler.providedBy(upstream):
config.error(
"upstream must be another Scheduler instance")
self.upstream_name = upstream.name
self._buildset_addition_subscr = None
self._buildset_completion_subscr = None
self._cached_upstream_bsids = None
# the subscription lock makes sure that we're done inserting a
# subcription into the DB before registering that the buildset is
# complete.
self._subscription_lock = defer.DeferredLock()
def startService(self):
self._buildset_addition_subscr = \
self.master.subscribeToBuildsets(self._buildsetAdded)
self._buildset_completion_subscr = \
self.master.subscribeToBuildsetCompletions(self._buildsetCompleted)
# check for any buildsets completed before we started
d = self._checkCompletedBuildsets(None, None)
d.addErrback(log.err, 'while checking for completed buildsets in start')
def stopService(self):
if self._buildset_addition_subscr:
self._buildset_addition_subscr.unsubscribe()
if self._buildset_completion_subscr:
self._buildset_completion_subscr.unsubscribe()
self._cached_upstream_bsids = None
return defer.succeed(None)
@util.deferredLocked('_subscription_lock')
def _buildsetAdded(self, bsid=None, properties=None, **kwargs):
# check if this was submitetted by our upstream by checking the
# scheduler property
submitter = properties.get('scheduler', (None, None))[0]
if submitter != self.upstream_name:
return
# record our interest in this buildset
d = self._addUpstreamBuildset(bsid)
d.addErrback(log.err, 'while subscribing to buildset %d' % bsid)
def _buildsetCompleted(self, bsid, result):
d = self._checkCompletedBuildsets(bsid, result)
d.addErrback(log.err, 'while checking for completed buildsets')
@util.deferredLocked('_subscription_lock')
@defer.inlineCallbacks
def _checkCompletedBuildsets(self, bsid, result):
subs = yield self._getUpstreamBuildsets()
sub_bsids = []
for (sub_bsid, sub_sssetid, sub_complete, sub_results) in subs:
# skip incomplete builds, handling the case where the 'complete'
# column has not been updated yet
if not sub_complete and sub_bsid != bsid:
continue
# build a dependent build if the status is appropriate
if sub_results in (SUCCESS, WARNINGS):
yield self.addBuildsetForSourceStamp(setid=sub_sssetid,
reason='downstream')
sub_bsids.append(sub_bsid)
# and regardless of status, remove the subscriptions
yield self._removeUpstreamBuildsets(sub_bsids)
@defer.inlineCallbacks
def _updateCachedUpstreamBuilds(self):
if self._cached_upstream_bsids is None:
bsids = yield self.master.db.state.getState(self.objectid,
'upstream_bsids', [])
self._cached_upstream_bsids = bsids
@defer.inlineCallbacks
def _getUpstreamBuildsets(self):
# get a list of (bsid, sssid, complete, results) for all
# upstream buildsets
yield self._updateCachedUpstreamBuilds()
changed = False
rv = []
for bsid in self._cached_upstream_bsids[:]:
bsdict = yield self.master.db.buildsets.getBuildset(bsid)
if not bsdict:
self._cached_upstream_bsids.remove(bsid)
changed = True
continue
rv.append((bsid, bsdict['sourcestampsetid'], bsdict['complete'],
bsdict['results']))
if changed:
yield self.master.db.state.setState(self.objectid,
'upstream_bsids', self._cached_upstream_bsids)
defer.returnValue(rv)
@defer.inlineCallbacks
def _addUpstreamBuildset(self, bsid):
yield self._updateCachedUpstreamBuilds()
if bsid not in self._cached_upstream_bsids:
self._cached_upstream_bsids.append(bsid)
yield self.master.db.state.setState(self.objectid,
'upstream_bsids', self._cached_upstream_bsids)
@defer.inlineCallbacks
def _removeUpstreamBuildsets(self, bsids):
yield self._updateCachedUpstreamBuilds()
old = set(self._cached_upstream_bsids)
self._cached_upstream_bsids = list(old - set(bsids))
yield self.master.db.state.setState(self.objectid,
'upstream_bsids', self._cached_upstream_bsids)
|