001 /** 002 * Copyright (c) 2010 Yahoo! Inc. All rights reserved. 003 * Licensed under the Apache License, Version 2.0 (the "License"); 004 * you may not use this file except in compliance with the License. 005 * You may obtain a copy of the License at 006 * 007 * http://www.apache.org/licenses/LICENSE-2.0 008 * 009 * Unless required by applicable law or agreed to in writing, software 010 * distributed under the License is distributed on an "AS IS" BASIS, 011 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 012 * See the License for the specific language governing permissions and 013 * limitations under the License. See accompanying LICENSE file. 014 */ 015 package org.apache.oozie.executor.jpa; 016 017 import java.util.List; 018 019 import javax.persistence.EntityManager; 020 import javax.persistence.Query; 021 022 import org.apache.oozie.ErrorCode; 023 import org.apache.oozie.WorkflowJobBean; 024 import org.apache.oozie.util.ParamChecker; 025 026 /** 027 * Load the WorkflowJob into a Bean and return it. 028 */ 029 public class WorkflowJobGetJPAExecutor implements JPAExecutor<WorkflowJobBean> { 030 031 private String wfJobId = null; 032 033 public WorkflowJobGetJPAExecutor(String wfJobId) { 034 ParamChecker.notNull(wfJobId, "wfJobId"); 035 this.wfJobId = wfJobId; 036 } 037 038 /* (non-Javadoc) 039 * @see org.apache.oozie.executor.jpa.JPAExecutor#getName() 040 */ 041 @Override 042 public String getName() { 043 return "WorkflowJobGetJPAExecutor"; 044 } 045 046 /* (non-Javadoc) 047 * @see org.apache.oozie.executor.jpa.JPAExecutor#execute(javax.persistence.EntityManager) 048 */ 049 @Override 050 @SuppressWarnings("unchecked") 051 public WorkflowJobBean execute(EntityManager em) throws JPAExecutorException { 052 List<WorkflowJobBean> wjBeans; 053 try { 054 Query q = em.createNamedQuery("GET_WORKFLOW"); 055 q.setParameter("id", wfJobId); 056 wjBeans = q.getResultList(); 057 } 058 catch (Exception e) { 059 throw new JPAExecutorException(ErrorCode.E0603, e); 060 } 061 WorkflowJobBean bean = null; 062 if (wjBeans != null && wjBeans.size() > 0) { 063 bean = wjBeans.get(0); 064 bean.setStatus(bean.getStatus()); 065 return bean; 066 } 067 else { 068 throw new JPAExecutorException(ErrorCode.E0604, wfJobId); 069 } 070 } 071 }