org.kuali.kpme.tklm.leave.accrual.bucket.YearToDateEarnedLeaveBalance.java Source code

Introduction

Here is the source code for org.kuali.kpme.tklm.leave.accrual.bucket.YearToDateEarnedLeaveBalance.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.accrual.bucket;

import java.util.ArrayList;
import java.util.List;

import org.joda.time.DateTime;
import org.joda.time.LocalDate;
import org.kuali.kpme.core.accrualcategory.AccrualCategory;
import org.kuali.kpme.core.earncode.EarnCode;
import org.kuali.kpme.core.principal.PrincipalHRAttributes;
import org.kuali.kpme.core.service.HrServiceLocator;
import org.kuali.kpme.core.util.HrConstants;
import org.kuali.kpme.tklm.leave.accrual.bucket.exception.MaximumBalanceException;
import org.kuali.kpme.tklm.leave.accrual.bucket.exception.NegativeBalanceException;
import org.kuali.kpme.tklm.leave.accrual.bucket.exception.UsageLimitException;
import org.kuali.kpme.tklm.leave.block.LeaveBlock;

public class YearToDateEarnedLeaveBalance extends LeaveBalance {

    private List<LeaveBlock> leaveBlocks;

    public YearToDateEarnedLeaveBalance(AccrualCategory accrualCategory, PrincipalHRAttributes principalCalendar) {
        super(accrualCategory, principalCalendar);
        asOfDate = LocalDate.now();
        leaveBlocks = new ArrayList<LeaveBlock>();
    }

    private final static String BALANCE_TYPE = "YTD_EARNED_BALANCE";

    @Override
    public void add(LeaveBlock leaveBlock) {
        EarnCode earnCode = HrServiceLocator.getEarnCodeService().getEarnCode(leaveBlock.getEarnCode(),
                LocalDate.fromDateFields(leaveBlock.getLeaveDate()));

        DateTime rolloverDate = HrServiceLocator.getLeavePlanService()
                .getFirstDayOfLeavePlan(principalCalendar.getLeavePlan(), asOfDate);
        if (earnCode != null) {
            if (leaveBlock.getLeaveDate().compareTo(asOfDate.toDate()) <= 0
                    && leaveBlock.getLeaveDate().compareTo(rolloverDate.toDate()) >= 0) {
                if (leaveBlock.getLeaveAmount().signum() >= 0) {

                    if (earnCode.getAccrualBalanceAction().equals(HrConstants.ACCRUAL_BALANCE_ACTION.USAGE)) {

                        add(leaveBlock.getLeaveAmount());

                        leaveBlocks.add(leaveBlock);

                    } else if (earnCode.getAccrualBalanceAction()
                            .equals(HrConstants.ACCRUAL_BALANCE_ACTION.ADJUSTMENT)) {
                        //does not validate against balances
                        add(leaveBlock.getLeaveAmount());

                        leaveBlocks.add(leaveBlock);
                    } else if (earnCode.getAccrualBalanceAction().equals(HrConstants.ACCRUAL_BALANCE_ACTION.NONE)) {

                        add(leaveBlock.getLeaveAmount());
                        //no balance validations, does not affect balances
                        leaveBlocks.add(leaveBlock);
                    }
                }
            }
            //otherwise, it goes into PendingLeaveBalance
        }
    }

    @Override
    public void remove(LeaveBlock leaveBlock) throws NegativeBalanceException {

        EarnCode earnCode = HrServiceLocator.getEarnCodeService().getEarnCode(leaveBlock.getEarnCode(),
                LocalDate.fromDateFields(leaveBlock.getLeaveDate()));

        DateTime rolloverDate = HrServiceLocator.getLeavePlanService()
                .getFirstDayOfLeavePlan(principalCalendar.getLeavePlan(), asOfDate);
        if (earnCode != null) {
            if (leaveBlock.getLeaveDate().compareTo(asOfDate.toDate()) <= 0
                    && leaveBlock.getLeaveDate().compareTo(rolloverDate.toDate()) >= 0) {
                if (leaveBlock.getLeaveAmount().signum() >= 0) {

                    if (earnCode.getAccrualBalanceAction().equals(HrConstants.ACCRUAL_BALANCE_ACTION.USAGE)) {

                        remove(leaveBlock.getLeaveAmount());

                        leaveBlocks.remove(leaveBlock);

                    } else if (earnCode.getAccrualBalanceAction()
                            .equals(HrConstants.ACCRUAL_BALANCE_ACTION.ADJUSTMENT)) {
                        //does not validate against balances
                        remove(leaveBlock.getLeaveAmount());

                        leaveBlocks.remove(leaveBlock);
                    } else if (earnCode.getAccrualBalanceAction().equals(HrConstants.ACCRUAL_BALANCE_ACTION.NONE)) {
                        //no balance validations, does not affect balances
                        leaveBlocks.remove(leaveBlock);
                    }
                }
            }
        }
    }

    @Override
    public String getBalanceType() {
        return BALANCE_TYPE;
    }

    @Override
    public void adjust(LeaveBlock leaveBlock)
            throws UsageLimitException, MaximumBalanceException, NegativeBalanceException {
        // TODO Auto-generated method stub

    }

    @Override
    public void clear() {
        super.clear();
        leaveBlocks.clear();
    }

}