/**
* 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.db;
import java.io.ByteArrayInputStream;
import java.io.DataInputStream;
import java.io.IOException;
import java.net.InetAddress;
import java.net.UnknownHostException;
import java.nio.ByteBuffer;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.apache.cassandra.net.IVerbHandler;
import org.apache.cassandra.net.Message;
import org.apache.cassandra.net.MessagingService;
import org.apache.cassandra.utils.ByteBufferUtil;
import org.apache.cassandra.utils.FBUtilities;
import org.apache.cassandra.config.DatabaseDescriptor;
public class RowMutationVerbHandler implements IVerbHandler
{
private static Logger logger_ = LoggerFactory.getLogger(RowMutationVerbHandler.class);
public void doVerb(Message message, String id)
{
byte[] bytes = message.getMessageBody();
ByteArrayInputStream buffer = new ByteArrayInputStream(bytes);
try
{
RowMutation rm = RowMutation.serializer().deserialize(new DataInputStream(buffer));
if (logger_.isDebugEnabled())
logger_.debug("Applying " + rm);
/* Check if there were any hints in this message */
byte[] hintedBytes = message.getHeader(RowMutation.HINT);
if (hintedBytes != null)
{
assert hintedBytes.length > 0;
DataInputStream dis = new DataInputStream(new ByteArrayInputStream(hintedBytes));
while (dis.available() > 0)
{
ByteBuffer addressBytes = ByteBufferUtil.readWithShortLength(dis);
if (logger_.isDebugEnabled())
logger_.debug("Adding hint for " + InetAddress.getByName(ByteBufferUtil.string(addressBytes)));
RowMutation hintedMutation = new RowMutation(Table.SYSTEM_TABLE, addressBytes);
hintedMutation.addHints(rm);
hintedMutation.apply();
}
}
// Check if there were any forwarding headers in this message
byte[] forwardBytes = message.getHeader(RowMutation.FORWARD_HEADER);
if (forwardBytes != null)
forwardToLocalNodes(message, forwardBytes);
if(DatabaseDescriptor.isBigtable(rm.getTable()))
{
Table.open(rm.getTable()).apply(rm, bytes, true);
}
else
{
Table.open(rm.getTable()).apply(rm, bytes, false);
}
WriteResponse response = new WriteResponse(rm.getTable(), rm.key(), true);
Message responseMessage = WriteResponse.makeWriteResponseMessage(message, response);
if (logger_.isDebugEnabled())
logger_.debug(rm + " applied. Sending response to " + id + "@" + message.getFrom());
MessagingService.instance().sendReply(responseMessage, id, message.getFrom());
}
catch (IOException e)
{
logger_.error("Error in row mutation", e);
}
}
private void forwardToLocalNodes(Message message, byte[] forwardBytes) throws UnknownHostException
{
// remove fwds from message to avoid infinite loop
message.removeHeader(RowMutation.FORWARD_HEADER);
int bytesPerInetAddress = FBUtilities.getLocalAddress().getAddress().length;
assert forwardBytes.length >= bytesPerInetAddress;
assert forwardBytes.length % bytesPerInetAddress == 0;
int offset = 0;
byte[] addressBytes = new byte[bytesPerInetAddress];
// Send a message to each of the addresses on our Forward List
while (offset < forwardBytes.length)
{
System.arraycopy(forwardBytes, offset, addressBytes, 0, bytesPerInetAddress);
InetAddress address = InetAddress.getByAddress(addressBytes);
if (logger_.isDebugEnabled())
logger_.debug("Forwarding message to " + address);
// Send the original message to the address specified by the FORWARD_HINT
// Let the response go back to the coordinator
MessagingService.instance().sendOneWay(message, address);
offset += bytesPerInetAddress;
}
}
}