net.ravendb.client.primitives.Timer Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of ravendb Show documentation

RavenDB Java Client

There is a newer version: 6.0.1

package net.ravendb.client.primitives;

import java.time.Duration;
import java.util.concurrent.CompletableFuture;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.ScheduledFuture;
import java.util.concurrent.TimeUnit;

public class Timer implements CleanCloseable {
    private final ExecutorService executorService;
    private final Runnable action;
    private ScheduledFuture scheduledFuture;
    private Duration period;

    public Timer(Runnable action, Duration dueTime, ExecutorService executorService) {
        this(action, dueTime, null, executorService);
    }

    public Timer(Runnable action, Duration dueTime, Duration period, ExecutorService executorService) {
        this.executorService = executorService;
        this.action = action;
        this.period = period;
        schedule(dueTime);
    }

    public void change(Duration dueTime) {
        change(dueTime, null);
    }

    public void change(Duration dueTime, Duration period) {
        this.period = period;
        this.scheduledFuture.cancel(false);
        this.schedule(dueTime);
    }

    private void schedule(Duration dueTime) {
        this.scheduledFuture = TimerService.service.schedule(() -> {
            CompletableFuture.runAsync(() -> {
                if (period != null) {
                    schedule(period);
                }
                this.action.run();
            }, executorService);
            return null;
        }, dueTime.toMillis(), TimeUnit.MILLISECONDS);
    }

    @Override
    public void close() {
        if (this.scheduledFuture != null) {
            this.scheduledFuture.cancel(false);
        }
    }
}