X-Git-Url: https://git.arvados.org/arvados-workbench2.git/blobdiff_plain/1826bc21a34b983778f6e221b6c4004fdea1ad1d..a6c092fe45fe0ff310800e82e224bc822890491a:/src/store/run-process-panel/run-process-panel-actions.ts diff --git a/src/store/run-process-panel/run-process-panel-actions.ts b/src/store/run-process-panel/run-process-panel-actions.ts index c39517fe..d9686feb 100644 --- a/src/store/run-process-panel/run-process-panel-actions.ts +++ b/src/store/run-process-panel/run-process-panel-actions.ts @@ -6,18 +6,21 @@ import { Dispatch } from 'redux'; import { unionize, ofType, UnionOf } from "~/common/unionize"; import { ServiceRepository } from "~/services/services"; import { RootState } from '~/store/store'; -import { WorkflowResource, getWorkflowInputs, parseWorkflowDefinition } from '~/models/workflow'; +import { getUserUuid } from "~/common/getuser"; +import { WorkflowResource, WorkflowRunnerResources, getWorkflow, getWorkflowInputs, parseWorkflowDefinition } from '~/models/workflow'; import { getFormValues, initialize } from 'redux-form'; import { RUN_PROCESS_BASIC_FORM, RunProcessBasicFormData } from '~/views/run-process-panel/run-process-basic-form'; import { RUN_PROCESS_INPUTS_FORM } from '~/views/run-process-panel/run-process-inputs-form'; import { WorkflowInputsData } from '~/models/workflow'; import { createWorkflowMounts } from '~/models/process'; import { ContainerRequestState } from '~/models/container-request'; -import { navigateToProcess } from '../navigation/navigation-action'; -import { RunProcessAdvancedFormData, RUN_PROCESS_ADVANCED_FORM, VCPUS_FIELD, RAM_FIELD, RUNTIME_FIELD, OUTPUT_FIELD, API_FIELD } from '~/views/run-process-panel/run-process-advanced-form'; +import { navigateTo } from '../navigation/navigation-action'; +import { + RunProcessAdvancedFormData, RUN_PROCESS_ADVANCED_FORM, VCPUS_FIELD, + KEEP_CACHE_RAM_FIELD, RAM_FIELD, RUNTIME_FIELD, OUTPUT_FIELD, RUNNER_IMAGE_FIELD +} from '~/views/run-process-panel/run-process-advanced-form'; import { dialogActions } from '~/store/dialog/dialog-actions'; import { setBreadcrumbs } from '~/store/breadcrumbs/breadcrumbs-actions'; -import { matchProjectRoute } from '~/routes/routes'; export const runProcessPanelActions = unionize({ SET_PROCESS_PATHNAME: ofType(), @@ -72,19 +75,40 @@ export const openSetWorkflowDialog = (workflow: WorkflowResource) => } }; +export const getWorkflowRunnerSettings = (workflow: WorkflowResource) => { + const advancedFormValues = {}; + Object.assign(advancedFormValues, DEFAULT_ADVANCED_FORM_VALUES); + + const wf = getWorkflow(parseWorkflowDefinition(workflow)); + const hints = wf ? wf.hints : undefined; + if (hints) { + const resc = hints.find(item => item.class === 'http://arvados.org/cwl#WorkflowRunnerResources') as WorkflowRunnerResources | undefined; + if (resc) { + if (resc.ramMin) { advancedFormValues[RAM_FIELD] = resc.ramMin * (1024 * 1024); } + if (resc.coresMin) { advancedFormValues[VCPUS_FIELD] = resc.coresMin; } + if (resc.keep_cache) { advancedFormValues[KEEP_CACHE_RAM_FIELD] = resc.keep_cache * (1024 * 1024); } + if (resc.acrContainerImage) { advancedFormValues[RUNNER_IMAGE_FIELD] = resc.acrContainerImage; } + } + } + return advancedFormValues; +}; + export const setWorkflow = (workflow: WorkflowResource, isWorkflowChanged = true) => (dispatch: Dispatch, getState: () => RootState) => { const isStepChanged = getState().runProcessPanel.isStepChanged; + + const advancedFormValues = getWorkflowRunnerSettings(workflow); + if (isStepChanged && isWorkflowChanged) { dispatch(runProcessPanelActions.SET_STEP_CHANGED(false)); dispatch(runProcessPanelActions.SET_SELECTED_WORKFLOW(workflow)); dispatch(loadPresets(workflow.uuid)); - dispatch(initialize(RUN_PROCESS_ADVANCED_FORM, DEFAULT_ADVANCED_FORM_VALUES)); + dispatch(initialize(RUN_PROCESS_ADVANCED_FORM, advancedFormValues)); } if (!isWorkflowChanged) { dispatch(runProcessPanelActions.SET_SELECTED_WORKFLOW(workflow)); dispatch(loadPresets(workflow.uuid)); - dispatch(initialize(RUN_PROCESS_ADVANCED_FORM, DEFAULT_ADVANCED_FORM_VALUES)); + dispatch(initialize(RUN_PROCESS_ADVANCED_FORM, advancedFormValues)); } }; @@ -117,13 +141,14 @@ export const runProcess = async (dispatch: Dispatch, getState: () => RootSt const state = getState(); const basicForm = getFormValues(RUN_PROCESS_BASIC_FORM)(state) as RunProcessBasicFormData; const inputsForm = getFormValues(RUN_PROCESS_INPUTS_FORM)(state) as WorkflowInputsData; - const advancedForm = getFormValues(RUN_PROCESS_ADVANCED_FORM)(state) as RunProcessAdvancedFormData || DEFAULT_ADVANCED_FORM_VALUES; - const userUuid = getState().auth.user!.uuid; - const pathname = getState().runProcessPanel.processPathname; + const userUuid = getUserUuid(getState()); + if (!userUuid) { return; } const { processOwnerUuid, selectedWorkflow } = state.runProcessPanel; + const ownerUUid = processOwnerUuid ? processOwnerUuid : userUuid; if (selectedWorkflow) { + const advancedForm = getFormValues(RUN_PROCESS_ADVANCED_FORM)(state) as RunProcessAdvancedFormData || getWorkflowRunnerSettings(selectedWorkflow); const newProcessData = { - ownerUuid: !matchProjectRoute(pathname) ? userUuid : processOwnerUuid, + ownerUuid: ownerUUid, name: basicForm.name, description: basicForm.description, state: ContainerRequestState.COMMITTED, @@ -131,18 +156,18 @@ export const runProcess = async (dispatch: Dispatch, getState: () => RootSt runtimeConstraints: { API: true, vcpus: advancedForm[VCPUS_FIELD], - ram: advancedForm[RAM_FIELD], - api: advancedForm[API_FIELD], + ram: (advancedForm[KEEP_CACHE_RAM_FIELD] + advancedForm[RAM_FIELD]), }, schedulingParameters: { - maxRunTime: advancedForm[RUNTIME_FIELD] + max_run_time: advancedForm[RUNTIME_FIELD] }, - containerImage: 'arvados/jobs', + containerImage: advancedForm[RUNNER_IMAGE_FIELD], cwd: '/var/spool/cwl', command: [ 'arvados-cwl-runner', '--api=containers', '--local', + `--project-uuid=${ownerUUid}`, '/var/lib/cwl/workflow.json#main', '/var/lib/cwl/cwl.input.json' ], @@ -152,17 +177,18 @@ export const runProcess = async (dispatch: Dispatch, getState: () => RootSt properties: { workflowUuid: selectedWorkflow.uuid, workflowName: selectedWorkflow.name - } + } }; const newProcess = await services.containerRequestService.create(newProcessData); - dispatch(navigateToProcess(newProcess.uuid)); + dispatch(navigateTo(newProcess.uuid)); } }; -export const DEFAULT_ADVANCED_FORM_VALUES: Partial = { +const DEFAULT_ADVANCED_FORM_VALUES: Partial = { [VCPUS_FIELD]: 1, [RAM_FIELD]: 1073741824, - [API_FIELD]: true, + [KEEP_CACHE_RAM_FIELD]: 268435456, + [RUNNER_IMAGE_FIELD]: "arvados/jobs" }; const normalizeInputKeys = (inputs: WorkflowInputsData): WorkflowInputsData =>