def _get_process_schedule(**kwargs):
queueing_mode = kwargs.get('queueing_mode')
restart_mode = kwargs.get('restart_mode')
execution_engine_id = kwargs.get('execution_engine_id')
node_exclusive = kwargs.get('node_exclusive')
constraints = kwargs.get('constraints')
process_schedule = ProcessSchedule()
if queueing_mode is not None:
try:
process_schedule.queueing_mode = ProcessQueueingMode._value_map[queueing_mode]
except KeyError:
msg = "%s is not a known ProcessQueueingMode" % (queueing_mode)
raise BadRequest(msg)
if restart_mode is not None:
try:
process_schedule.restart_mode = ProcessRestartMode._value_map[restart_mode]
except KeyError:
msg = "%s is not a known ProcessRestartMode" % (restart_mode)
raise BadRequest(msg)
else:
# if restart mode isn't specified, use NEVER. HA Agent itself will reschedule failures.
process_schedule.restart_mode = ProcessRestartMode.NEVER
target = ProcessTarget()
if execution_engine_id is not None:
target.execution_engine_id = execution_engine_id
if node_exclusive is not None:
target.node_exclusive = node_exclusive
if constraints is not None:
target.constraints = constraints
process_schedule.target = target
return process_schedule
def _get_process_schedule(**kwargs):
queueing_mode = kwargs.get('queueing_mode')
restart_mode = kwargs.get('restart_mode')
execution_engine_id = kwargs.get('execution_engine_id')
node_exclusive = kwargs.get('node_exclusive')
constraints = kwargs.get('constraints')
process_schedule = ProcessSchedule()
if queueing_mode is not None:
try:
process_schedule.queueing_mode = ProcessQueueingMode._value_map[queueing_mode]
except KeyError:
msg = "%s is not a known ProcessQueueingMode" % (queueing_mode)
raise BadRequest(msg)
if restart_mode is not None:
try:
process_schedule.restart_mode = ProcessRestartMode._value_map[restart_mode]
except KeyError:
msg = "%s is not a known ProcessRestartMode" % (restart_mode)
raise BadRequest(msg)
else:
# if restart mode isn't specified, use NEVER. HA Agent itself will reschedule failures.
process_schedule.restart_mode = ProcessRestartMode.NEVER
target = ProcessTarget()
if execution_engine_id is not None:
target.execution_engine_id = execution_engine_id
if node_exclusive is not None:
target.node_exclusive = node_exclusive
if constraints is not None:
target.constraints = constraints
process_schedule.target = target
return process_schedule
def schedule_process(self, upid, definition_id, configuration=None,
subscribers=None, constraints=None, queueing_mode=None,
restart_mode=None, execution_engine_id=None, node_exclusive=None):
definition = self.real_client.read_process_definition(definition_id)
self.event_pub.publish_event(event_type="ProcessLifecycleEvent",
origin=definition.name, origin_type="DispatchedHAProcess",
state=ProcessStateEnum.RUNNING)
create_upid = self.real_client.create_process(definition_id)
process_schedule = ProcessSchedule()
if queueing_mode is not None:
try:
process_schedule.queueing_mode = ProcessQueueingMode._value_map[queueing_mode]
except KeyError:
msg = "%s is not a known ProcessQueueingMode" % (queueing_mode)
raise BadRequest(msg)
if restart_mode is not None:
try:
process_schedule.restart_mode = ProcessRestartMode._value_map[restart_mode]
except KeyError:
msg = "%s is not a known ProcessRestartMode" % (restart_mode)
raise BadRequest(msg)
target = ProcessTarget()
if execution_engine_id is not None:
target.execution_engine_id = execution_engine_id
if node_exclusive is not None:
target.node_exclusive = node_exclusive
if constraints is not None:
target.constraints = constraints
process_schedule.target = target
sched_pid = self.real_client.schedule_process(definition_id,
process_schedule, configuration=configuration, process_id=create_upid)
proc = self.real_client.read_process(sched_pid)
self._associate_process(proc)
dict_proc = {'upid': proc.process_id,
'state': self.state_map.get(proc.process_state, self.unknown_state),
}
return dict_proc
