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 package org.apache.oozie.executor.jpa;
019
020 import javax.persistence.EntityManager;
021 import javax.persistence.Query;
022
023 import org.apache.oozie.CoordinatorActionBean;
024 import org.apache.oozie.ErrorCode;
025 import org.apache.oozie.client.CoordinatorAction;
026 import org.apache.oozie.util.ParamChecker;
027
028 /**
029 * JPAExecutor to get attributes of CoordinatorActionBean required by CoordActionStartXCommand
030 */
031 public class CoordActionGetForStartJPAExecutor implements JPAExecutor<CoordinatorActionBean> {
032
033 private String coordActionId = null;
034
035 public CoordActionGetForStartJPAExecutor(String coordActionId) {
036 ParamChecker.notNull(coordActionId, "coordActionId");
037 this.coordActionId = coordActionId;
038 }
039
040 /* (non-Javadoc)
041 * @see org.apache.oozie.executor.jpa.JPAExecutor#getName()
042 */
043 @Override
044 public String getName() {
045 return "CoordActionGetForStartExecutor";
046 }
047
048 /* (non-Javadoc)
049 * @see org.apache.oozie.executor.jpa.JPAExecutor#execute(javax.persistence.EntityManager)
050 */
051 @Override
052 public CoordinatorActionBean execute(EntityManager em) throws JPAExecutorException {
053 try {
054 Query q = em.createNamedQuery("GET_COORD_ACTION_FOR_START");
055 q.setParameter("id", coordActionId);
056 Object[] obj = (Object[]) q.getSingleResult();
057 CoordinatorActionBean caBean = getBeanForRunningCoordAction(obj);
058 return caBean;
059 }
060 catch (Exception e) {
061 throw new JPAExecutorException(ErrorCode.E0603, e);
062 }
063
064 }
065
066 private CoordinatorActionBean getBeanForRunningCoordAction(Object[] arr) {
067 CoordinatorActionBean bean = new CoordinatorActionBean();
068 if (arr[0] != null) {
069 bean.setId((String) arr[0]);
070 }
071 if (arr[1] != null) {
072 bean.setJobId((String) arr[1]);
073 }
074 if (arr[2] != null) {
075 bean.setStatus(CoordinatorAction.Status.valueOf((String) arr[2]));
076 }
077 if (arr[3] != null) {
078 bean.setPending((Integer) arr[3]);
079 }
080 if (arr[4] != null) {
081 bean.setCreatedConf((String) arr[4]);
082 }
083 if (arr[5] != null) {
084 bean.setSlaXml((String) arr[5]);
085 }
086 if (arr[6] != null) {
087 bean.setActionXml((String) arr[6]);
088 }
089 if (arr[7] != null) {
090 bean.setExternalId((String) arr[7]);
091 }
092 if (arr[8] != null) {
093 bean.setErrorMessage((String) arr[8]);
094 }
095 if (arr[9] != null) {
096 bean.setErrorCode((String) arr[9]);
097 }
098 return bean;
099 }
100 }