/*
 * Copyright DataStax, 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.datastax.oss.protocol.internal.response.error;

import static com.datastax.oss.protocol.internal.ProtocolConstants.Version.V5;

import com.datastax.oss.protocol.internal.Message;
import com.datastax.oss.protocol.internal.PrimitiveCodec;
import com.datastax.oss.protocol.internal.PrimitiveSizes;
import com.datastax.oss.protocol.internal.ProtocolConstants;
import com.datastax.oss.protocol.internal.response.Error;
import java.net.InetAddress;
import java.util.Collections;
import java.util.Map;

public class WriteFailure extends Error {
  
The consistency level of the query that triggered the exception.
/** The consistency level of the query that triggered the exception. */
public final int consistencyLevel;
The number of nodes having acknowledged the request.
/** The number of nodes having acknowledged the request. */
public final int received;
The number of replicas whose acknowledgment is required to achieve consistencyLevel.
/** * The number of replicas whose acknowledgment is required to achieve {@code consistencyLevel}. */
public final int blockFor;
The number of nodes that experienced a failure while executing the request.
/** The number of nodes that experienced a failure while executing the request. */
public final int numFailures;
A map of endpoint to failure reason codes. This maps the endpoints of the replica nodes that failed when executing the request to a code representing the reason for the failure.
/** * A map of endpoint to failure reason codes. This maps the endpoints of the replica nodes that * failed when executing the request to a code representing the reason for the failure. */
public final Map<InetAddress, Integer> reasonMap;
The type of the write that timed out.
/** The type of the write that timed out. */
public final String writeType; public WriteFailure( String message, int consistencyLevel, int received, int blockFor, int numFailures, Map<InetAddress, Integer> reasonMap, String writeType) { super(ProtocolConstants.ErrorCode.WRITE_FAILURE, message); this.consistencyLevel = consistencyLevel; this.received = received; this.blockFor = blockFor; this.numFailures = numFailures; this.reasonMap = reasonMap; this.writeType = writeType; } public static class SubCodec extends Error.SubCodec { public SubCodec(int protocolVersion) { super(ProtocolConstants.ErrorCode.WRITE_FAILURE, protocolVersion); } @Override public <B> void encode(B dest, Message message, PrimitiveCodec<B> encoder) { WriteFailure writeFailure = (WriteFailure) message; encoder.writeString(writeFailure.message, dest); encoder.writeUnsignedShort(writeFailure.consistencyLevel, dest); encoder.writeInt(writeFailure.received, dest); encoder.writeInt(writeFailure.blockFor, dest); if (protocolVersion >= V5) { ReadFailure.SubCodec.writeReasonMap(writeFailure.reasonMap, dest, encoder); } else { encoder.writeInt(writeFailure.numFailures, dest); } encoder.writeString(writeFailure.writeType, dest); } @Override public int encodedSize(Message message) { WriteFailure writeFailure = (WriteFailure) message; int size = PrimitiveSizes.sizeOfString(writeFailure.message) + PrimitiveSizes.SHORT // consistencyLevel + PrimitiveSizes.INT // received + PrimitiveSizes.INT // blockFor + PrimitiveSizes.sizeOfString(writeFailure.writeType); if (protocolVersion >= V5) { size += ReadFailure.SubCodec.sizeOfReasonMap(writeFailure.reasonMap); } else { size += PrimitiveSizes.INT; } return size; } @Override public <B> Message decode(B source, PrimitiveCodec<B> decoder) { String message = decoder.readString(source); int consistencyLevel = decoder.readUnsignedShort(source); int received = decoder.readInt(source); int blockFor = decoder.readInt(source); int numFailures; Map<InetAddress, Integer> reasonMap; if (protocolVersion >= V5) { reasonMap = ReadFailure.SubCodec.readReasonMap(source, decoder); numFailures = reasonMap.size(); } else { reasonMap = Collections.emptyMap(); numFailures = decoder.readInt(source); } String writeType = decoder.readString(source); return new WriteFailure( message, consistencyLevel, received, blockFor, numFailures, reasonMap, writeType); } } }