/* * Licensed to the Apache Software Foundation (ASF) under one * or more contributor license agreements. See the NOTICE file * distributed with this work for additional information * regarding copyright ownership. The ASF licenses this file * to you 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 org.apache.cassandra.service; import java.nio.ByteBuffer; import java.util.Set; import org.cliffc.high_scale_lib.NonBlockingHashSet; import org.slf4j.Logger; import org.slf4j.LoggerFactory; import org.apache.cassandra.db.DecoratedKey; import org.apache.cassandra.db.ReadResponse; import org.apache.cassandra.db.Row; import org.apache.cassandra.net.MessageIn; public abstract class AbstractRowResolver implements IResponseResolver<ReadResponse, Row> { protected static final Logger logger = LoggerFactory.getLogger(AbstractRowResolver.class); protected final String keyspaceName; protected final Set<MessageIn<ReadResponse>> replies = new NonBlockingHashSet<MessageIn<ReadResponse>>(); protected final DecoratedKey key; public AbstractRowResolver(ByteBuffer key, String keyspaceName) { this.key = StorageService.getPartitioner().decorateKey(key); this.keyspaceName = keyspaceName; } public boolean preprocess(MessageIn<ReadResponse> message) { MessageIn<ReadResponse> toReplace = null; for (MessageIn<ReadResponse> reply : replies) { if (reply.from.equals(message.from)) { if (!message.payload.isDigestQuery()) toReplace = reply; break; } } // replace old message if (toReplace != null) { replies.remove(toReplace); replies.add(message); return false; } replies.add(message); return true; } public Iterable<MessageIn<ReadResponse>> getMessages() { return replies; } }