org.kuali.kpme.tklm.leave.batch.AccrualJob.java Source code

Introduction

Here is the source code for org.kuali.kpme.tklm.leave.batch.AccrualJob.java

Source

/**
 * Copyright 2004-2013 The Kuali Foundation
 *
 * Licensed under the Educational Community License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.opensource.org/licenses/ecl2.php
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.kuali.kpme.tklm.leave.batch;

import java.util.List;

import org.apache.log4j.Logger;
import org.joda.time.DateTime;
import org.joda.time.LocalDate;
import org.kuali.kpme.core.assignment.Assignment;
import org.kuali.kpme.core.assignment.service.AssignmentService;
import org.kuali.kpme.core.leaveplan.LeavePlan;
import org.kuali.kpme.core.leaveplan.service.LeavePlanService;
import org.kuali.kpme.core.principal.PrincipalHRAttributes;
import org.kuali.kpme.core.principal.service.PrincipalHRAttributesService;
import org.kuali.kpme.tklm.common.TkConstants;
import org.kuali.kpme.tklm.leave.accrual.service.AccrualService;
import org.kuali.kpme.tklm.leave.calendar.service.LeaveCalendarServiceImpl;
import org.kuali.rice.core.api.config.property.ConfigContext;
import org.kuali.rice.kim.api.identity.principal.Principal;
import org.kuali.rice.kim.api.services.KimApiServiceLocator;
import org.quartz.Job;
import org.quartz.JobExecutionContext;
import org.quartz.JobExecutionException;

public class AccrualJob implements Job {

    private static final Logger LOG = Logger.getLogger(LeaveCalendarServiceImpl.class);

    private static AccrualService ACCRUAL_SERVICE;
    private static AssignmentService ASSIGNMENT_SERVICE;
    private static LeavePlanService LEAVE_PLAN_SERVICE;
    private static PrincipalHRAttributesService PRINCIPAL_HR_ATTRIBUTES_SERVICE;

    @Override
    public void execute(JobExecutionContext context) throws JobExecutionException {
        String batchUserPrincipalId = getBatchUserPrincipalId();

        if (batchUserPrincipalId != null) {
            LocalDate asOfDate = LocalDate.now();
            List<Assignment> assignments = getAssignmentService().getActiveAssignments(asOfDate);

            for (Assignment assignment : assignments) {
                if (assignment.getJob().isEligibleForLeave()) {
                    String principalId = assignment.getPrincipalId();

                    PrincipalHRAttributes principalHRAttributes = getPrincipalHRAttributesService()
                            .getPrincipalCalendar(principalId, asOfDate);
                    if (principalHRAttributes != null) {
                        LeavePlan leavePlan = getLeavePlanService().getLeavePlan(
                                principalHRAttributes.getLeavePlan(),
                                principalHRAttributes.getEffectiveLocalDate());
                        if (leavePlan != null) {
                            DateTime endDate = asOfDate.toDateTimeAtStartOfDay()
                                    .plusMonths(Integer.parseInt(leavePlan.getPlanningMonths()));
                            getAccrualService().runAccrual(principalId, asOfDate.toDateTimeAtStartOfDay(), endDate,
                                    true, batchUserPrincipalId);
                        }
                    }
                }
            }
        } else {
            String principalName = ConfigContext.getCurrentContextConfig()
                    .getProperty(TkConstants.BATCH_USER_PRINCIPAL_NAME);
            LOG.error("Could not run batch jobs due to missing batch user " + principalName);
        }
    }

    public static AccrualService getAccrualService() {
        return ACCRUAL_SERVICE;
    }

    public static void setAccrualService(AccrualService accrualService) {
        ACCRUAL_SERVICE = accrualService;
    }

    public static AssignmentService getAssignmentService() {
        return ASSIGNMENT_SERVICE;
    }

    public static void setAssignmentService(AssignmentService assignmentService) {
        ASSIGNMENT_SERVICE = assignmentService;
    }

    public static LeavePlanService getLeavePlanService() {
        return LEAVE_PLAN_SERVICE;
    }

    public static void setLeavePlanService(LeavePlanService leavePlanService) {
        LEAVE_PLAN_SERVICE = leavePlanService;
    }

    public static PrincipalHRAttributesService getPrincipalHRAttributesService() {
        return PRINCIPAL_HR_ATTRIBUTES_SERVICE;
    }

    public static void setPrincipalHRAttributesService(PrincipalHRAttributesService principalHRAttributesService) {
        PRINCIPAL_HR_ATTRIBUTES_SERVICE = principalHRAttributesService;
    }

    private String getBatchUserPrincipalId() {
        String principalName = ConfigContext.getCurrentContextConfig()
                .getProperty(TkConstants.BATCH_USER_PRINCIPAL_NAME);
        Principal principal = KimApiServiceLocator.getIdentityService().getPrincipalByPrincipalName(principalName);
        return principal == null ? null : principal.getPrincipalId();
    }

}