forked from M-Labs/artiq
1
0
Fork 0

artiq_run: implement mockups of new scheduler API

This commit is contained in:
Sebastien Bourdeauducq 2015-05-28 17:21:20 +08:00
parent 737f6d4485
commit 4a7c6954c3
1 changed files with 14 additions and 19 deletions

View File

@ -48,28 +48,21 @@ class DummyWatchdog:
class DummyScheduler: class DummyScheduler:
def __init__(self): def __init__(self, expid):
self.next_rid = 0 self.next_rid = 0
self.next_trid = 0 self.next_trid = 0
self.pipeline_name = "main"
self.priority = 0
self.expid = expid
def run_queued(self, run_params): def submit(self, pipeline_name, expid, priority, due_date, flush):
rid = self.next_rid rid = self.next_rid
self.next_rid += 1 self.next_rid += 1
logger.info("Queuing: %s, RID=%s", run_params, rid) logger.info("Submitting: %s, RID=%s", expid, rid)
return rid return rid
def cancel_queued(self, rid): def delete(self, rid):
logger.info("Cancelling RID %s", rid) logger.info("Deleting RID %s", rid)
def run_timed(self, run_params, next_run):
trid = self.next_trid
self.next_trid += 1
next_run_s = time.strftime("%m/%d %H:%M:%S", time.localtime(next_run))
logger.info("Timing: %s at %s, TRID=%s", run_params, next_run_s, trid)
return trid
def cancel_timed(self, trid):
logger.info("Cancelling TRID %s", trid)
watchdog = DummyWatchdog watchdog = DummyWatchdog
@ -115,11 +108,13 @@ def _build_experiment(dbh, args):
file = getattr(module, "__file__") file = getattr(module, "__file__")
exp = get_experiment(module, args.experiment) exp = get_experiment(module, args.experiment)
arguments = parse_arguments(args.arguments) arguments = parse_arguments(args.arguments)
expid = {
"file": file,
"experiment": args.experiment,
"arguments": arguments
}
return exp(dbh, return exp(dbh,
scheduler=DummyScheduler(), scheduler=DummyScheduler(expid),
run_params=dict(file=file,
experiment=args.experiment,
arguments=arguments),
**arguments) **arguments)