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