1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16 package org.kuali.hr.time.batch.web;
17
18 import java.util.HashMap;
19 import java.util.List;
20 import java.util.Map;
21
22 import javax.servlet.http.HttpServletRequest;
23 import javax.servlet.http.HttpServletResponse;
24
25 import org.apache.commons.lang.StringUtils;
26 import org.apache.log4j.Logger;
27 import org.apache.struts.action.ActionForm;
28 import org.apache.struts.action.ActionForward;
29 import org.apache.struts.action.ActionMapping;
30 import org.kuali.hr.time.base.web.TkAction;
31 import org.kuali.hr.time.batch.BatchApproveMissedPunchJob;
32 import org.kuali.hr.time.batch.BatchApproveMissedPunchJobRunnable;
33 import org.kuali.hr.time.batch.BatchJob;
34 import org.kuali.hr.time.batch.BatchJobEntry;
35 import org.kuali.hr.time.batch.EmployeeApprovalBatchJob;
36 import org.kuali.hr.time.batch.EmployeeApprovalBatchJobRunnable;
37 import org.kuali.hr.time.batch.InitiateBatchJob;
38 import org.kuali.hr.time.batch.InitiateBatchJobRunnable;
39 import org.kuali.hr.time.batch.PayPeriodEndBatchJob;
40 import org.kuali.hr.time.batch.PayPeriodEndBatchJobRunnable;
41 import org.kuali.hr.time.batch.SupervisorApprovalBatchJob;
42 import org.kuali.hr.time.batch.SupervisorApprovalBatchJobRunnable;
43 import org.kuali.hr.time.calendar.CalendarEntries;
44 import org.kuali.hr.time.service.base.TkServiceLocator;
45 import org.kuali.hr.time.util.TkConstants;
46
47 public class BatchJobAction extends TkAction {
48
49 private static final Logger LOG = Logger.getLogger(BatchJobAction.class);
50
51 public ActionForward getBatchJobEntryStatus(ActionMapping mapping, ActionForm form, HttpServletRequest request, HttpServletResponse response) throws Exception {
52 BatchJobActionForm bjaf = (BatchJobActionForm) form;
53 Map<String, Object> searchCrit = new HashMap<String, Object>();
54 searchCrit.put("tkBatchJobId", bjaf.getBatchJobId());
55 searchCrit.put("batchJobName", bjaf.getBatchJobName());
56 searchCrit.put("batchJobEntryStatus", bjaf.getBatchJobEntryStatus());
57 searchCrit.put("hrPyCalendarEntryId", bjaf.getHrPyCalendarEntryId());
58 searchCrit.put("ipAddress", bjaf.getIpAddress());
59 searchCrit.put("documentId", bjaf.getDocumentId());
60 searchCrit.put("principalId", bjaf.getPrincipalId());
61
62 bjaf.setBatchJobEntries(TkServiceLocator.getBatchJobEntryService().getBatchJobEntries(searchCrit));
63
64 return mapping.findForward("basic");
65 }
66
67 public ActionForward changeIpAddress(ActionMapping mapping, ActionForm form, HttpServletRequest request, HttpServletResponse response) throws Exception {
68 BatchJobActionForm bjaf = (BatchJobActionForm) form;
69
70 BatchJobEntry batchJobEntry = TkServiceLocator.getBatchJobEntryService().getBatchJobEntry(Long.valueOf(bjaf.getTkBatchJobEntryId()));
71 batchJobEntry.setIpAddress(bjaf.getIpToChange());
72 TkServiceLocator.getBatchJobEntryService().saveBatchJobEntry(batchJobEntry);
73
74 return mapping.findForward("basic");
75 }
76
77 public ActionForward runBatchJob(ActionMapping mapping, ActionForm form, HttpServletRequest request, HttpServletResponse response) throws Exception {
78 BatchJobActionForm bjaf = (BatchJobActionForm) form;
79
80 createBatchJobEntry(bjaf.getSelectedBatchJob(), bjaf.getHrPyCalendarEntryId());
81
82 runBatchJobEntry(bjaf.getSelectedBatchJob(), bjaf.getHrPyCalendarEntryId());
83
84 return mapping.findForward("basic");
85 }
86
87 private void createBatchJobEntry(String selectedBatchJob, String hrPyCalendarEntryId) {
88 Map<String, Object> searchCrit = new HashMap<String, Object>();
89 searchCrit.put("batchJobName", selectedBatchJob);
90 searchCrit.put("hrPyCalendarEntryId", hrPyCalendarEntryId);
91 searchCrit.put("batchJobEntryStatus", TkConstants.BATCH_JOB_ENTRY_STATUS.SCHEDULED);
92
93 List<BatchJobEntry> batchJobEntries = TkServiceLocator.getBatchJobEntryService().getBatchJobEntries(searchCrit);
94
95 if (batchJobEntries.isEmpty()) {
96 CalendarEntries calendarEntry = TkServiceLocator.getCalendarEntriesService().getCalendarEntries(hrPyCalendarEntryId);
97
98 if (StringUtils.equals(selectedBatchJob, TkConstants.BATCH_JOB_NAMES.APPROVE)) {
99 BatchJob job = new EmployeeApprovalBatchJob(calendarEntry);
100 TkServiceLocator.getBatchJobService().saveBatchJob(job);
101 job.runJob();
102 } else if (StringUtils.equals(selectedBatchJob, TkConstants.BATCH_JOB_NAMES.PAY_PERIOD_END)) {
103 BatchJob job = new PayPeriodEndBatchJob(calendarEntry);
104 TkServiceLocator.getBatchJobService().saveBatchJob(job);
105 job.runJob();
106 } else if (StringUtils.equals(selectedBatchJob, TkConstants.BATCH_JOB_NAMES.SUPERVISOR_APPROVAL)) {
107 BatchJob job = new SupervisorApprovalBatchJob(calendarEntry);
108 TkServiceLocator.getBatchJobService().saveBatchJob(job);
109 job.runJob();
110 } else if (StringUtils.equals(selectedBatchJob, TkConstants.BATCH_JOB_NAMES.INITIATE)) {
111 BatchJob job = new InitiateBatchJob(calendarEntry);
112 TkServiceLocator.getBatchJobService().saveBatchJob(job);
113 job.runJob();
114 } else if (StringUtils.equals(selectedBatchJob, TkConstants.BATCH_JOB_NAMES.BATCH_APPROVE_MISSED_PUNCH)) {
115 BatchJob job = new BatchApproveMissedPunchJob(calendarEntry);
116 TkServiceLocator.getBatchJobService().saveBatchJob(job);
117 job.runJob();
118 }
119 }
120 }
121
122 private void runBatchJobEntry(String selectedBatchJob, String hrPyCalendarEntryId) throws Exception {
123 Map<String, Object> searchCrit = new HashMap<String, Object>();
124 searchCrit.put("batchJobName", selectedBatchJob);
125 searchCrit.put("hrPyCalendarEntryId", hrPyCalendarEntryId);
126 searchCrit.put("batchJobEntryStatus", TkConstants.BATCH_JOB_ENTRY_STATUS.SCHEDULED);
127
128 List<BatchJobEntry> batchJobEntries = TkServiceLocator.getBatchJobEntryService().getBatchJobEntries(searchCrit);
129
130 for (BatchJobEntry entry : batchJobEntries) {
131 long startTime = System.currentTimeMillis();
132 LOG.debug("Before run.");
133 entry.setBatchJobEntryStatus(TkConstants.BATCH_JOB_ENTRY_STATUS.RUNNING);
134 TkServiceLocator.getBatchJobEntryService().saveBatchJobEntry(entry);
135
136 if (StringUtils.equals(entry.getBatchJobName(), TkConstants.BATCH_JOB_NAMES.APPROVE)) {
137 LOG.debug("Creating EmployeeApprovalBatchJobRunnable.");
138 new EmployeeApprovalBatchJobRunnable(entry).doWork();
139 } else if (StringUtils.equals(entry.getBatchJobName(), TkConstants.BATCH_JOB_NAMES.PAY_PERIOD_END)) {
140 LOG.debug("Creating PayPeriodEndBatchJobRunnable.");
141 new PayPeriodEndBatchJobRunnable(entry).doWork();
142 } else if (StringUtils.equals(entry.getBatchJobName(), TkConstants.BATCH_JOB_NAMES.SUPERVISOR_APPROVAL)) {
143 LOG.debug("Creating SupervisorApprovalBatchJobRunnabble.");
144 new SupervisorApprovalBatchJobRunnable(entry).doWork();
145 } else if (StringUtils.equals(entry.getBatchJobName(), TkConstants.BATCH_JOB_NAMES.INITIATE)) {
146 LOG.debug("Creating InitiateBatchJobRunnable.");
147 new InitiateBatchJobRunnable(entry).doWork();
148 } else if (StringUtils.equals(entry.getBatchJobName(), TkConstants.BATCH_JOB_NAMES.BATCH_APPROVE_MISSED_PUNCH)) {
149 LOG.debug("Creating BatchApproveMissedPunchJobRunnable.");
150 new BatchApproveMissedPunchJobRunnable(entry).doWork();
151 } else {
152 LOG.warn("Unknown BatchJobEntryRunnable found in BatchJobEntry table. Unable to create Runnable.");
153 }
154
155 long endTime = System.currentTimeMillis();
156 long runtime = endTime - startTime;
157 runtime = (runtime > 0) ? runtime : 1;
158 LOG.debug("Job finished in " + runtime / 1000 + " seconds.");
159
160 if (StringUtils.isEmpty(entry.getBatchJobException())) {
161 entry.setBatchJobEntryStatus(TkConstants.BATCH_JOB_ENTRY_STATUS.FINISHED);
162 } else {
163 entry.setBatchJobEntryStatus(TkConstants.BATCH_JOB_ENTRY_STATUS.EXCEPTION);
164 }
165 TkServiceLocator.getBatchJobEntryService().saveBatchJobEntry(entry);
166 }
167 }
168
169
170 }