yocto-autobuilder2/steps/observer.py
Richard Purdie 3d960690e4 steps/observer: Update to work with new buildbot versions
Signed-off-by: Richard Purdie <richard.purdie@linuxfoundation.org>
2023-05-02 09:59:14 +01:00

94 lines
3.0 KiB
Python

#
# SPDX-License-Identifier: GPL-2.0-only
#
from twisted.internet import defer
from twisted.python import log
from buildbot.process import logobserver
from buildbot.process.results import FAILURE
from buildbot.process.results import SKIPPED
from buildbot.process.results import SUCCESS
from buildbot.process.results import WARNINGS
from buildbot.steps.shell import ShellCommand
from functools import partial
#
# Monitor the step 1-X logs and stdio, collecting up any warnings and errors seen
# and publish them at the end in their own 'logfile' for ease of access to the user
#
class SimpleLogObserver(ShellCommand):
warnOnWarnings = True
warnOnFailure = True
warnCount = 0
errorCount = 0
def __init__(self, maxsteps=10, *args, **kwargs):
super().__init__(*args, **kwargs)
self.warningLines = []
self.errorLines = []
if "description" in kwargs:
self.description = kwargs["description"]
else:
self.description = "run-config"
self.addLogObserver('stdio', logobserver.LineConsumerLogObserver(partial(self.logConsumer, 'stdio')))
def describe(self, done=False):
return self.description
def logConsumer(self, logname):
while True:
stream, line = yield
if line.startswith("WARNING:"):
self.warnCount += 1
self.warningLines.append(logname + ": " + line)
if line.startswith("ERROR:"):
self.errorCount += 1
self.errorLines.append(logname + ": " + line)
@defer.inlineCallbacks
def finish_logs(self):
stdio_log = yield self.getLog('stdio')
yield stdio_log.finish()
@defer.inlineCallbacks
def createSummary(self):
if self.warnCount:
yield self.addCompleteLog('warnings', '\n'.join(self.warningLines) + '\n')
if self.errorLines:
yield self.addCompleteLog('errors', '\n'.join(self.errorLines) + '\n')
warnings_stat = self.getStatistic('warnings', 0)
self.setStatistic('warnings', warnings_stat + self.warnCount)
old_count = self.getProperty("warnings-count", 0)
self.setProperty(
"warnings-count", old_count + self.warnCount, "SimpleLogObserver")
@defer.inlineCallbacks
def run(self):
cmd = yield self.makeRemoteShellCommand()
yield self.runCommand(cmd)
yield self.finish_logs()
yield self.createSummary()
return self.evaluateCommand(cmd)
def evaluateCommand(self, cmd):
if cmd.didFail() or self.errorCount:
return FAILURE
if self.warnCount:
return WARNINGS
return SUCCESS
class RunConfigLogObserver(SimpleLogObserver):
def __init__(self, maxsteps=10, *args, **kwargs):
super().__init__(*args, **kwargs)
for i in range(1, maxsteps):
for j in ['a', 'b', 'c', 'd']:
name = 'step' + str(i) + str(j)
self.addLogObserver(name, logobserver.LineConsumerLogObserver(partial(self.logConsumer, name)))