forked from M-Labs/artiq
artiq_run: implement mockups of new scheduler API
This commit is contained in:
parent
737f6d4485
commit
4a7c6954c3
|
@ -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)
|
||||||
|
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue