org.wisdom.akka.impl.Job.java Source code

Introduction

Here is the source code for org.wisdom.akka.impl.Job.java

Source

/*
 * #%L
 * Wisdom-Framework
 * %%
 * Copyright (C) 2013 - 2014 Wisdom Framework
 * %%
 * 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.
 * #L%
 */
package org.wisdom.akka.impl;

import akka.actor.Cancellable;
import com.google.common.base.Strings;
import org.joda.time.Period;
import org.joda.time.format.PeriodFormatter;
import org.joda.time.format.PeriodFormatterBuilder;
import org.wisdom.api.annotations.scheduler.Every;
import org.wisdom.api.scheduler.Scheduled;
import scala.concurrent.duration.Duration;
import scala.concurrent.duration.FiniteDuration;

import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import java.util.concurrent.TimeUnit;

/**
 * Structure holding a job.
 */
public class Job {

    public static final PeriodFormatter PERIOD_FORMATTER = new PeriodFormatterBuilder().printZeroRarelyFirst()
            .appendDays().appendSuffix("d", "d").printZeroRarelyLast().appendHours().appendSuffix("h", "h")
            .printZeroRarelyLast().appendMinutes().appendSuffix("m", "m").printZeroRarelyLast().appendSeconds()
            .appendSuffix("s", "s").toFormatter();
    private final Method method;
    private final FiniteDuration duration;
    private final Scheduled scheduled;
    private Cancellable cancellable;

    /**
     * Creates a new instance of Job.
     *
     * @param scheduled the scheduled object, must not be {@literal null}
     * @param method    the method to call on this scheduled object, must not be {@literal null}
     * @param every     the every annotation
     */
    public Job(Scheduled scheduled, Method method, Every every) {
        this.method = method;
        this.scheduled = scheduled;
        this.duration = getDurationFromPeriod(every);
    }

    /**
     * Translates the given (Joda) Period to (Scala) duration.
     *
     * @param period the period
     * @return the duration representing the same amount of time
     */
    public static FiniteDuration toDuration(Period period) {
        return Duration.create(period.getDays(), TimeUnit.DAYS)
                .plus(Duration.create(period.getHours(), TimeUnit.HOURS)
                        .plus(Duration.create(period.getMinutes(), TimeUnit.MINUTES)
                                .plus(Duration.create(period.getSeconds(), TimeUnit.SECONDS))));
    }

    /**
     * Parses the given String as Period.
     * The given String must follows this syntax: XdYhZmTs.
     *
     * @param every the period to parse
     * @return the parsed period
     */
    public static FiniteDuration getDurationFromPeriod(Every every) {
        if (!Strings.isNullOrEmpty(every.value())) {
            return toDuration(PERIOD_FORMATTER.parsePeriod(every.value()));
        } else {
            // We have a period and time unit.
            if (every.period() <= 0) {
                throw new IllegalArgumentException(
                        "Invalid period for an @Every task, the period must be greater " + "than 0.");
            }
            return new FiniteDuration(every.period(), every.unit());
        }
    }

    public Method method() {
        return method;
    }

    public FiniteDuration duration() {
        return duration;
    }

    /**
     * Gets the runnable invoking the scheduled method.
     *
     * @return the runnable.
     */
    public Runnable function() {
        return new Runnable() {
            @Override
            public void run() {
                try {
                    method.invoke(scheduled);
                } catch (IllegalAccessException e) {
                    AkkaScheduler.getLogger().error("Error while accessing to the scheduled method {}.{}",
                            scheduled.getClass().getName(), method.getName(), e);
                } catch (InvocationTargetException e) {
                    AkkaScheduler.getLogger().error("Error in scheduled method {}.{}",
                            scheduled.getClass().getName(), method.getName(), e);
                }
            }
        };
    }

    /**
     * Method called when the job is submitted. It provides a reference to the cancellable object.
     *
     * @param cancellable the object used to cancel the task.
     */
    public void submitted(Cancellable cancellable) {
        this.cancellable = cancellable;
    }

    /**
     * @return the Cancellable object, {@literal null} if the job is not yet submitted.
     */
    public Cancellable cancellable() {
        return cancellable;
    }

    /**
     * @return the scheduled object.
     */
    public Scheduled scheduled() {
        return scheduled;
    }
}