ExternalVendorCode/onefuzz
1
0
Fork 0
mirror of https://github.com/microsoft/onefuzz.git synced 2025-06-17 04:18:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

make user triggered reimaging happen immediately (#566)

Browse Source
This commit is contained in:
bmc-msft
2021-02-18 14:08:25 -05:00
committed by GitHub
parent 21f08f6a98
commit 929d9ce496
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/api-service/__app__/node/__init__.py
View File

@ -85,7 +85,7 @@ def delete(req: func.HttpRequest) -> func.HttpResponse:
def patch(req: func.HttpRequest) -> func.HttpResponse: def patch(req: func.HttpRequest) -> func.HttpResponse:
request = parse_request(NodeGet, req) request = parse_request(NodeGet, req)
if isinstance(request, Error): if isinstance(request, Error):
return not_ok(request, context="NodeRestart") return not_ok(request, context="NodeReimage")
node = Node.get_by_machine_id(request.machine_id) node = Node.get_by_machine_id(request.machine_id)
if not node: if not node:
@ -94,7 +94,7 @@ def patch(req: func.HttpRequest) -> func.HttpResponse:
context=request.machine_id, context=request.machine_id,
) )
node.stop() node.stop(done=True)
if node.debug_keep_node: if node.debug_keep_node:
node.debug_keep_node = False node.debug_keep_node = False
node.save() node.save()

4
src/api-service/__app__/onefuzzlib/workers/nodes.py
View File

@ -281,8 +281,8 @@ class Node(BASE_NODE, ORMMixin):
) )
return None return None
def stop(self) -> None: def stop(self, done: bool = False) -> None:
self.to_reimage() self.to_reimage(done=done)
self.send_message(NodeCommand(stop=StopNodeCommand())) self.send_message(NodeCommand(stop=StopNodeCommand()))
def set_shutdown(self) -> None: def set_shutdown(self) -> None: