com.basho.riak.client.core.PBStreamingFutureOperation Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of riak-client Show documentation
HttpClient-based client for Riak
The newest version!
/*
 * Copyright 2016 Basho Technologies, Inc.
 *
 * 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.basho.riak.client.core;

import com.basho.riak.client.core.operations.PBFutureOperation;
import com.basho.riak.protobuf.RiakKvPB;
import com.google.protobuf.GeneratedMessage;

import java.util.concurrent.LinkedTransferQueue;
import java.util.concurrent.TransferQueue;

/**
 * @author Alex Moore 
 * @param  The type returned by the streaming and non-streaming operation versions
 * @param  The protocol type returned
 * @param  Query info type
 * @since 2.1.0
 */
public abstract class PBStreamingFutureOperation
        extends PBFutureOperation
        implements StreamingRiakFuture
{
    private final TransferQueue responseQueue;
    private boolean streamResults;

    protected PBStreamingFutureOperation(final byte reqMessageCode,
                                         final byte respMessageCode,
                                         final GeneratedMessage.Builder reqBuilder,
                                         com.google.protobuf.Parser respParser,
                                         boolean streamResults)
    {
        super(reqMessageCode, respMessageCode, reqBuilder, respParser);
        this.streamResults = streamResults;
        this.responseQueue = streamResults ? new LinkedTransferQueue<>() : null;
    }

    @Override
    protected void processMessage(ResponseType decodedMessage)
    {
        if (!streamResults)
        {
            super.processMessage(decodedMessage);
            return;
        }

        final ReturnType r = processStreamingChunk(decodedMessage);
        assert this.responseQueue != null;
        final boolean chunkAdded = responseQueue.offer(r);
        assert chunkAdded;
    }

    abstract protected ReturnType processStreamingChunk(ResponseType rawResponseChunk);

    public final TransferQueue getResultsQueue()
    {
        assert this.responseQueue != null;
        return this.responseQueue;
    }
}