001 /**
002 * Copyright 2004-2013 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 @SuppressWarnings("unchecked")
182 public List<CalendarEntries> getCalendarEntriesEndingBetweenBeginAndEndDate(String hrCalendarId, Date beginDate, Date endDate) {
183 List<CalendarEntries> results = new ArrayList<CalendarEntries>();
184
185 Criteria root = new Criteria();
186
187 root.addEqualTo("hrCalendarId", hrCalendarId);
188 root.addGreaterOrEqualThan("endPeriodDateTime", beginDate);
189 root.addLessOrEqualThan("endPeriodDateTime", endDate);
190 Query query = QueryFactory.newQuery(CalendarEntries.class, root);
191
192 results.addAll(getPersistenceBrokerTemplate().getCollectionByQuery(query));
193
194 return results;
195 }
196
197 public List<CalendarEntries> getAllCalendarEntriesForCalendarId(String hrCalendarId) {
198 Criteria root = new Criteria();
199 root.addEqualTo("hrCalendarId", hrCalendarId);
200 Query query = QueryFactory.newQuery(CalendarEntries.class, root);
201 List<CalendarEntries> ceList = new ArrayList<CalendarEntries> (this.getPersistenceBrokerTemplate().getCollectionByQuery(query));
202 return ceList;
203 }
204
205 public List<CalendarEntries> getAllCalendarEntriesForCalendarIdUpToCutOffTime(String hrCalendarId, Date cutOffTime) {
206 Criteria root = new Criteria();
207 root.addEqualTo("hrCalendarId", hrCalendarId);
208 root.addLessOrEqualThan("endPeriodDateTime", cutOffTime);
209 Query query = QueryFactory.newQuery(CalendarEntries.class, root);
210 List<CalendarEntries> ceList = new ArrayList<CalendarEntries> (this.getPersistenceBrokerTemplate().getCollectionByQuery(query));
211 return ceList;
212 }
213
214 public List<CalendarEntries> getAllCalendarEntriesForCalendarIdAndYear(String hrCalendarId, String year) {
215 Criteria crit = new Criteria();
216 List<CalendarEntries> ceList = new ArrayList<CalendarEntries>();
217 try {
218 crit.addEqualTo("hrCalendarId", hrCalendarId);
219 DateFormat df = new SimpleDateFormat("yyyy");
220 java.util.Date cYear = df.parse(year);
221 String nextYear = Integer.toString((Integer.parseInt(year) + 1));
222 java.util.Date nYear = df.parse(nextYear);
223
224 crit.addGreaterOrEqualThan("beginPeriodDateTime", cYear);
225 crit.addLessThan("beginPeriodDateTime", nYear );
226 QueryByCriteria query = new QueryByCriteria(CalendarEntries.class, crit);
227 Collection c = this.getPersistenceBrokerTemplate().getCollectionByQuery(query);
228 if (c != null) {
229 ceList.addAll(c);
230 }
231 } catch (ParseException e) {
232 e.printStackTrace();
233 }
234 return ceList;
235 }
236
237
238 }