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 java.sql.Timestamp; 021 022 import javax.persistence.EntityManager; 023 import javax.persistence.Query; 024 025 import org.apache.oozie.ErrorCode; 026 027 /** 028 * Count the number of Workflow children of a parent Workflow or Coordinator that are not ready to be purged 029 */ 030 public class WorkflowJobsCountNotForPurgeFromParentIdJPAExecutor implements JPAExecutor<Long> { 031 032 private static final long DAY_IN_MS = 24 * 60 * 60 * 1000; 033 private long olderThanDays; 034 private String parentId; 035 036 public WorkflowJobsCountNotForPurgeFromParentIdJPAExecutor(long olderThanDays, String parentId) { 037 this.olderThanDays = olderThanDays; 038 this.parentId = parentId; 039 } 040 041 /* (non-Javadoc) 042 * @see org.apache.oozie.executor.jpa.JPAExecutor#getName() 043 */ 044 @Override 045 public String getName() { 046 return "WorkflowChildrenCountNotForPurgeJPAExecutor"; 047 } 048 049 /* (non-Javadoc) 050 * @see org.apache.oozie.executor.jpa.JPAExecutor#execute(javax.persistence.EntityManager) 051 */ 052 @Override 053 @SuppressWarnings("unchecked") 054 public Long execute(EntityManager em) throws JPAExecutorException { 055 Long count = 0L; 056 try { 057 Timestamp maxEndTime = new Timestamp(System.currentTimeMillis() - (olderThanDays * DAY_IN_MS)); 058 Query jobQ = em.createNamedQuery("GET_WORKFLOWS_COUNT_WITH_PARENT_ID_NOT_READY_FOR_PURGE"); 059 jobQ.setParameter("parentId", parentId); 060 jobQ.setParameter("endTime", maxEndTime); 061 count = (Long) jobQ.getSingleResult(); 062 } 063 catch (Exception e) { 064 throw new JPAExecutorException(ErrorCode.E0603, e.getMessage(), e); 065 } 066 return count; 067 } 068 069 }