/***************************************************************************
* Copyright (c) 2014 VMware, 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.
***************************************************************************/
package com.vmware.bdd.service.job;
import org.apache.log4j.Logger;
import org.springframework.batch.core.ExitStatus;
import org.springframework.batch.core.JobExecution;
import org.springframework.transaction.annotation.Transactional;
import com.vmware.bdd.entity.ClusterEntity;
import com.vmware.bdd.manager.intf.IClusterEntityManager;
public class UpgradeClusterJobExecutionListener extends ClusterJobExecutionListener {
private static final Logger logger = Logger.getLogger(UpgradeClusterJobExecutionListener.class);
@Transactional
public void afterJob(JobExecution je) {
super.afterJob(je);
Boolean success =
TrackableTasklet.getFromJobExecutionContext(
je.getExecutionContext(),
JobConstants.CLUSTER_OPERATION_SUCCESS, Boolean.class);
final String clusterName = getJobParameters(je).getString(JobConstants.CLUSTER_NAME_JOB_PARAM);
if (success == null || success) {
success = (je.getExitStatus().equals(ExitStatus.COMPLETED));
}
if (success) {
IClusterEntityManager clusterEntityMgr = getClusterEntityMgr();
ClusterEntity cluster = clusterEntityMgr.findByName(clusterName);
cluster.setVersion(clusterEntityMgr.getServerVersion());
cluster.setStatus(cluster.getLastStatus());
cluster.setLastStatus(null);
logger.info("cluster " + clusterName + "upgrade job succeeds");
}
}
}