Search in sources :

Example 21 with DistributionQueueEntry

use of org.apache.sling.distribution.queue.DistributionQueueEntry in project sling by apache.

the class JobHandlingDistributionQueue method remove.

public DistributionQueueEntry remove(@Nonnull String id) {
    boolean removed = false;
    Job job = getJob(id);
    DistributionQueueEntry entry = null;
    if (job != null) {
        entry = JobHandlingUtils.getEntry(job);
        removed = jobManager.removeJobById(job.getId());
    }
    log.debug("item with id {} removed from the queue: {}", id, removed);
    return entry;
}
Also used : DistributionQueueEntry(org.apache.sling.distribution.queue.DistributionQueueEntry) Job(org.apache.sling.event.jobs.Job)

Example 22 with DistributionQueueEntry

use of org.apache.sling.distribution.queue.DistributionQueueEntry in project sling by apache.

the class DistributionAgentQueueServlet method deleteItems.

private void deleteItems(ResourceResolver resourceResolver, DistributionQueue queue, String[] ids) {
    for (String id : ids) {
        DistributionQueueEntry entry = queue.getItem(id);
        deleteItem(resourceResolver, queue, entry);
    }
}
Also used : DistributionQueueEntry(org.apache.sling.distribution.queue.DistributionQueueEntry)

Example 23 with DistributionQueueEntry

use of org.apache.sling.distribution.queue.DistributionQueueEntry in project sling by apache.

the class AsyncDeliveryDispatchingStrategyTest method testAddWithOneItemInRunningQueues.

@Test
public void testAddWithOneItemInRunningQueues() throws Exception {
    Map<String, String> deliveryMappings = new HashMap<String, String>();
    String queue1 = "queue1";
    deliveryMappings.put(queue1, "delivery1");
    String queue2 = "queue2";
    deliveryMappings.put(queue2, "delivery2");
    AsyncDeliveryDispatchingStrategy asyncDeliveryDispatchingStrategy = new AsyncDeliveryDispatchingStrategy(deliveryMappings);
    // setup package
    SharedDistributionPackage distributionPackage = mock(SharedDistributionPackage.class);
    when(distributionPackage.getId()).thenReturn("1221312");
    DistributionPackageInfo info = new DistributionPackageInfo("dummy", new HashMap<String, Object>());
    when(distributionPackage.getInfo()).thenReturn(info);
    // setup queues
    DistributionQueueProvider queueProvider = mock(DistributionQueueProvider.class);
    DistributionQueue dq1 = mock(DistributionQueue.class);
    DistributionQueueItemStatus status = mock(DistributionQueueItemStatus.class);
    DistributionQueueEntry entry = new DistributionQueueEntry("1242112", new DistributionQueueItem(distributionPackage.getId(), new HashMap<String, Object>()), status);
    when(dq1.add(any(DistributionQueueItem.class))).thenReturn(entry);
    DistributionQueueStatus status1 = new DistributionQueueStatus(1, DistributionQueueState.RUNNING);
    when(dq1.getStatus()).thenReturn(status1);
    when(queueProvider.getQueue(queue1)).thenReturn(dq1);
    DistributionQueue dq2 = mock(DistributionQueue.class);
    when(dq2.add(any(DistributionQueueItem.class))).thenReturn(entry);
    DistributionQueueStatus status2 = new DistributionQueueStatus(1, DistributionQueueState.RUNNING);
    when(dq2.getStatus()).thenReturn(status2);
    when(queueProvider.getQueue(queue2)).thenReturn(dq2);
    Iterable<DistributionQueueItemStatus> statuses = asyncDeliveryDispatchingStrategy.add(distributionPackage, queueProvider);
    assertNotNull(statuses);
}
Also used : DistributionQueueStatus(org.apache.sling.distribution.queue.DistributionQueueStatus) DistributionQueueItemStatus(org.apache.sling.distribution.queue.DistributionQueueItemStatus) DistributionPackageInfo(org.apache.sling.distribution.packaging.DistributionPackageInfo) HashMap(java.util.HashMap) DistributionQueue(org.apache.sling.distribution.queue.DistributionQueue) DistributionQueueEntry(org.apache.sling.distribution.queue.DistributionQueueEntry) DistributionQueueItem(org.apache.sling.distribution.queue.DistributionQueueItem) SharedDistributionPackage(org.apache.sling.distribution.packaging.impl.SharedDistributionPackage) DistributionQueueProvider(org.apache.sling.distribution.queue.DistributionQueueProvider) Test(org.junit.Test)

