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.sql.Timestamp;
018    import java.util.ArrayList;
019    import java.util.List;
020    
021    import javax.persistence.EntityManager;
022    import javax.persistence.Query;
023    
024    import org.apache.oozie.CoordinatorActionBean;
025    import org.apache.oozie.ErrorCode;
026    import org.apache.oozie.util.ParamChecker;
027    
028    public class CoordActionsGetForRecoveryJPAExecutor implements JPAExecutor<List<CoordinatorActionBean>> {
029    
030        private long checkAgeSecs = 0;
031    
032        public CoordActionsGetForRecoveryJPAExecutor(final long checkAgeSecs) {
033            ParamChecker.notNull(checkAgeSecs, "checkAgeSecs");
034            this.checkAgeSecs = checkAgeSecs;
035        }
036    
037        /* (non-Javadoc)
038         * @see org.apache.oozie.executor.jpa.JPAExecutor#getName()
039         */
040        @Override
041        public String getName() {
042            return "CoordActionsGetForRecoveryJPAExecutor";
043        }
044    
045        /* (non-Javadoc)
046         * @see org.apache.oozie.executor.jpa.JPAExecutor#execute(javax.persistence.EntityManager)
047         */
048        @Override
049        public List<CoordinatorActionBean> execute(EntityManager em) throws JPAExecutorException {
050            List<CoordinatorActionBean> allActions = new ArrayList<CoordinatorActionBean>();
051            List<CoordinatorActionBean> actions;
052            try {
053                Query q = em.createNamedQuery("GET_COORD_ACTIONS_FOR_RECOVERY_OLDER_THAN");
054                Timestamp ts = new Timestamp(System.currentTimeMillis() - this.checkAgeSecs * 1000);
055                q.setParameter("lastModifiedTime", ts);
056                actions = q.getResultList();
057                for (CoordinatorActionBean action : actions) {
058                    allActions.add(action);
059                }
060    
061                q = em.createNamedQuery("GET_COORD_ACTIONS_WAITING_SUBMITTED_OLDER_THAN");
062                q.setParameter("lastModifiedTime", ts);
063                actions = q.getResultList();
064                for (CoordinatorActionBean action : actions) {
065                    allActions.add(action);
066                }
067                
068                return allActions;
069            }
070            catch (IllegalStateException e) {
071                throw new JPAExecutorException(ErrorCode.E0601, e.getMessage(), e);
072            }
073        }
074    }