com.lambdaworks.redis.protocol.QueuedCommands Maven / Gradle / Ivy

Go to download

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

Advanced and thread-safe Java Redis client for synchronous, asynchronous, and reactive usage. Supports Cluster, Sentinel, Pipelining, Auto-Reconnect, Codecs and much more.

The newest version!

/*
 * Copyright 2011-2016 the original author or authors.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.lambdaworks.redis.protocol;

import java.util.ArrayList;
import java.util.Collection;
import java.util.List;
import java.util.Set;

import io.netty.util.internal.ConcurrentSet;

/**
 * Registry for all queued commands. This class allows to register/unregister {@link HasQueuedCommands queue holders}. It
 * provides queue maintenance commands.
 * 
 * @author Mark Paluch
 */
class QueuedCommands {

    private Set queues = new ConcurrentSet<>();

    /**
     * Register queue holder.
     * 
     * @param queueHolder the queue holder, must not be {@literal null}.
     */
    void register(HasQueuedCommands queueHolder) {
        queues.add(queueHolder);
    }

    /**
     * Unregister queue holder.
     *
     * @param queueHolder the queue holder, must not be {@literal null}.
     */
    void unregister(HasQueuedCommands queueHolder) {
        queues.remove(queueHolder);
    }

    /**
     * Remove a command from all registered queues.
     * 
     * @param command the command.
     */
    public void remove(RedisCommand command) {

        for (HasQueuedCommands queue : queues) {
            queue.getQueue().remove(command);
        }
    }

    /**
     * Remove multiple commands from all registered queues.
     *
     * @param commands the commands.
     */
    void removeAll(Collection> commands) {

        for (HasQueuedCommands queue : queues) {
            queue.getQueue().removeAll(commands);
        }
    }

    /**
     * Drain commands from all queues.
     * 
     * @return the commands.
     */
    List> drainCommands() {

        int size = 0;

        for (HasQueuedCommands queue : queues) {
            size += queue.getQueue().size();
        }

        List> target = new ArrayList<>(size);

        RedisCommand cmd;
        for (HasQueuedCommands queue : queues) {
            while ((cmd = queue.getQueue().poll()) != null) {
                target.add(cmd);
            }
        }

        return target;
    }

    /**
     * Check whether the registered queues exceed a global sized of the {@code requestQueueSize}.
     * 
     * @param requestQueueSize the queue size limit.
     * @return {@literal true} if there are more queued commands than {@code requestQueueSize}
     */
    boolean exceedsLimit(int requestQueueSize) {

        if (requestQueueSize == Integer.MAX_VALUE) {
            return false;
        }

        int current = 0;

        for (HasQueuedCommands queue : queues) {
            current += queue.getQueue().size();
            if (current >= requestQueueSize) {
                return true;
            }
        }

        return false;
    }
}