org.duracloud.common.queue.local.LocalTaskQueue Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of common-queue Show documentation
There is a newer version: 8.1.0
/*
 * The contents of this file are subject to the license and copyright
 * detailed in the LICENSE and NOTICE files at the root of the source
 * tree and available online at
 *
 *     http://duracloud.org/license/
 */
package org.duracloud.common.queue.local;

import java.util.Arrays;
import java.util.HashSet;
import java.util.LinkedList;
import java.util.List;
import java.util.NoSuchElementException;
import java.util.Queue;
import java.util.Set;
import java.util.concurrent.LinkedBlockingQueue;

import org.duracloud.common.queue.TaskException;
import org.duracloud.common.queue.TaskNotFoundException;
import org.duracloud.common.queue.TaskQueue;
import org.duracloud.common.queue.TimeoutException;
import org.duracloud.common.queue.task.Task;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * This class implements a local, in-memory task queue. (It is intended to use
 * for testing purposes.)
 *
 * @author Daniel Bernstein Date: Oct 24, 2013
 */
public class LocalTaskQueue implements TaskQueue {
    private Queue queue;
    private Logger log = LoggerFactory.getLogger(LocalTaskQueue.class);
    private List inprocess;
    private long completedCount = 0;
    private String name = "local-task-queue-" + System.currentTimeMillis();

    /**
     *
     */

    @Override
    public String getName() {
        return this.name;
    }

    public LocalTaskQueue() {
        queue = new LinkedBlockingQueue<>();
        inprocess = new LinkedList<>();
    }

    /* (non-Javadoc)
     * @see org.duracloud.queue.TaskQueue#put(org.duracloud.queue.task.Task)
     */
    @Override
    public synchronized void put(Task task) {
        queue.add(task);
    }

    @Override
    public synchronized void put(Task... tasks) {
        Set taskSet = new HashSet<>();
        taskSet.addAll(Arrays.asList(tasks));
        this.put(taskSet);
    }

    @Override
    public synchronized void put(Set tasks) {
        queue.addAll(tasks);
    }

    /* (non-Javadoc)
     * @see org.duracloud.queue.TaskQueue#take()
     */
    @Override
    public synchronized Task take() throws TimeoutException {
        try {
            Task task = queue.remove();
            inprocess.add(task);
            return task;
        } catch (NoSuchElementException ex) {
            throw new TimeoutException(ex);
        }
    }

    /* (non-Javadoc)
     * @see org.duracloud.queue.TaskQueue#extendVisibilityTimeout(org.duracloud.queue.task.Task)
     */
    @Override
    public void extendVisibilityTimeout(Task task) throws TaskNotFoundException {
        log.info("extending visibility time on {}", task);
    }

    /* (non-Javadoc)
     * @see org.duracloud.queue.TaskQueue#deleteTask(org.duracloud.queue.task.Task)
     */
    @Override
    public synchronized void deleteTask(Task task) throws TaskNotFoundException {
        if (!this.inprocess.contains(task)) {
            log.error("{} not found.", task);

            throw new TaskNotFoundException("task not found:" + task.toString());
        }

        this.inprocess.remove(task);
        this.completedCount++;
        log.info("{} complete", task);
    }

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

    public int getInprocessCount() {
        return this.inprocess.size();
    }

    public long getCompletedCount() {
        return completedCount;
    }

    @Override
    public Integer sizeIncludingInvisibleAndDelayed() {
        return size();
    }

    /* (non-Javadoc)
     * @see org.duracloud.queue.TaskQueue#requeue(org.duracloud.queue.task.Task)
     */
    @Override
    public void requeue(Task task) {
        this.inprocess.remove(task);
        task.incrementAttempts();
        this.queue.add(task);
    }

    @Override
    public void deleteTasks(Set tasks) throws TaskException {
        for (Task task : tasks) {
            deleteTask(task);
        }
    }

    @Override
    public Set take(int maxTasks) throws TimeoutException {
        Set set = new HashSet(1);
        set.add(this.take());
        return set;
    }

}