001/**
002 * Licensed to the Apache Software Foundation (ASF) under one
003 * or more contributor license agreements.  See the NOTICE file
004 * distributed with this work for additional information
005 * regarding copyright ownership.  The ASF licenses this file
006 * to you under the Apache License, Version 2.0 (the
007 * "License"); you may not use this file except in compliance
008 * with the License.  You may obtain a copy of the License at
009 *
010 *      http://www.apache.org/licenses/LICENSE-2.0
011 *
012 * Unless required by applicable law or agreed to in writing, software
013 * distributed under the License is distributed on an "AS IS" BASIS,
014 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
015 * See the License for the specific language governing permissions and
016 * limitations under the License.
017 */
018package org.apache.oozie.command;
019
020import org.apache.oozie.client.Job;
021
022public abstract class SuspendTransitionXCommand extends TransitionXCommand<Void> {
023
024    /**
025     * Suspend all children of the job
026     *
027     * @throws CommandException
028     */
029    public abstract void suspendChildren() throws CommandException;
030
031    public SuspendTransitionXCommand(String name, String type, int priority) {
032        super(name, type, priority);
033    }
034
035    public SuspendTransitionXCommand(String name, String type, int priority, boolean dryrun) {
036        super(name, type, priority, dryrun);
037    }
038
039    /**
040     * Transit job to suspended from running or to prepsuspended from prep.
041     *
042     * @see org.apache.oozie.command.TransitionXCommand#transitToNext()
043     */
044    @Override
045    public void transitToNext() {
046        if (job == null) {
047            job = this.getJob();
048        }
049        if (job.getStatus() == Job.Status.PREP) {
050            job.setStatus(Job.Status.PREPSUSPENDED);
051        }
052        else if (job.getStatus() == Job.Status.RUNNING) {
053            job.setStatus(Job.Status.SUSPENDED);
054        }
055        else if (job.getStatus() == Job.Status.RUNNINGWITHERROR || job.getStatus() == Job.Status.PAUSEDWITHERROR){
056            job.setStatus(Job.Status.SUSPENDEDWITHERROR);
057        }
058        else if (job.getStatus() == Job.Status.PAUSED) {
059            job.setStatus(Job.Status.SUSPENDED);
060        }
061        else if (job.getStatus() == Job.Status.PREPPAUSED) {
062            job.setStatus(Job.Status.PREPSUSPENDED);
063        }
064        job.setPending();
065    }
066
067    /* (non-Javadoc)
068     * @see org.apache.oozie.command.XCommand#execute()
069     */
070    @Override
071    protected Void execute() throws CommandException {
072        transitToNext();
073        try {
074            suspendChildren();
075            updateJob();
076            performWrites();
077        } finally {
078            notifyParent();
079        }
080        return null;
081    }
082}