com.effektif.workflow.api.model.AfterRelativeTime.java Source code

Introduction

Here is the source code for com.effektif.workflow.api.model.AfterRelativeTime.java

Source

/* Copyright (c) 2014, Effektif GmbH.
 * 
 * Licensed under the Apache 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.apache.org/licenses/LICENSE-2.0
 *
 * 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 com.effektif.workflow.api.model;

import org.joda.time.Days;
import org.joda.time.Hours;
import org.joda.time.LocalDateTime;
import org.joda.time.Minutes;
import org.joda.time.Months;
import org.joda.time.ReadablePeriod;
import org.joda.time.Weeks;
import org.joda.time.Years;

import com.effektif.workflow.api.bpmn.BpmnReader;
import com.effektif.workflow.api.bpmn.BpmnWriter;
import com.effektif.workflow.api.json.TypeName;
import com.effektif.workflow.api.workflow.Binding;

/**
 * @author Tom Baeyens
 */
@TypeName("after")
public class AfterRelativeTime extends RelativeTime {

    protected Integer duration;
    public String durationUnit;

    @Override
    public void readBpmn(BpmnReader r) {
        super.readBpmn(r);
        this.duration = new Integer(r.readStringAttributeEffektif("duration"));
        this.durationUnit = r.readStringAttributeEffektif("durationUnit");
    }

    @Override
    public void writeBpmn(BpmnWriter w) {
        super.writeBpmn(w);
        w.writeStringAttributeEffektif("type", getType());
        w.writeStringAttributeEffektif("duration", duration);
        w.writeStringAttributeEffektif("durationUnit", durationUnit);
    }

    public String getType() {
        return AFTER;
    }

    public Integer getDurationAsInt() {
        return duration;
    }

    public Integer getDuration() {
        return this.duration;
    }

    public void setDuration(Integer duration) {
        this.duration = duration;
    }

    public AfterRelativeTime duration(Integer duration) {
        this.duration = duration;
        return this;
    }

    public String getDurationUnit() {
        return durationUnit;
    }

    public void setDurationUnit(String durationUnit) {
        this.durationUnit = durationUnit;
    }

    public AfterRelativeTime durationUnit(String durationUnit) {
        this.durationUnit = durationUnit;
        return this;
    }

    public boolean isDayResolutionOrBigger() {
        return !(MINUTES.equals(durationUnit) || HOURS.equals(durationUnit));
    }

    public LocalDateTime resolve(LocalDateTime base) {
        if (this.duration == null || this.durationUnit == null) {
            return null;
        }

        ReadablePeriod period = null;
        if (DAYS.equals(durationUnit)) {
            period = Days.days(getDurationAsInt());
        } else if (WEEKS.equals(durationUnit)) {
            period = Weeks.weeks(getDurationAsInt());
        } else if (HOURS.equals(durationUnit)) {
            period = Hours.hours(getDurationAsInt());
        } else if (MONTHS.equals(durationUnit)) {
            period = Months.months(getDurationAsInt());
        } else if (YEARS.equals(durationUnit)) {
            period = Years.years(getDurationAsInt());
        } else if (MINUTES.equals(durationUnit)) {
            period = Minutes.minutes(getDurationAsInt());
        } else {
            return null;
        }

        LocalDateTime time = base.plus(period);

        if (atHour != null) {
            LocalDateTime atTime = time.withTime(atHour, atMinute != null ? atMinute : 0, 0, 0);
            if (atTime.isBefore(time)) {
                time = atTime.plusDays(1);
            } else {
                time = atTime;
            }
        } else if (isDayResolutionOrBigger()) {
            time = time.withTime(23, 59, 59, 999);
        }

        return time;
    }

    @Override
    public boolean valid() {
        if (duration == null || durationUnit == null) {
            return false;
        }
        return MINUTES.equals(durationUnit) || HOURS.equals(durationUnit) || DAYS.equals(durationUnit)
                || WEEKS.equals(durationUnit) || MONTHS.equals(durationUnit) || YEARS.equals(durationUnit);
    }

    @Override
    public AfterRelativeTime base(Binding<LocalDateTime> base) {
        return (AfterRelativeTime) super.base(base);
    }

}