001 /**
002 * Copyright (c) 2010 Yahoo! Inc. All rights reserved.
003 * Licensed under the Apache License, Version 2.0 (the "License");
004 * you may not use this file except in compliance with the License.
005 * You may obtain a copy of the License at
006 *
007 * http://www.apache.org/licenses/LICENSE-2.0
008 *
009 * Unless required by applicable law or agreed to in writing, software
010 * distributed under the License is distributed on an "AS IS" BASIS,
011 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
012 * See the License for the specific language governing permissions and
013 * limitations under the License. See accompanying LICENSE file.
014 */
015 package org.apache.oozie.command.wf;
016
017 import java.util.Date;
018
019 import org.apache.oozie.WorkflowActionBean;
020 import org.apache.oozie.WorkflowJobBean;
021 import org.apache.oozie.client.WorkflowJob;
022 import org.apache.oozie.command.CommandException;
023 import org.apache.oozie.store.StoreException;
024 import org.apache.oozie.store.WorkflowStore;
025 import org.apache.oozie.util.ParamChecker;
026 import org.apache.oozie.util.XLog;
027 import org.apache.oozie.workflow.WorkflowException;
028 import org.apache.oozie.workflow.WorkflowInstance;
029 import org.apache.oozie.workflow.lite.LiteWorkflowInstance;
030
031 public class ResumeCommand extends WorkflowCommand<Void> {
032
033 private String id;
034
035 public ResumeCommand(String id) {
036 super("resume", "resume", 1, XLog.STD);
037 this.id = ParamChecker.notEmpty(id, "id");
038 }
039
040 @Override
041 protected Void call(WorkflowStore store) throws StoreException, CommandException {
042 try {
043 WorkflowJobBean workflow = store.getWorkflow(id, false);
044 setLogInfo(workflow);
045 if (workflow.getStatus() == WorkflowJob.Status.SUSPENDED) {
046 incrJobCounter(1);
047 workflow.getWorkflowInstance().resume();
048 WorkflowInstance wfInstance = workflow.getWorkflowInstance();
049 ((LiteWorkflowInstance) wfInstance).setStatus(WorkflowInstance.Status.RUNNING);
050 workflow.setWorkflowInstance(wfInstance);
051 workflow.setStatus(WorkflowJob.Status.RUNNING);
052
053 for (WorkflowActionBean action : store.getActionsForWorkflow(id, false)) {
054
055 // Set pending flag to true for the actions that are START_RETRY or
056 // START_MANUAL or END_RETRY or END_MANUAL
057 if (action.isRetryOrManual()) {
058 action.setPendingOnly();
059 store.updateAction(action);
060 }
061
062 if (action.isPending()) {
063 if (action.getStatus() == WorkflowActionBean.Status.PREP
064 || action.getStatus() == WorkflowActionBean.Status.START_MANUAL) {
065 queueCallable(new ActionStartCommand(action.getId(), action.getType()));
066 }
067 else {
068 if (action.getStatus() == WorkflowActionBean.Status.START_RETRY) {
069 Date nextRunTime = action.getPendingAge();
070 queueCallable(new ActionStartCommand(action.getId(), action.getType()),
071 nextRunTime.getTime() - System.currentTimeMillis());
072 }
073 else {
074 if (action.getStatus() == WorkflowActionBean.Status.DONE
075 || action.getStatus() == WorkflowActionBean.Status.END_MANUAL) {
076 queueCallable(new ActionEndCommand(action.getId(), action.getType()));
077 }
078 else {
079 if (action.getStatus() == WorkflowActionBean.Status.END_RETRY) {
080 Date nextRunTime = action.getPendingAge();
081 queueCallable(new ActionEndCommand(action.getId(), action.getType()),
082 nextRunTime.getTime() - System.currentTimeMillis());
083 }
084 }
085 }
086 }
087
088 }
089 }
090
091 store.updateWorkflow(workflow);
092 queueCallable(new NotificationCommand(workflow));
093 }
094 return null;
095 }
096 catch (WorkflowException ex) {
097 throw new CommandException(ex);
098 }
099 }
100
101 @Override
102 protected Void execute(WorkflowStore store) throws CommandException, StoreException {
103 XLog.getLog(getClass()).debug("STARTED ResumeCommand for action " + id);
104 try {
105 if (lock(id)) {
106 call(store);
107 }
108 else {
109 queueCallable(new KillCommand(id), LOCK_FAILURE_REQUEUE_INTERVAL);
110 XLog.getLog(getClass()).warn("Resume lock was not acquired - failed {0}", id);
111 }
112 }
113 catch (InterruptedException e) {
114 queueCallable(new KillCommand(id), LOCK_FAILURE_REQUEUE_INTERVAL);
115 XLog.getLog(getClass()).warn("ResumeCommand lock was not acquired - interrupted exception failed {0}", id);
116 }
117 finally {
118 XLog.getLog(getClass()).debug("ENDED ResumeCommand for action " + id);
119 }
120 return null;
121 }
122 }