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 */
018
019package org.apache.oozie.command;
020
021import org.apache.oozie.client.Job;
022
023public abstract class SuspendTransitionXCommand extends TransitionXCommand<Void> {
024
025    /**
026     * Suspend all children of the job
027     *
028     * @throws CommandException
029     */
030    public abstract void suspendChildren() throws CommandException;
031
032    public SuspendTransitionXCommand(String name, String type, int priority) {
033        super(name, type, priority);
034    }
035
036    public SuspendTransitionXCommand(String name, String type, int priority, boolean dryrun) {
037        super(name, type, priority, dryrun);
038    }
039
040    /**
041     * Transit job to suspended from running or to prepsuspended from prep.
042     *
043     * @see org.apache.oozie.command.TransitionXCommand#transitToNext()
044     */
045    @Override
046    public void transitToNext() {
047        if (job == null) {
048            job = this.getJob();
049        }
050        if (job.getStatus() == Job.Status.PREP) {
051            job.setStatus(Job.Status.PREPSUSPENDED);
052        }
053        else if (job.getStatus() == Job.Status.RUNNING) {
054            job.setStatus(Job.Status.SUSPENDED);
055        }
056        else if (job.getStatus() == Job.Status.RUNNINGWITHERROR || job.getStatus() == Job.Status.PAUSEDWITHERROR){
057            job.setStatus(Job.Status.SUSPENDEDWITHERROR);
058        }
059        else if (job.getStatus() == Job.Status.PAUSED) {
060            job.setStatus(Job.Status.SUSPENDED);
061        }
062        else if (job.getStatus() == Job.Status.PREPPAUSED) {
063            job.setStatus(Job.Status.PREPSUSPENDED);
064        }
065        job.setPending();
066    }
067
068    /* (non-Javadoc)
069     * @see org.apache.oozie.command.XCommand#execute()
070     */
071    @Override
072    protected Void execute() throws CommandException {
073        transitToNext();
074        try {
075            suspendChildren();
076            updateJob();
077            performWrites();
078        } finally {
079            notifyParent();
080        }
081        return null;
082    }
083}