com.bazaarvoice.emodb.queue.client.DedupQueueClient Maven / Gradle / Ivy

Go to download
package com.bazaarvoice.emodb.queue.client;

import com.bazaarvoice.emodb.client.EmoClient;
import com.bazaarvoice.emodb.queue.api.AuthDedupQueueService;
import com.bazaarvoice.emodb.queue.api.Message;
import com.bazaarvoice.emodb.queue.api.MoveQueueStatus;
import com.bazaarvoice.ostrich.partition.PartitionKey;

import java.net.URI;
import java.time.Duration;
import java.util.Collection;
import java.util.List;

/**
 * Dedup queue client implementation that routes API calls to the EmoDB service.  The actual HTTP communication
 * is managed by the {@link EmoClient} implementation to allow for flexible usage by variety of HTTP client
 * implementations, such as Jersey.
 */
public class DedupQueueClient extends AbstractQueueClient implements AuthDedupQueueService {

    /** Must match the service name in the EmoService class. */
    static final String BASE_SERVICE_NAME = "emodb-dedupq-1";

    /** Must match the @Path annotation on the DedupQueueResource1 class. */
    static final String SERVICE_PATH = "/dedupq/1";

    public DedupQueueClient(URI endPoint, EmoClient client) {
        super(endPoint, false, client);
    }

    public DedupQueueClient(URI endPoint, boolean partitionSafe, EmoClient client) {
        super(endPoint, partitionSafe, client);
    }

    // Compared to QueueClient, the DedupQueueClient requires partition-based routing for more methods.

    @Override
    public long getMessageCount(String apiKey, @PartitionKey String queue) {
        return getMessageCountUpTo(apiKey, queue, Long.MAX_VALUE);
    }

    @Override
    public long getMessageCountUpTo(String apiKey, @PartitionKey String queue, long limit) {
        return doGetMessageCountUpTo(apiKey, queue, limit, true);
    }

    @Override
    public long getClaimCount(String apiKey, @PartitionKey String queue) {
        return doGetClaimCount(apiKey, queue);
    }

    @Override
    public List peek(String apiKey, @PartitionKey String queue, int limit) {
        return doPeek(apiKey, queue, limit, true);
    }

    @Override
    public List poll(String apiKey, @PartitionKey String queue, Duration claimTtl, int limit) {
        return doPoll(apiKey, queue, claimTtl, limit);
    }

    @Override
    public void renew(String apiKey, @PartitionKey String queue, Collection messageIds, Duration claimTtl) {
        doRenew(apiKey, queue, messageIds, claimTtl);
    }

    @Override
    public void acknowledge(String apiKey, @PartitionKey String queue, Collection messageIds) {
        doAcknowledge(apiKey, queue, messageIds);
    }

    @Override
    public String moveAsync(String apiKey, String from, String to) {
        // Any server can handle this request, no need for @PartitionKey
        return doMoveAsync(apiKey, from, to);
    }

    @Override
    public MoveQueueStatus getMoveStatus(String apiKey, String reference) {
        // Any server can handle this request, no need for @PartitionKey
        return doGetMoveStatus(apiKey, reference);
    }

    @Override
    public void unclaimAll(String apiKey, @PartitionKey String queue) {
        doUnclaimAll(apiKey, queue);
    }

    @Override
    public void purge(String apiKey, @PartitionKey String queue) {
        doPurge(apiKey, queue);
    }
}