Brad Bishop | c342db3 | 2019-05-15 21:57:59 -0400 | [diff] [blame] | 1 | # |
| 2 | # SPDX-License-Identifier: GPL-2.0-only |
| 3 | # |
| 4 | |
Brad Bishop | d7bf8c1 | 2018-02-25 22:55:05 -0500 | [diff] [blame] | 5 | from django.core.management.base import BaseCommand |
Patrick Williams | c124f4f | 2015-09-15 14:41:29 -0500 | [diff] [blame] | 6 | from django.db import transaction |
Patrick Williams | d8c66bc | 2016-06-20 12:57:21 -0500 | [diff] [blame] | 7 | from django.db.models import Q |
| 8 | |
| 9 | from bldcontrol.bbcontroller import getBuildEnvironmentController |
Patrick Williams | d8c66bc | 2016-06-20 12:57:21 -0500 | [diff] [blame] | 10 | from bldcontrol.models import BuildRequest, BuildEnvironment |
| 11 | from bldcontrol.models import BRError, BRVariable |
| 12 | |
Patrick Williams | c0f7c04 | 2017-02-23 20:41:17 -0600 | [diff] [blame] | 13 | from orm.models import Build, LogMessage, Target |
Patrick Williams | d8c66bc | 2016-06-20 12:57:21 -0500 | [diff] [blame] | 14 | |
Patrick Williams | c124f4f | 2015-09-15 14:41:29 -0500 | [diff] [blame] | 15 | import logging |
Patrick Williams | d8c66bc | 2016-06-20 12:57:21 -0500 | [diff] [blame] | 16 | import traceback |
Patrick Williams | c0f7c04 | 2017-02-23 20:41:17 -0600 | [diff] [blame] | 17 | import signal |
Brad Bishop | 6e60e8b | 2018-02-01 10:27:11 -0500 | [diff] [blame] | 18 | import os |
Patrick Williams | c124f4f | 2015-09-15 14:41:29 -0500 | [diff] [blame] | 19 | |
Patrick Williams | d8c66bc | 2016-06-20 12:57:21 -0500 | [diff] [blame] | 20 | logger = logging.getLogger("toaster") |
Patrick Williams | c124f4f | 2015-09-15 14:41:29 -0500 | [diff] [blame] | 21 | |
Brad Bishop | 6e60e8b | 2018-02-01 10:27:11 -0500 | [diff] [blame] | 22 | |
Brad Bishop | d7bf8c1 | 2018-02-25 22:55:05 -0500 | [diff] [blame] | 23 | class Command(BaseCommand): |
Patrick Williams | c0f7c04 | 2017-02-23 20:41:17 -0600 | [diff] [blame] | 24 | args = "" |
| 25 | help = "Schedules and executes build requests as possible. "\ |
| 26 | "Does not return (interrupt with Ctrl-C)" |
Patrick Williams | c124f4f | 2015-09-15 14:41:29 -0500 | [diff] [blame] | 27 | |
Patrick Williams | d8c66bc | 2016-06-20 12:57:21 -0500 | [diff] [blame] | 28 | @transaction.atomic |
Patrick Williams | c124f4f | 2015-09-15 14:41:29 -0500 | [diff] [blame] | 29 | def _selectBuildEnvironment(self): |
Patrick Williams | c0f7c04 | 2017-02-23 20:41:17 -0600 | [diff] [blame] | 30 | bec = getBuildEnvironmentController(lock=BuildEnvironment.LOCK_FREE) |
Patrick Williams | c124f4f | 2015-09-15 14:41:29 -0500 | [diff] [blame] | 31 | bec.be.lock = BuildEnvironment.LOCK_LOCK |
| 32 | bec.be.save() |
| 33 | return bec |
| 34 | |
Patrick Williams | d8c66bc | 2016-06-20 12:57:21 -0500 | [diff] [blame] | 35 | @transaction.atomic |
Patrick Williams | c124f4f | 2015-09-15 14:41:29 -0500 | [diff] [blame] | 36 | def _selectBuildRequest(self): |
Patrick Williams | d8c66bc | 2016-06-20 12:57:21 -0500 | [diff] [blame] | 37 | br = BuildRequest.objects.filter(state=BuildRequest.REQ_QUEUED).first() |
Patrick Williams | c124f4f | 2015-09-15 14:41:29 -0500 | [diff] [blame] | 38 | return br |
| 39 | |
| 40 | def schedule(self): |
Patrick Williams | c124f4f | 2015-09-15 14:41:29 -0500 | [diff] [blame] | 41 | try: |
Patrick Williams | d8c66bc | 2016-06-20 12:57:21 -0500 | [diff] [blame] | 42 | # select the build environment and the request to build |
| 43 | br = self._selectBuildRequest() |
| 44 | if br: |
| 45 | br.state = BuildRequest.REQ_INPROGRESS |
| 46 | br.save() |
| 47 | else: |
Patrick Williams | c124f4f | 2015-09-15 14:41:29 -0500 | [diff] [blame] | 48 | return |
Patrick Williams | d8c66bc | 2016-06-20 12:57:21 -0500 | [diff] [blame] | 49 | |
Patrick Williams | c124f4f | 2015-09-15 14:41:29 -0500 | [diff] [blame] | 50 | try: |
| 51 | bec = self._selectBuildEnvironment() |
| 52 | except IndexError as e: |
| 53 | # we could not find a BEC; postpone the BR |
| 54 | br.state = BuildRequest.REQ_QUEUED |
| 55 | br.save() |
Brad Bishop | 1a4b7ee | 2018-12-16 17:11:34 -0800 | [diff] [blame] | 56 | logger.debug("runbuilds: No build env (%s)" % e) |
Patrick Williams | c124f4f | 2015-09-15 14:41:29 -0500 | [diff] [blame] | 57 | return |
| 58 | |
Brad Bishop | 6e60e8b | 2018-02-01 10:27:11 -0500 | [diff] [blame] | 59 | logger.info("runbuilds: starting build %s, environment %s" % |
Patrick Williams | c0f7c04 | 2017-02-23 20:41:17 -0600 | [diff] [blame] | 60 | (br, bec.be)) |
Patrick Williams | c124f4f | 2015-09-15 14:41:29 -0500 | [diff] [blame] | 61 | |
| 62 | # let the build request know where it is being executed |
| 63 | br.environment = bec.be |
| 64 | br.save() |
| 65 | |
| 66 | # this triggers an async build |
Patrick Williams | d8c66bc | 2016-06-20 12:57:21 -0500 | [diff] [blame] | 67 | bec.triggerBuild(br.brbitbake, br.brlayer_set.all(), |
| 68 | br.brvariable_set.all(), br.brtarget_set.all(), |
| 69 | "%d:%d" % (br.pk, bec.be.pk)) |
Patrick Williams | c124f4f | 2015-09-15 14:41:29 -0500 | [diff] [blame] | 70 | |
| 71 | except Exception as e: |
| 72 | logger.error("runbuilds: Error launching build %s" % e) |
Patrick Williams | c0f7c04 | 2017-02-23 20:41:17 -0600 | [diff] [blame] | 73 | traceback.print_exc() |
Patrick Williams | c124f4f | 2015-09-15 14:41:29 -0500 | [diff] [blame] | 74 | if "[Errno 111] Connection refused" in str(e): |
| 75 | # Connection refused, read toaster_server.out |
| 76 | errmsg = bec.readServerLogFile() |
| 77 | else: |
| 78 | errmsg = str(e) |
| 79 | |
Patrick Williams | c0f7c04 | 2017-02-23 20:41:17 -0600 | [diff] [blame] | 80 | BRError.objects.create(req=br, errtype=str(type(e)), errmsg=errmsg, |
| 81 | traceback=traceback.format_exc()) |
Patrick Williams | c124f4f | 2015-09-15 14:41:29 -0500 | [diff] [blame] | 82 | br.state = BuildRequest.REQ_FAILED |
| 83 | br.save() |
| 84 | bec.be.lock = BuildEnvironment.LOCK_FREE |
| 85 | bec.be.save() |
Brad Bishop | d7bf8c1 | 2018-02-25 22:55:05 -0500 | [diff] [blame] | 86 | # Cancel the pending build and report the exception to the UI |
| 87 | log_object = LogMessage.objects.create( |
| 88 | build = br.build, |
| 89 | level = LogMessage.EXCEPTION, |
| 90 | message = errmsg) |
| 91 | log_object.save() |
| 92 | br.build.outcome = Build.FAILED |
| 93 | br.build.save() |
Patrick Williams | c124f4f | 2015-09-15 14:41:29 -0500 | [diff] [blame] | 94 | |
| 95 | def archive(self): |
Patrick Williams | c0f7c04 | 2017-02-23 20:41:17 -0600 | [diff] [blame] | 96 | for br in BuildRequest.objects.filter(state=BuildRequest.REQ_ARCHIVE): |
Brad Bishop | 6e60e8b | 2018-02-01 10:27:11 -0500 | [diff] [blame] | 97 | if br.build is None: |
Patrick Williams | c124f4f | 2015-09-15 14:41:29 -0500 | [diff] [blame] | 98 | br.state = BuildRequest.REQ_FAILED |
Patrick Williams | d7e9631 | 2015-09-22 08:09:05 -0500 | [diff] [blame] | 99 | else: |
| 100 | br.state = BuildRequest.REQ_COMPLETED |
Patrick Williams | c124f4f | 2015-09-15 14:41:29 -0500 | [diff] [blame] | 101 | br.save() |
| 102 | |
| 103 | def cleanup(self): |
| 104 | from django.utils import timezone |
| 105 | from datetime import timedelta |
Patrick Williams | d8c66bc | 2016-06-20 12:57:21 -0500 | [diff] [blame] | 106 | # environments locked for more than 30 seconds |
| 107 | # they should be unlocked |
| 108 | BuildEnvironment.objects.filter( |
| 109 | Q(buildrequest__state__in=[BuildRequest.REQ_FAILED, |
| 110 | BuildRequest.REQ_COMPLETED, |
| 111 | BuildRequest.REQ_CANCELLING]) & |
| 112 | Q(lock=BuildEnvironment.LOCK_LOCK) & |
Patrick Williams | c0f7c04 | 2017-02-23 20:41:17 -0600 | [diff] [blame] | 113 | Q(updated__lt=timezone.now() - timedelta(seconds=30)) |
Patrick Williams | d8c66bc | 2016-06-20 12:57:21 -0500 | [diff] [blame] | 114 | ).update(lock=BuildEnvironment.LOCK_FREE) |
Patrick Williams | c124f4f | 2015-09-15 14:41:29 -0500 | [diff] [blame] | 115 | |
Patrick Williams | d8c66bc | 2016-06-20 12:57:21 -0500 | [diff] [blame] | 116 | # update all Builds that were in progress and failed to start |
Brad Bishop | 6e60e8b | 2018-02-01 10:27:11 -0500 | [diff] [blame] | 117 | for br in BuildRequest.objects.filter( |
| 118 | state=BuildRequest.REQ_FAILED, |
| 119 | build__outcome=Build.IN_PROGRESS): |
Patrick Williams | c124f4f | 2015-09-15 14:41:29 -0500 | [diff] [blame] | 120 | # transpose the launch errors in ToasterExceptions |
| 121 | br.build.outcome = Build.FAILED |
| 122 | for brerror in br.brerror_set.all(): |
| 123 | logger.debug("Saving error %s" % brerror) |
Patrick Williams | d8c66bc | 2016-06-20 12:57:21 -0500 | [diff] [blame] | 124 | LogMessage.objects.create(build=br.build, |
| 125 | level=LogMessage.EXCEPTION, |
| 126 | message=brerror.errmsg) |
Patrick Williams | c124f4f | 2015-09-15 14:41:29 -0500 | [diff] [blame] | 127 | br.build.save() |
| 128 | |
Patrick Williams | d8c66bc | 2016-06-20 12:57:21 -0500 | [diff] [blame] | 129 | # we don't have a true build object here; hence, toasterui |
| 130 | # didn't have a change to release the BE lock |
Patrick Williams | c124f4f | 2015-09-15 14:41:29 -0500 | [diff] [blame] | 131 | br.environment.lock = BuildEnvironment.LOCK_FREE |
| 132 | br.environment.save() |
| 133 | |
Patrick Williams | c124f4f | 2015-09-15 14:41:29 -0500 | [diff] [blame] | 134 | # update all BuildRequests without a build created |
Patrick Williams | c0f7c04 | 2017-02-23 20:41:17 -0600 | [diff] [blame] | 135 | for br in BuildRequest.objects.filter(build=None): |
Patrick Williams | d8c66bc | 2016-06-20 12:57:21 -0500 | [diff] [blame] | 136 | br.build = Build.objects.create(project=br.project, |
| 137 | completed_on=br.updated, |
| 138 | started_on=br.created) |
Patrick Williams | c124f4f | 2015-09-15 14:41:29 -0500 | [diff] [blame] | 139 | br.build.outcome = Build.FAILED |
| 140 | try: |
| 141 | br.build.machine = br.brvariable_set.get(name='MACHINE').value |
| 142 | except BRVariable.DoesNotExist: |
| 143 | pass |
| 144 | br.save() |
| 145 | # transpose target information |
| 146 | for brtarget in br.brtarget_set.all(): |
Patrick Williams | d8c66bc | 2016-06-20 12:57:21 -0500 | [diff] [blame] | 147 | Target.objects.create(build=br.build, |
| 148 | target=brtarget.target, |
| 149 | task=brtarget.task) |
Patrick Williams | c124f4f | 2015-09-15 14:41:29 -0500 | [diff] [blame] | 150 | # transpose the launch errors in ToasterExceptions |
| 151 | for brerror in br.brerror_set.all(): |
Patrick Williams | d8c66bc | 2016-06-20 12:57:21 -0500 | [diff] [blame] | 152 | LogMessage.objects.create(build=br.build, |
| 153 | level=LogMessage.EXCEPTION, |
| 154 | message=brerror.errmsg) |
Patrick Williams | c124f4f | 2015-09-15 14:41:29 -0500 | [diff] [blame] | 155 | |
| 156 | br.build.save() |
Patrick Williams | d8c66bc | 2016-06-20 12:57:21 -0500 | [diff] [blame] | 157 | |
| 158 | # Make sure the LOCK is removed for builds which have been fully |
| 159 | # cancelled |
Brad Bishop | 6e60e8b | 2018-02-01 10:27:11 -0500 | [diff] [blame] | 160 | for br in BuildRequest.objects.filter( |
Patrick Williams | c0f7c04 | 2017-02-23 20:41:17 -0600 | [diff] [blame] | 161 | Q(build__outcome=Build.CANCELLED) & |
| 162 | Q(state=BuildRequest.REQ_CANCELLING) & |
| 163 | ~Q(environment=None)): |
Patrick Williams | d8c66bc | 2016-06-20 12:57:21 -0500 | [diff] [blame] | 164 | br.environment.lock = BuildEnvironment.LOCK_FREE |
| 165 | br.environment.save() |
Patrick Williams | c124f4f | 2015-09-15 14:41:29 -0500 | [diff] [blame] | 166 | |
Patrick Williams | c0f7c04 | 2017-02-23 20:41:17 -0600 | [diff] [blame] | 167 | def runbuild(self): |
| 168 | try: |
| 169 | self.cleanup() |
| 170 | except Exception as e: |
Andrew Geissler | 82c905d | 2020-04-13 13:39:40 -0500 | [diff] [blame] | 171 | logger.warning("runbuilds: cleanup exception %s" % str(e)) |
Patrick Williams | c0f7c04 | 2017-02-23 20:41:17 -0600 | [diff] [blame] | 172 | |
| 173 | try: |
| 174 | self.archive() |
| 175 | except Exception as e: |
Andrew Geissler | 82c905d | 2020-04-13 13:39:40 -0500 | [diff] [blame] | 176 | logger.warning("runbuilds: archive exception %s" % str(e)) |
Patrick Williams | c0f7c04 | 2017-02-23 20:41:17 -0600 | [diff] [blame] | 177 | |
| 178 | try: |
| 179 | self.schedule() |
| 180 | except Exception as e: |
Andrew Geissler | 82c905d | 2020-04-13 13:39:40 -0500 | [diff] [blame] | 181 | logger.warning("runbuilds: schedule exception %s" % str(e)) |
Patrick Williams | c124f4f | 2015-09-15 14:41:29 -0500 | [diff] [blame] | 182 | |
Brad Bishop | d7bf8c1 | 2018-02-25 22:55:05 -0500 | [diff] [blame] | 183 | def handle(self, **options): |
Brad Bishop | 6e60e8b | 2018-02-01 10:27:11 -0500 | [diff] [blame] | 184 | pidfile_path = os.path.join(os.environ.get("BUILDDIR", "."), |
| 185 | ".runbuilds.pid") |
| 186 | |
| 187 | with open(pidfile_path, 'w') as pidfile: |
| 188 | pidfile.write("%s" % os.getpid()) |
| 189 | |
Patrick Williams | c0f7c04 | 2017-02-23 20:41:17 -0600 | [diff] [blame] | 190 | self.runbuild() |
| 191 | |
| 192 | signal.signal(signal.SIGUSR1, lambda sig, frame: None) |
| 193 | |
Patrick Williams | f1e5d69 | 2016-03-30 15:21:19 -0500 | [diff] [blame] | 194 | while True: |
Patrick Williams | c0f7c04 | 2017-02-23 20:41:17 -0600 | [diff] [blame] | 195 | signal.pause() |
| 196 | self.runbuild() |