/**
* Copyright (c) 2010 Yahoo! Inc. All rights reserved.
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License. See accompanying LICENSE file.
*/
package org.apache.oozie.command.bundle;
import java.util.Date;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;
import org.apache.oozie.BundleActionBean;
import org.apache.oozie.BundleJobBean;
import org.apache.oozie.ErrorCode;
import org.apache.oozie.XException;
import org.apache.oozie.client.Job;
import org.apache.oozie.client.OozieClient;
import org.apache.oozie.command.CommandException;
import org.apache.oozie.command.PreconditionException;
import org.apache.oozie.command.XCommand;
import org.apache.oozie.command.coord.CoordChangeXCommand;
import org.apache.oozie.executor.jpa.BundleActionUpdateJPAExecutor;
import org.apache.oozie.executor.jpa.BundleActionsGetJPAExecutor;
import org.apache.oozie.executor.jpa.BundleJobGetJPAExecutor;
import org.apache.oozie.executor.jpa.BundleJobUpdateJPAExecutor;
import org.apache.oozie.service.JPAService;
import org.apache.oozie.service.Services;
import org.apache.oozie.util.DateUtils;
import org.apache.oozie.util.JobUtils;
import org.apache.oozie.util.LogUtils;
import org.apache.oozie.util.ParamChecker;
public class BundleJobChangeXCommand extends XCommand<Void> {
private String jobId;
private String changeValue;
private JPAService jpaService;
private List<BundleActionBean> bundleActions;
private BundleJobBean bundleJob;
private Date newPauseTime = null;
private Date newEndTime = null;
boolean isChangePauseTime = false;
boolean isChangeEndTime = false;
private static final Set<String> ALLOWED_CHANGE_OPTIONS = new HashSet<String>();
static {
ALLOWED_CHANGE_OPTIONS.add("pausetime");
ALLOWED_CHANGE_OPTIONS.add("endtime");
}
/**
* @param id bundle job id
* @param changeValue change value
*
* @throws CommandException thrown if failed to change bundle
*/
public BundleJobChangeXCommand(String id, String changeValue) throws CommandException {
super("bundle_change", "bundle_change", 1);
this.jobId = ParamChecker.notEmpty(id, "id");
this.changeValue = ParamChecker.notEmpty(changeValue, "changeValue");
}
/**
* Check if new pause time is future time.
*
* @param newPauseTime new pause time.
* @throws CommandException thrown if new pause time is not valid.
*/
private void checkPauseTime(Date newPauseTime) throws CommandException {
// New pauseTime has to be a non-past time.
Date d = new Date();
if (newPauseTime.before(d)) {
throw new CommandException(ErrorCode.E1317, newPauseTime, "must be a non-past time");
}
}
/**
* Check if new pause time is future time.
*
* @param newEndTime new end time, can be null meaning no change on end time.
* @throws CommandException thrown if new end time is not valid.
*/
private void checkEndTime(Date newEndTime) throws CommandException {
// New endTime has to be a non-past start time.
Date startTime = bundleJob.getKickoffTime();
if (startTime != null && newEndTime.before(startTime)) {
throw new CommandException(ErrorCode.E1317, newEndTime, "must be greater then kickoff time");
}
}
/**
* validate if change value is valid.
*
* @param changeValue change value.
* @throws CommandException thrown if changeValue cannot be parsed properly.
*/
private void validateChangeValue(String changeValue) throws CommandException {
Map<String, String> map = JobUtils.parseChangeValue(changeValue);
if (map.size() > ALLOWED_CHANGE_OPTIONS.size() || !(map.containsKey(OozieClient.CHANGE_VALUE_PAUSETIME) || map.containsKey(OozieClient.CHANGE_VALUE_ENDTIME))) {
throw new CommandException(ErrorCode.E1317, changeValue, "can only change pausetime or end time");
}
if (map.containsKey(OozieClient.CHANGE_VALUE_PAUSETIME)) {
isChangePauseTime = true;
}
else if(map.containsKey(OozieClient.CHANGE_VALUE_ENDTIME)){
isChangeEndTime = true;
}
else {
throw new CommandException(ErrorCode.E1317, changeValue, "should change pausetime or endtime");
}
if(isChangePauseTime){
String value = map.get(OozieClient.CHANGE_VALUE_PAUSETIME);
if (!value.equals("")) {
try {
newPauseTime = DateUtils.parseDateUTC(value);
}
catch (Exception ex) {
throw new CommandException(ErrorCode.E1317, value, "is not a valid date");
}
checkPauseTime(newPauseTime);
}
}
else if (isChangeEndTime){
String value = map.get(OozieClient.CHANGE_VALUE_ENDTIME);
if (!value.equals("")) {
try {
newEndTime = DateUtils.parseDateUTC(value);
}
catch (Exception ex) {
throw new CommandException(ErrorCode.E1317, value, "is not a valid date");
}
checkEndTime(newEndTime);
}
}
}
/* (non-Javadoc)
* @see org.apache.oozie.command.XCommand#execute()
*/
@Override
protected Void execute() throws CommandException {
try {
if (isChangePauseTime || isChangeEndTime) {
if (isChangePauseTime) {
bundleJob.setPauseTime(newPauseTime);
}
else if (isChangeEndTime) {
bundleJob.setEndTime(newEndTime);
bundleJob.setStatus(Job.Status.RUNNING);
}
for (BundleActionBean action : this.bundleActions) {
// queue coord change commands;
if (action.getStatus() != Job.Status.KILLED && action.getCoordId() != null) {
queue(new CoordChangeXCommand(action.getCoordId(), changeValue));
LOG.info("Queuing CoordChangeXCommand coord job = " + action.getCoordId() + " to change "
+ changeValue);
action.setPending(action.getPending() + 1);
jpaService.execute(new BundleActionUpdateJPAExecutor(action));
}
}
jpaService.execute(new BundleJobUpdateJPAExecutor(bundleJob));
}
return null;
}
catch (XException ex) {
throw new CommandException(ex);
}
}
/* (non-Javadoc)
* @see org.apache.oozie.command.XCommand#getEntityKey()
*/
@Override
protected String getEntityKey() {
return this.jobId;
}
/* (non-Javadoc)
* @see org.apache.oozie.command.XCommand#isLockRequired()
*/
@Override
protected boolean isLockRequired() {
return true;
}
/* (non-Javadoc)
* @see org.apache.oozie.command.XCommand#loadState()
*/
@Override
protected void loadState() throws CommandException {
try{
eagerLoadState();
this.bundleActions = jpaService.execute(new BundleActionsGetJPAExecutor(jobId));
}
catch(Exception Ex){
throw new CommandException(ErrorCode.E1311,this.jobId);
}
}
/* (non-Javadoc)
* @see org.apache.oozie.command.XCommand#verifyPrecondition()
*/
@Override
protected void verifyPrecondition() throws CommandException, PreconditionException {
}
/* (non-Javadoc)
* @see org.apache.oozie.command.XCommand#eagerLoadState()
*/
@Override
protected void eagerLoadState() throws CommandException {
try {
jpaService = Services.get().get(JPAService.class);
if (jpaService != null) {
this.bundleJob = jpaService.execute(new BundleJobGetJPAExecutor(jobId));
LogUtils.setLogInfo(bundleJob, logInfo);
}
else {
throw new CommandException(ErrorCode.E0610);
}
}
catch (XException ex) {
throw new CommandException(ex);
}
}
/* (non-Javadoc)
* @see org.apache.oozie.command.XCommand#eagerVerifyPrecondition()
*/
@Override
protected void eagerVerifyPrecondition() throws CommandException, PreconditionException {
validateChangeValue(changeValue);
if (bundleJob == null) {
LOG.info("BundleChangeCommand not succeeded - " + "job " + jobId + " does not exist");
throw new PreconditionException(ErrorCode.E1314, jobId);
}
if (isChangePauseTime) {
if (bundleJob.getStatus() == Job.Status.SUCCEEDED || bundleJob.getStatus() == Job.Status.FAILED
|| bundleJob.getStatus() == Job.Status.KILLED || bundleJob.getStatus() == Job.Status.DONEWITHERROR
|| bundleJob == null) {
LOG.info("BundleChangeCommand not succeeded for changing pausetime- " + "job " + jobId + " finished, status is "
+ bundleJob.getStatusStr());
throw new PreconditionException(ErrorCode.E1312, jobId, bundleJob.getStatus().toString());
}
}
else if(isChangeEndTime){
if (bundleJob.getStatus() == Job.Status.KILLED || bundleJob == null) {
LOG.info("BundleChangeCommand not succeeded for changing endtime- " + "job " + jobId + " finished, status is "
+ bundleJob.getStatusStr());
throw new PreconditionException(ErrorCode.E1312, jobId, bundleJob.getStatus().toString());
}
}
}
}