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 org.apache.oozie.client.WorkflowJob;
018    import org.apache.oozie.client.SLAEvent.SlaAppType;
019    import org.apache.oozie.client.SLAEvent.Status;
020    import org.apache.oozie.WorkflowActionBean;
021    import org.apache.oozie.WorkflowJobBean;
022    import org.apache.oozie.command.Command;
023    import org.apache.oozie.command.CommandException;
024    import org.apache.oozie.store.StoreException;
025    import org.apache.oozie.store.WorkflowStore;
026    import org.apache.oozie.store.Store;
027    import org.apache.oozie.workflow.WorkflowException;
028    import org.apache.oozie.workflow.WorkflowInstance;
029    import org.apache.oozie.workflow.lite.LiteWorkflowInstance;
030    import org.apache.oozie.util.ParamChecker;
031    import org.apache.oozie.util.XLog;
032    import org.apache.oozie.util.db.SLADbOperations;
033    
034    import java.util.Date;
035    
036    public class KillCommand extends WorkflowCommand<Void> {
037    
038        private String id;
039        private final XLog log = XLog.getLog(getClass());
040    
041        public KillCommand(String id) {
042            super("kill", "kill", 1, XLog.STD);
043            this.id = ParamChecker.notEmpty(id, "id");
044        }
045    
046        @Override
047        protected Void call(WorkflowStore store) throws StoreException, CommandException {
048            try {
049                log.info("In Workflow KillCommand.call() for jobId=" + id);
050                WorkflowJobBean workflow = store.getWorkflow(id, false);
051                setLogInfo(workflow);
052    
053                if (workflow.getStatus() == WorkflowJob.Status.PREP || workflow.getStatus() == WorkflowJob.Status.RUNNING
054                        || workflow.getStatus() == WorkflowJob.Status.SUSPENDED
055                        || workflow.getStatus() == WorkflowJob.Status.FAILED) {
056                    workflow.setEndTime(new Date());
057    
058                    if (workflow.getStatus() != WorkflowJob.Status.FAILED) {
059                        incrJobCounter(1);
060                        workflow.setStatus(WorkflowJob.Status.KILLED);
061                        SLADbOperations.writeStausEvent(workflow.getSlaXml(), workflow.getId(), store, Status.KILLED,
062                                                        SlaAppType.WORKFLOW_JOB);
063                        workflow.getWorkflowInstance().kill();
064                        WorkflowInstance wfInstance = workflow.getWorkflowInstance();
065                        ((LiteWorkflowInstance) wfInstance).setStatus(WorkflowInstance.Status.KILLED);
066                        workflow.setWorkflowInstance(wfInstance);
067                    }
068                    for (WorkflowActionBean action : store.getActionsForWorkflow(id, true)) {
069                        if (action.getStatus() == WorkflowActionBean.Status.RUNNING
070                                || action.getStatus() == WorkflowActionBean.Status.DONE) {
071                            action.setPending();
072                            action.setStatus(WorkflowActionBean.Status.KILLED);
073                            store.updateAction(action);
074                            queueCallable(new ActionKillCommand(action.getId(), action.getType()));
075                        }
076                        if (action.getStatus() == WorkflowActionBean.Status.PREP
077                                || action.getStatus() == WorkflowActionBean.Status.START_RETRY
078                                || action.getStatus() == WorkflowActionBean.Status.START_MANUAL
079                                || action.getStatus() == WorkflowActionBean.Status.END_RETRY
080                                || action.getStatus() == WorkflowActionBean.Status.END_MANUAL) {
081                            action.setStatus(WorkflowActionBean.Status.KILLED);
082                            action.resetPending();
083                            SLADbOperations.writeStausEvent(action.getSlaXml(), action.getId(), store, Status.KILLED,
084                                                            SlaAppType.WORKFLOW_ACTION);
085                            store.updateAction(action);
086                        }
087                    }
088                    store.updateWorkflow(workflow);
089                    queueCallable(new NotificationCommand(workflow));
090                }
091                return null;
092            }
093            catch (WorkflowException ex) {
094                throw new CommandException(ex);
095            }
096        }
097    
098        @Override
099        protected Void execute(WorkflowStore store) throws CommandException, StoreException {
100            try {
101                XLog.getLog(getClass()).debug("STARTED KillCommand for job " + id);
102                if (lock(id)) {
103                    call(store);
104                }
105                else {
106                    queueCallable(new KillCommand(id), LOCK_FAILURE_REQUEUE_INTERVAL);
107                    XLog.getLog(getClass()).warn("KillCommand lock was not acquired - failed {0}", id);
108                }
109            }
110            catch (InterruptedException e) {
111                queueCallable(new KillCommand(id), LOCK_FAILURE_REQUEUE_INTERVAL);
112                XLog.getLog(getClass()).warn("KillCommand lock was not acquired - interrupted exception failed {0}", id);
113            }
114            XLog.getLog(getClass()).debug("ENDED KillCommand for job " + id);
115            return null;
116        }
117    }