run.halo.app.extension.controller.DefaultQueue Maven / Gradle / Ivy

Go to download
package run.halo.app.extension.controller;

import java.time.Duration;
import java.time.Instant;
import java.util.HashSet;
import java.util.Optional;
import java.util.Set;
import java.util.concurrent.DelayQueue;
import java.util.concurrent.locks.Lock;
import java.util.concurrent.locks.ReentrantLock;
import java.util.function.Supplier;
import lombok.extern.slf4j.Slf4j;

@Slf4j
public class DefaultQueue implements RequestQueue {

    private final Lock lock;

    private final DelayQueue> queue;

    private final Supplier nowSupplier;

    private volatile boolean disposed = false;

    private final Duration minDelay;

    private final Set processing;

    private final Set dirty;

    public DefaultQueue(Supplier nowSupplier) {
        this(nowSupplier, Duration.ZERO);
    }

    public DefaultQueue(Supplier nowSupplier, Duration minDelay) {
        this.lock = new ReentrantLock();
        this.nowSupplier = nowSupplier;
        this.minDelay = minDelay;
        this.processing = new HashSet<>();
        this.dirty = new HashSet<>();
        this.queue = new DelayQueue<>();
    }

    @Override
    public boolean addImmediately(R request) {
        log.debug("Adding request {} immediately", request);
        var delayedEntry = new DelayedEntry<>(request, minDelay, nowSupplier);
        return add(delayedEntry);
    }

    @Override
    public boolean add(DelayedEntry entry) {
        lock.lock();
        try {
            if (isDisposed()) {
                return false;
            }
            log.debug("Adding request {} after {}", entry.getEntry(), entry.getRetryAfter());
            if (entry.getRetryAfter().compareTo(minDelay) < 0) {
                log.warn("Request {} will be retried after {} ms, but minimum delay is {} ms",
                    entry.getEntry(), entry.getRetryAfter().toMillis(), minDelay.toMillis());
                entry = new DelayedEntry<>(entry.getEntry(), minDelay, nowSupplier);
            }
            if (dirty.contains(entry.getEntry())) {
                var oldEntry = findOldEntry(entry);
                if (oldEntry.isEmpty()) {
                    return false;
                }
                var oldReadyAt = oldEntry.get().getReadyAt();
                var readyAt = entry.getReadyAt();
                if (!readyAt.isBefore(oldReadyAt)) {
                    return false;
                }
            }
            dirty.add(entry.getEntry());
            if (processing.contains(entry.getEntry())) {
                return false;
            }

            boolean added = queue.add(entry);
            log.debug("Added request {} after {}", entry.getEntry(), entry.getRetryAfter());
            return added;
        } finally {
            lock.unlock();
        }
    }

    @Override
    public DelayedEntry take() throws InterruptedException {
        var entry = queue.take();
        log.debug("Take request {} at {}", entry.getEntry(), Instant.now());
        lock.lockInterruptibly();
        try {
            if (isDisposed()) {
                throw new InterruptedException(
                    "Queue has been disposed. Cannot take any elements now");
            }
            processing.add(entry.getEntry());
            dirty.remove(entry.getEntry());
            return entry;
        } finally {
            lock.unlock();
        }
    }

    @Override
    public void done(R request) {
        lock.lock();
        try {
            if (isDisposed()) {
                return;
            }
            processing.remove(request);
            if (dirty.contains(request)) {
                queue.add(new DelayedEntry<>(request, minDelay, nowSupplier));
            }
        } finally {
            lock.unlock();
        }
    }

    @Override
    public long size() {
        return queue.size();
    }

    @Override
    public DelayedEntry peek() {
        return queue.peek();
    }

    @Override
    public void dispose() {
        lock.lock();
        try {
            disposed = true;
            queue.clear();
            processing.clear();
            dirty.clear();
        } finally {
            lock.unlock();
        }
    }

    @Override
    public boolean isDisposed() {
        return this.disposed;
    }

    private Optional> findOldEntry(DelayedEntry entry) {
        for (DelayedEntry element : queue) {
            if (element.equals(entry)) {
                return Optional.of(element);
            }
        }
        return Optional.empty();
    }

}