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.CoordinatorJobBean;
023    import org.apache.oozie.ErrorCode;
024    import org.apache.oozie.util.ParamChecker;
025    
026    /**
027     * Load the coordinators for specified bundle in the Coordinator job bean
028     */
029    public class BundleJobGetCoordinatorsJPAExecutor implements JPAExecutor<List<CoordinatorJobBean>> {
030        private String bundleId = null;
031    
032        public BundleJobGetCoordinatorsJPAExecutor(String bundleId) {
033            ParamChecker.notNull(bundleId, "bundleId");
034            this.bundleId = bundleId;
035        }
036    
037        /* (non-Javadoc)
038         * @see org.apache.oozie.executor.jpa.JPAExecutor#getName()
039         */
040        @Override
041        public String getName() {
042            return "BundleJobGetCoordinatorsJPAExecutor";
043        }
044    
045        /* (non-Javadoc)
046         * @see org.apache.oozie.executor.jpa.JPAExecutor#execute(javax.persistence.EntityManager)
047         */
048        @Override
049        @SuppressWarnings("unchecked")
050        public List<CoordinatorJobBean> execute(EntityManager em) throws JPAExecutorException {
051            List<CoordinatorJobBean> coordJobBeans;
052            try {
053                Query q = em.createNamedQuery("GET_COORD_JOBS_FOR_BUNDLE");
054                q.setParameter("bundleId", bundleId);
055                coordJobBeans = q.getResultList();
056    
057                for (CoordinatorJobBean cjBean : coordJobBeans) {
058                    cjBean.setStatus(cjBean.getStatus());
059                    cjBean.setTimeUnit(cjBean.getTimeUnit());
060                    cjBean.setStartTimestamp(cjBean.getStartTimestamp());
061                    cjBean.setEndTimestamp(cjBean.getEndTimestamp());
062                    cjBean.setNextMaterializedTimestamp(cjBean.getNextMaterializedTimestamp());
063                }
064    
065                return coordJobBeans;
066            }
067            catch (Exception e) {
068                throw new JPAExecutorException(ErrorCode.E0603, e);
069            }
070        }
071    }