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.ArrayList;
018    import java.util.List;
019    
020    import javax.persistence.EntityManager;
021    import javax.persistence.Query;
022    
023    import org.apache.oozie.BundleJobBean;
024    import org.apache.oozie.ErrorCode;
025    
026    /**
027     * Get a list of Bundle Jobs that are are RUNNING or RUNNINGWITHERROR status. The result is ordered by lastModifiedTimestamp.
028     */
029    public class BundleJobsGetRunningJPAExecutor implements JPAExecutor<List<BundleJobBean>> {
030        private int limit;
031    
032        public BundleJobsGetRunningJPAExecutor(int limit) {
033            this.limit = limit;
034        }
035    
036        /* (non-Javadoc)
037         * @see org.apache.oozie.executor.jpa.JPAExecutor#getName()
038         */
039        @Override
040        public String getName() {
041            return "BundleJobsGetRunningJPAExecutor";
042        }
043    
044        /* (non-Javadoc)
045         * @see org.apache.oozie.executor.jpa.JPAExecutor#execute(javax.persistence.EntityManager)
046         */
047        @Override
048        @SuppressWarnings("unchecked")
049        public List<BundleJobBean> execute(EntityManager em) throws JPAExecutorException {
050            List<BundleJobBean> bjBeans;
051            List<BundleJobBean> jobList = new ArrayList<BundleJobBean>();
052            try {
053                Query q = em.createNamedQuery("GET_BUNDLE_JOBS_RUNNING");
054                if (limit > 0) {
055                    q.setMaxResults(limit);
056                }
057                bjBeans = q.getResultList();
058                for (BundleJobBean j : bjBeans) {
059                    jobList.add(j);
060                }
061            }
062            catch (Exception e) {
063                throw new JPAExecutorException(ErrorCode.E0603, e);
064            }
065            return jobList;
066        }
067    }