001 /** 002 * Copyright 2004-2012 The Kuali Foundation 003 * 004 * Licensed under the Educational Community License, Version 2.0 (the "License"); 005 * you may not use this file except in compliance with the License. 006 * You may obtain a copy of the License at 007 * 008 * http://www.opensource.org/licenses/ecl2.php 009 * 010 * Unless required by applicable law or agreed to in writing, software 011 * distributed under the License is distributed on an "AS IS" BASIS, 012 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 013 * See the License for the specific language governing permissions and 014 * limitations under the License. 015 */ 016 package org.kuali.hr.time.calendar.dao; 017 018 import java.text.DateFormat; 019 import java.text.ParseException; 020 import java.text.SimpleDateFormat; 021 import java.util.ArrayList; 022 import java.util.Collection; 023 import java.util.Date; 024 import java.util.List; 025 026 import org.apache.ojb.broker.query.Criteria; 027 import org.apache.ojb.broker.query.Query; 028 import org.apache.ojb.broker.query.QueryByCriteria; 029 import org.apache.ojb.broker.query.QueryFactory; 030 import org.apache.ojb.broker.query.ReportQueryByCriteria; 031 import org.joda.time.DateTime; 032 import org.kuali.hr.time.calendar.CalendarEntries; 033 import org.kuali.rice.core.framework.persistence.ojb.dao.PlatformAwareDaoBaseOjb; 034 035 public class CalendarEntriesDaoSpringOjbImpl extends PlatformAwareDaoBaseOjb implements CalendarEntriesDao { 036 037 038 public void saveOrUpdate(CalendarEntries calendarEntries) { 039 this.getPersistenceBrokerTemplate().store(calendarEntries); 040 } 041 042 public CalendarEntries getCalendarEntries(String hrCalendarEntriesId) { 043 Criteria currentRecordCriteria = new Criteria(); 044 currentRecordCriteria.addEqualTo("hrCalendarEntriesId", hrCalendarEntriesId); 045 046 return (CalendarEntries) this.getPersistenceBrokerTemplate().getObjectByQuery(QueryFactory.newQuery(CalendarEntries.class, currentRecordCriteria)); 047 } 048 049 @Override 050 public CalendarEntries getCalendarEntriesByIdAndPeriodEndDate(String hrCalendarId, Date endPeriodDate) { 051 Criteria root = new Criteria(); 052 root.addEqualTo("hrCalendarId", hrCalendarId); 053 root.addEqualTo("endPeriodDateTime", endPeriodDate); 054 055 Query query = QueryFactory.newQuery(CalendarEntries.class, root); 056 CalendarEntries pce = (CalendarEntries) this.getPersistenceBrokerTemplate().getObjectByQuery(query); 057 return pce; 058 } 059 060 @Override 061 public CalendarEntries getCurrentCalendarEntriesByCalendarId( 062 String hrCalendarId, Date currentDate) { 063 Criteria root = new Criteria(); 064 // Criteria beginDate = new Criteria(); 065 // Criteria endDate = new Criteria(); 066 067 // beginDate.addEqualToField("hrPyCalendarId", Criteria.PARENT_QUERY_PREFIX + "hrPyCalendarId"); 068 // beginDate.addLessOrEqualThan("beginPeriodDateTime", currentDate); 069 // ReportQueryByCriteria beginDateSubQuery = QueryFactory.newReportQuery(CalendarEntries.class, beginDate); 070 // beginDateSubQuery.setAttributes(new String[] { "max(beginPeriodDateTime)" }); 071 072 // endDate.addEqualToField("hrPyCalendarId", Criteria.PARENT_QUERY_PREFIX + "hrPyCalendarId"); 073 // endDate.addGreaterOrEqualThan("endPeriodDateTime", currentDate); 074 // ReportQueryByCriteria endDateSubQuery = QueryFactory.newReportQuery(CalendarEntries.class, endDate); 075 // endDateSubQuery.setAttributes(new String[] { "min(endPeriodDateTime)" }); 076 077 root.addEqualTo("hrCalendarId", hrCalendarId); 078 //root.addEqualTo("beginPeriodDateTime", beginDateSubQuery); 079 root.addLessOrEqualThan("beginPeriodDateTime", currentDate); 080 root.addGreaterThan("endPeriodDateTime", currentDate); 081 // root.addEqualTo("endPeriodDateTime", endDateSubQuery); 082 083 Query query = QueryFactory.newQuery(CalendarEntries.class, root); 084 085 CalendarEntries pce = (CalendarEntries) this.getPersistenceBrokerTemplate().getObjectByQuery(query); 086 return pce; 087 } 088 089 @Override 090 public CalendarEntries getNextCalendarEntriesByCalendarId(String hrCalendarId, CalendarEntries calendarEntries) { 091 Criteria root = new Criteria(); 092 Criteria beginDate = new Criteria(); 093 Criteria endDate = new Criteria(); 094 095 beginDate.addEqualToField("hrCalendarId", Criteria.PARENT_QUERY_PREFIX + "hrCalendarId"); 096 beginDate.addGreaterThan("beginPeriodDateTime", calendarEntries.getBeginPeriodDateTime()); 097 ReportQueryByCriteria beginDateSubQuery = QueryFactory.newReportQuery(CalendarEntries.class, beginDate); 098 beginDateSubQuery.setAttributes(new String[]{"min(beginPeriodDateTime)"}); 099 100 endDate.addEqualToField("hrCalendarId", Criteria.PARENT_QUERY_PREFIX + "hrCalendarId"); 101 endDate.addGreaterThan("endPeriodDateTime", calendarEntries.getEndPeriodDateTime()); 102 ReportQueryByCriteria endDateSubQuery = QueryFactory.newReportQuery(CalendarEntries.class, endDate); 103 endDateSubQuery.setAttributes(new String[]{"min(endPeriodDateTime)"}); 104 105 root.addEqualTo("hrCalendarId", hrCalendarId); 106 root.addEqualTo("beginPeriodDateTime", beginDateSubQuery); 107 root.addEqualTo("endPeriodDateTime", endDateSubQuery); 108 109 Query query = QueryFactory.newQuery(CalendarEntries.class, root); 110 111 CalendarEntries pce = (CalendarEntries) this.getPersistenceBrokerTemplate().getObjectByQuery(query); 112 return pce; 113 } 114 115 @Override 116 public CalendarEntries getPreviousCalendarEntriesByCalendarId(String hrCalendarId, CalendarEntries calendarEntries) { 117 Criteria root = new Criteria(); 118 Criteria beginDate = new Criteria(); 119 Criteria endDate = new Criteria(); 120 121 beginDate.addEqualToField("hrCalendarId", Criteria.PARENT_QUERY_PREFIX + "hrCalendarId"); 122 beginDate.addLessThan("beginPeriodDateTime", calendarEntries.getBeginPeriodDateTime()); 123 ReportQueryByCriteria beginDateSubQuery = QueryFactory.newReportQuery(CalendarEntries.class, beginDate); 124 beginDateSubQuery.setAttributes(new String[]{"max(beginPeriodDateTime)"}); 125 126 endDate.addEqualToField("hrCalendarId", Criteria.PARENT_QUERY_PREFIX + "hrCalendarId"); 127 endDate.addLessThan("endPeriodDateTime", calendarEntries.getEndPeriodDateTime()); 128 ReportQueryByCriteria endDateSubQuery = QueryFactory.newReportQuery(CalendarEntries.class, endDate); 129 endDateSubQuery.setAttributes(new String[]{"max(endPeriodDateTime)"}); 130 131 root.addEqualTo("hrCalendarId", hrCalendarId); 132 root.addEqualTo("beginPeriodDateTime", beginDateSubQuery); 133 root.addEqualTo("endPeriodDateTime", endDateSubQuery); 134 135 Query query = QueryFactory.newQuery(CalendarEntries.class, root); 136 137 CalendarEntries pce = (CalendarEntries) this.getPersistenceBrokerTemplate().getObjectByQuery(query); 138 return pce; 139 } 140 141 public List<CalendarEntries> getCurrentCalendarEntryNeedsScheduled(int thresholdDays, Date asOfDate) { 142 DateTime current = new DateTime(asOfDate.getTime()); 143 DateTime windowStart = current.minusDays(thresholdDays); 144 DateTime windowEnd = current.plusDays(thresholdDays); 145 146 Criteria root = new Criteria(); 147 148 root.addGreaterOrEqualThan("beginPeriodDateTime", windowStart.toDate()); 149 root.addLessOrEqualThan("beginPeriodDateTime", windowEnd.toDate()); 150 151 Query query = QueryFactory.newQuery(CalendarEntries.class, root); 152 Collection c = this.getPersistenceBrokerTemplate().getCollectionByQuery(query); 153 154 List<CalendarEntries> pce = new ArrayList<CalendarEntries>(c.size()); 155 pce.addAll(c); 156 157 return pce; 158 } 159 160 public List<CalendarEntries> getFutureCalendarEntries(String hrCalendarId, Date currentDate, int numberOfEntries) { 161 Criteria root = new Criteria(); 162 root.addEqualTo("hrCalendarId", hrCalendarId); 163 root.addGreaterOrEqualThan("beginPeriodDateTime", currentDate); 164 QueryByCriteria q = QueryFactory.newReportQuery(CalendarEntries.class, root); 165 q.addOrderByAscending("beginPeriodDateTime"); 166 q.setStartAtIndex(1); 167 q.setEndAtIndex(numberOfEntries); 168 List<CalendarEntries> calendarEntries = new ArrayList<CalendarEntries>(this.getPersistenceBrokerTemplate().getCollectionByQuery(q)); 169 return calendarEntries; 170 } 171 172 public CalendarEntries getCalendarEntriesByBeginAndEndDate(Date beginPeriodDate, Date endPeriodDate) { 173 Criteria root = new Criteria(); 174 root.addEqualTo("beginPeriodDateTime", beginPeriodDate); 175 root.addEqualTo("endPeriodDateTime", endPeriodDate); 176 Query query = QueryFactory.newQuery(CalendarEntries.class, root); 177 178 return (CalendarEntries) this.getPersistenceBrokerTemplate().getObjectByQuery(query); 179 } 180 181 public List<CalendarEntries> getAllCalendarEntriesForCalendarId(String hrCalendarId) { 182 Criteria root = new Criteria(); 183 root.addEqualTo("hrCalendarId", hrCalendarId); 184 Query query = QueryFactory.newQuery(CalendarEntries.class, root); 185 List<CalendarEntries> ceList = new ArrayList<CalendarEntries> (this.getPersistenceBrokerTemplate().getCollectionByQuery(query)); 186 return ceList; 187 } 188 189 public List<CalendarEntries> getAllCalendarEntriesForCalendarIdAndYear(String hrCalendarId, String year) { 190 Criteria crit = new Criteria(); 191 List<CalendarEntries> ceList = new ArrayList<CalendarEntries>(); 192 try { 193 crit.addEqualTo("hrCalendarId", hrCalendarId); 194 DateFormat df = new SimpleDateFormat("yyyy"); 195 java.util.Date cYear = df.parse(year); 196 String nextYear = Integer.toString((Integer.parseInt(year) + 1)); 197 java.util.Date nYear = df.parse(nextYear); 198 199 crit.addGreaterOrEqualThan("beginPeriodDateTime", cYear); 200 crit.addLessThan("beginPeriodDateTime", nYear ); 201 QueryByCriteria query = new QueryByCriteria(CalendarEntries.class, crit); 202 Collection c = this.getPersistenceBrokerTemplate().getCollectionByQuery(query); 203 if (c != null) { 204 ceList.addAll(c); 205 } 206 } catch (ParseException e) { 207 e.printStackTrace(); 208 } 209 return ceList; 210 } 211 212 213 }