Example 24 with DistributionQueueEntry

use of org.apache.sling.distribution.queue.DistributionQueueEntry in project sling by apache.

the class SimpleDistributionQueueCheckpointTest method testRunWithNothingInQueue.

@Test
public void testRunWithNothingInQueue() throws Exception {
    DistributionQueue queue = mock(DistributionQueue.class);
    Iterable<DistributionQueueEntry> items = new LinkedList<DistributionQueueEntry>();
    when(queue.getItems(0, -1)).thenReturn(items);
    File checkpointDirectory = FileUtils.getTempDirectory();
    SimpleDistributionQueueCheckpoint simpleDistributionQueueCheckpoint = new SimpleDistributionQueueCheckpoint(queue, checkpointDirectory);
    simpleDistributionQueueCheckpoint.run();
}
Also used : DistributionQueue(org.apache.sling.distribution.queue.DistributionQueue) DistributionQueueEntry(org.apache.sling.distribution.queue.DistributionQueueEntry) File(java.io.File) LinkedList(java.util.LinkedList) Test(org.junit.Test)

Example 25 with DistributionQueueEntry

use of org.apache.sling.distribution.queue.DistributionQueueEntry in project sling by apache.

the class SimpleDistributionQueueCheckpointTest method testRunWithOneItemInTheQueue.

@Test
public void testRunWithOneItemInTheQueue() throws Exception {
    DistributionQueue queue = mock(DistributionQueue.class);
    String queueName = "sample-queue";
    when(queue.getName()).thenReturn(queueName);
    LinkedList<DistributionQueueEntry> entries = new LinkedList<DistributionQueueEntry>();
    Map<String, Object> base = new HashMap<String, Object>();
    base.put("here", "there");
    base.put("foo", "bar");
    base.put("multi", new String[] { "1", "2" });
    entries.add(new DistributionQueueEntry("123", new DistributionQueueItem("pid123", base), new DistributionQueueItemStatus(DistributionQueueItemState.QUEUED, queueName)));
    when(queue.getItems(0, -1)).thenReturn(entries);
    File checkpointDirectory = FileUtils.getTempDirectory();
    SimpleDistributionQueueCheckpoint simpleDistributionQueueCheckpoint = new SimpleDistributionQueueCheckpoint(queue, checkpointDirectory);
    simpleDistributionQueueCheckpoint.run();
    File checkpointFile = new File(checkpointDirectory, "sample-queue-checkpoint");
    assertTrue(checkpointFile.exists());
}
Also used : DistributionQueueItemStatus(org.apache.sling.distribution.queue.DistributionQueueItemStatus) DistributionQueue(org.apache.sling.distribution.queue.DistributionQueue) DistributionQueueEntry(org.apache.sling.distribution.queue.DistributionQueueEntry) HashMap(java.util.HashMap) File(java.io.File) LinkedList(java.util.LinkedList) DistributionQueueItem(org.apache.sling.distribution.queue.DistributionQueueItem) Test(org.junit.Test)

Aggregations

DistributionQueueEntry (org.apache.sling.distribution.queue.DistributionQueueEntry)30 DistributionQueueItem (org.apache.sling.distribution.queue.DistributionQueueItem)20 DistributionQueue (org.apache.sling.distribution.queue.DistributionQueue)18 DistributionQueueItemStatus (org.apache.sling.distribution.queue.DistributionQueueItemStatus)16 Test (org.junit.Test)13 ArrayList (java.util.ArrayList)6 DistributionPackage (org.apache.sling.distribution.packaging.DistributionPackage)6 HashMap (java.util.HashMap)5 LinkedList (java.util.LinkedList)5 DistributionQueueProvider (org.apache.sling.distribution.queue.DistributionQueueProvider)5 File (java.io.File)4 DistributionException (org.apache.sling.distribution.common.DistributionException)4 Map (java.util.Map)3 CheckForNull (javax.annotation.CheckForNull)3 DistributionAgent (org.apache.sling.distribution.agent.DistributionAgent)3 DistributionPackageInfo (org.apache.sling.distribution.packaging.DistributionPackageInfo)3 SharedDistributionPackage (org.apache.sling.distribution.packaging.impl.SharedDistributionPackage)3 DistributionQueueStatus (org.apache.sling.distribution.queue.DistributionQueueStatus)3 Result (org.apache.sling.hc.api.Result)3 DistributionPackageBuilder (org.apache.sling.distribution.packaging.DistributionPackageBuilder)2