|
|
@@ -0,0 +1,544 @@
|
|
|
+/**
|
|
|
+ * DBSyncer Copyright 2020-2024 All Rights Reserved.
|
|
|
+ */
|
|
|
+package org.dbsyncer.connector.elasticsearch.api.bulk;
|
|
|
+
|
|
|
+import org.dbsyncer.common.util.StringUtil;
|
|
|
+import org.elasticsearch.ElasticsearchException;
|
|
|
+import org.elasticsearch.ExceptionsHelper;
|
|
|
+import org.elasticsearch.Version;
|
|
|
+import org.elasticsearch.action.DocWriteRequest.OpType;
|
|
|
+import org.elasticsearch.action.DocWriteResponse;
|
|
|
+import org.elasticsearch.action.bulk.BulkItemRequest;
|
|
|
+import org.elasticsearch.action.bulk.BulkResponse;
|
|
|
+import org.elasticsearch.action.delete.DeleteResponse;
|
|
|
+import org.elasticsearch.action.index.IndexResponse;
|
|
|
+import org.elasticsearch.action.update.UpdateResponse;
|
|
|
+import org.elasticsearch.common.CheckedConsumer;
|
|
|
+import org.elasticsearch.common.ParseField;
|
|
|
+import org.elasticsearch.common.Strings;
|
|
|
+import org.elasticsearch.common.io.stream.StreamInput;
|
|
|
+import org.elasticsearch.common.io.stream.StreamOutput;
|
|
|
+import org.elasticsearch.common.io.stream.Writeable;
|
|
|
+import org.elasticsearch.common.xcontent.ConstructingObjectParser;
|
|
|
+import org.elasticsearch.common.xcontent.StatusToXContentObject;
|
|
|
+import org.elasticsearch.common.xcontent.ToXContentFragment;
|
|
|
+import org.elasticsearch.common.xcontent.XContentBuilder;
|
|
|
+import org.elasticsearch.common.xcontent.XContentParser;
|
|
|
+import org.elasticsearch.index.seqno.SequenceNumbers;
|
|
|
+import org.elasticsearch.index.shard.ShardId;
|
|
|
+import org.elasticsearch.rest.RestStatus;
|
|
|
+
|
|
|
+import java.io.IOException;
|
|
|
+
|
|
|
+import static org.elasticsearch.common.xcontent.ConstructingObjectParser.constructorArg;
|
|
|
+import static org.elasticsearch.common.xcontent.ConstructingObjectParser.optionalConstructorArg;
|
|
|
+import static org.elasticsearch.common.xcontent.XContentParserUtils.ensureExpectedToken;
|
|
|
+import static org.elasticsearch.common.xcontent.XContentParserUtils.throwUnknownField;
|
|
|
+
|
|
|
+/**
|
|
|
+ * Represents a single item response for an action executed as part of the bulk API. Holds the index/type/id
|
|
|
+ * of the relevant action, and if it has failed or not (with the failure message in case it failed).
|
|
|
+ */
|
|
|
+public class BulkItemResponse implements Writeable, StatusToXContentObject {
|
|
|
+
|
|
|
+ private static final String _INDEX = "_index";
|
|
|
+ private static final String _TYPE = "_type";
|
|
|
+ private static final String _ID = "_id";
|
|
|
+ private static final String STATUS = "status";
|
|
|
+ private static final String ERROR = "error";
|
|
|
+
|
|
|
+ @Override
|
|
|
+ public RestStatus status() {
|
|
|
+ return failure == null ? response.status() : failure.getStatus();
|
|
|
+ }
|
|
|
+
|
|
|
+ @Override
|
|
|
+ public XContentBuilder toXContent(XContentBuilder builder, Params params) throws IOException {
|
|
|
+ builder.startObject();
|
|
|
+ builder.startObject(opType.getLowercase());
|
|
|
+ if (failure == null) {
|
|
|
+ response.innerToXContent(builder, params);
|
|
|
+ builder.field(STATUS, response.status().getStatus());
|
|
|
+ } else {
|
|
|
+ builder.field(_INDEX, failure.getIndex());
|
|
|
+ builder.field(_TYPE, failure.getType());
|
|
|
+ builder.field(_ID, failure.getId());
|
|
|
+ builder.field(STATUS, failure.getStatus().getStatus());
|
|
|
+ builder.startObject(ERROR);
|
|
|
+ ElasticsearchException.generateThrowableXContent(builder, params, failure.getCause());
|
|
|
+ builder.endObject();
|
|
|
+ }
|
|
|
+ builder.endObject();
|
|
|
+ builder.endObject();
|
|
|
+ return builder;
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * Reads a {@link BulkItemResponse} from a {@link XContentParser}.
|
|
|
+ *
|
|
|
+ * @param parser the {@link XContentParser}
|
|
|
+ * @param id the id to assign to the parsed {@link BulkItemResponse}. It is usually the index of
|
|
|
+ * the item in the {@link BulkResponse#getItems} array.
|
|
|
+ */
|
|
|
+ public static BulkItemResponse fromXContent(XContentParser parser, int id) throws IOException {
|
|
|
+ ensureExpectedToken(XContentParser.Token.START_OBJECT, parser.currentToken(), parser);
|
|
|
+
|
|
|
+ XContentParser.Token token = parser.nextToken();
|
|
|
+ ensureExpectedToken(XContentParser.Token.FIELD_NAME, token, parser);
|
|
|
+
|
|
|
+ String currentFieldName = parser.currentName();
|
|
|
+ token = parser.nextToken();
|
|
|
+
|
|
|
+ final OpType opType = OpType.fromString(currentFieldName);
|
|
|
+ ensureExpectedToken(XContentParser.Token.START_OBJECT, token, parser);
|
|
|
+
|
|
|
+ DocWriteResponse.Builder builder = null;
|
|
|
+ CheckedConsumer<XContentParser, IOException> itemParser = null;
|
|
|
+
|
|
|
+ if (opType == OpType.INDEX || opType == OpType.CREATE) {
|
|
|
+ final IndexResponse.Builder indexResponseBuilder = new IndexResponse.Builder();
|
|
|
+ builder = indexResponseBuilder;
|
|
|
+ itemParser = (indexParser) -> IndexResponse.parseXContentFields(indexParser, indexResponseBuilder);
|
|
|
+
|
|
|
+ } else if (opType == OpType.UPDATE) {
|
|
|
+ final UpdateResponse.Builder updateResponseBuilder = new UpdateResponse.Builder();
|
|
|
+ builder = updateResponseBuilder;
|
|
|
+ itemParser = (updateParser) -> UpdateResponse.parseXContentFields(updateParser, updateResponseBuilder);
|
|
|
+
|
|
|
+ } else if (opType == OpType.DELETE) {
|
|
|
+ final DeleteResponse.Builder deleteResponseBuilder = new DeleteResponse.Builder();
|
|
|
+ builder = deleteResponseBuilder;
|
|
|
+ itemParser = (deleteParser) -> DeleteResponse.parseXContentFields(deleteParser, deleteResponseBuilder);
|
|
|
+ } else {
|
|
|
+ throwUnknownField(currentFieldName, parser.getTokenLocation());
|
|
|
+ }
|
|
|
+
|
|
|
+ RestStatus status = null;
|
|
|
+ ElasticsearchException exception = null;
|
|
|
+ while ((token = parser.nextToken()) != XContentParser.Token.END_OBJECT) {
|
|
|
+ if (token == XContentParser.Token.FIELD_NAME) {
|
|
|
+ currentFieldName = parser.currentName();
|
|
|
+ }
|
|
|
+
|
|
|
+ if (ERROR.equals(currentFieldName)) {
|
|
|
+ if (token == XContentParser.Token.START_OBJECT) {
|
|
|
+ exception = ElasticsearchException.fromXContent(parser);
|
|
|
+ }
|
|
|
+ } else if (STATUS.equals(currentFieldName)) {
|
|
|
+ if (token == XContentParser.Token.VALUE_NUMBER) {
|
|
|
+ status = RestStatus.fromCode(parser.intValue());
|
|
|
+ }
|
|
|
+ } else {
|
|
|
+ itemParser.accept(parser);
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ ensureExpectedToken(XContentParser.Token.END_OBJECT, token, parser);
|
|
|
+ token = parser.nextToken();
|
|
|
+ ensureExpectedToken(XContentParser.Token.END_OBJECT, token, parser);
|
|
|
+
|
|
|
+ BulkItemResponse bulkItemResponse;
|
|
|
+ if (exception != null) {
|
|
|
+ BulkItemResponse.Failure failure = new BulkItemResponse.Failure(builder.getShardId().getIndexName(), builder.getType(), builder.getId(), exception, status);
|
|
|
+ bulkItemResponse = new BulkItemResponse(id, opType, failure);
|
|
|
+ } else {
|
|
|
+ builder.setType(builder.getType() == null ? StringUtil.EMPTY : builder.getType());
|
|
|
+ bulkItemResponse = new BulkItemResponse(id, opType, builder.build());
|
|
|
+ }
|
|
|
+ return bulkItemResponse;
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * Represents a failure.
|
|
|
+ */
|
|
|
+ public static class Failure implements Writeable, ToXContentFragment {
|
|
|
+ public static final String INDEX_FIELD = "index";
|
|
|
+ public static final String TYPE_FIELD = "type";
|
|
|
+ public static final String ID_FIELD = "id";
|
|
|
+ public static final String CAUSE_FIELD = "cause";
|
|
|
+ public static final String STATUS_FIELD = "status";
|
|
|
+
|
|
|
+ private final String index;
|
|
|
+ private final String type;
|
|
|
+ private final String id;
|
|
|
+ private final Exception cause;
|
|
|
+ private final RestStatus status;
|
|
|
+ private final long seqNo;
|
|
|
+ private final long term;
|
|
|
+ private final boolean aborted;
|
|
|
+
|
|
|
+ public static final ConstructingObjectParser<BulkItemResponse.Failure, Void> PARSER =
|
|
|
+ new ConstructingObjectParser<>(
|
|
|
+ "bulk_failures",
|
|
|
+ true,
|
|
|
+ a ->
|
|
|
+ new BulkItemResponse.Failure(
|
|
|
+ (String)a[0], (String)a[1], (String)a[2], (Exception)a[3], RestStatus.fromCode((int)a[4])
|
|
|
+ )
|
|
|
+ );
|
|
|
+ static {
|
|
|
+ PARSER.declareString(constructorArg(), new ParseField(INDEX_FIELD));
|
|
|
+ PARSER.declareString(constructorArg(), new ParseField(TYPE_FIELD));
|
|
|
+ PARSER.declareString(optionalConstructorArg(), new ParseField(ID_FIELD));
|
|
|
+ PARSER.declareObject(constructorArg(), (p, c) -> ElasticsearchException.fromXContent(p), new ParseField(CAUSE_FIELD));
|
|
|
+ PARSER.declareInt(constructorArg(), new ParseField(STATUS_FIELD));
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * For write failures before operation was assigned a sequence number.
|
|
|
+ *
|
|
|
+ * use @{link {@link #Failure(String, String, String, Exception, long, long)}}
|
|
|
+ * to record operation sequence no with failure
|
|
|
+ */
|
|
|
+ public Failure(String index, String type, String id, Exception cause) {
|
|
|
+ this(index, type, id, cause, ExceptionsHelper.status(cause), SequenceNumbers.UNASSIGNED_SEQ_NO,
|
|
|
+ SequenceNumbers.UNASSIGNED_PRIMARY_TERM, false);
|
|
|
+ }
|
|
|
+
|
|
|
+ public Failure(String index, String type, String id, Exception cause, boolean aborted) {
|
|
|
+ this(index, type, id, cause, ExceptionsHelper.status(cause), SequenceNumbers.UNASSIGNED_SEQ_NO,
|
|
|
+ SequenceNumbers.UNASSIGNED_PRIMARY_TERM, aborted);
|
|
|
+ }
|
|
|
+
|
|
|
+ public Failure(String index, String type, String id, Exception cause, RestStatus status) {
|
|
|
+ this(index, type, id, cause, status, SequenceNumbers.UNASSIGNED_SEQ_NO, SequenceNumbers.UNASSIGNED_PRIMARY_TERM, false);
|
|
|
+ }
|
|
|
+
|
|
|
+ /** For write failures after operation was assigned a sequence number. */
|
|
|
+ public Failure(String index, String type, String id, Exception cause, long seqNo, long term) {
|
|
|
+ this(index, type, id, cause, ExceptionsHelper.status(cause), seqNo, term, false);
|
|
|
+ }
|
|
|
+
|
|
|
+ private Failure(String index, String type, String id, Exception cause, RestStatus status, long seqNo, long term, boolean aborted) {
|
|
|
+ this.index = index;
|
|
|
+ this.type = type;
|
|
|
+ this.id = id;
|
|
|
+ this.cause = cause;
|
|
|
+ this.status = status;
|
|
|
+ this.seqNo = seqNo;
|
|
|
+ this.term = term;
|
|
|
+ this.aborted = aborted;
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * Read from a stream.
|
|
|
+ */
|
|
|
+ public Failure(StreamInput in) throws IOException {
|
|
|
+ index = in.readString();
|
|
|
+ type = in.readString();
|
|
|
+ id = in.readOptionalString();
|
|
|
+ cause = in.readException();
|
|
|
+ status = ExceptionsHelper.status(cause);
|
|
|
+ seqNo = in.readZLong();
|
|
|
+ if (in.getVersion().onOrAfter(Version.V_7_6_0)) {
|
|
|
+ term = in.readVLong();
|
|
|
+ } else {
|
|
|
+ term = SequenceNumbers.UNASSIGNED_PRIMARY_TERM;
|
|
|
+ }
|
|
|
+ aborted = in.readBoolean();
|
|
|
+ }
|
|
|
+
|
|
|
+ @Override
|
|
|
+ public void writeTo(StreamOutput out) throws IOException {
|
|
|
+ out.writeString(index);
|
|
|
+ out.writeString(type);
|
|
|
+ out.writeOptionalString(id);
|
|
|
+ out.writeException(cause);
|
|
|
+ out.writeZLong(seqNo);
|
|
|
+ if (out.getVersion().onOrAfter(Version.V_7_6_0)) {
|
|
|
+ out.writeVLong(term);
|
|
|
+ }
|
|
|
+ out.writeBoolean(aborted);
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * The index name of the action.
|
|
|
+ */
|
|
|
+ public String getIndex() {
|
|
|
+ return this.index;
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * The type of the action.
|
|
|
+ */
|
|
|
+ public String getType() {
|
|
|
+ return type;
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * The id of the action.
|
|
|
+ */
|
|
|
+ public String getId() {
|
|
|
+ return id;
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * The failure message.
|
|
|
+ */
|
|
|
+ public String getMessage() {
|
|
|
+ return this.cause.toString();
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * The rest status.
|
|
|
+ */
|
|
|
+ public RestStatus getStatus() {
|
|
|
+ return this.status;
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * The actual cause of the failure.
|
|
|
+ */
|
|
|
+ public Exception getCause() {
|
|
|
+ return cause;
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * The operation sequence number generated by primary
|
|
|
+ * NOTE: {@link SequenceNumbers#UNASSIGNED_SEQ_NO}
|
|
|
+ * indicates sequence number was not generated by primary
|
|
|
+ */
|
|
|
+ public long getSeqNo() {
|
|
|
+ return seqNo;
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * The operation primary term of the primary
|
|
|
+ * NOTE: {@link SequenceNumbers#UNASSIGNED_PRIMARY_TERM}
|
|
|
+ * indicates primary term was not assigned by primary
|
|
|
+ */
|
|
|
+ public long getTerm() {
|
|
|
+ return term;
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * Whether this failure is the result of an <em>abort</em>.
|
|
|
+ * If {@code true}, the request to which this failure relates should never be retried, regardless of the {@link #getCause() cause}.
|
|
|
+ * @see BulkItemRequest#abort(String, Exception)
|
|
|
+ */
|
|
|
+ public boolean isAborted() {
|
|
|
+ return aborted;
|
|
|
+ }
|
|
|
+
|
|
|
+ @Override
|
|
|
+ public XContentBuilder toXContent(XContentBuilder builder, Params params) throws IOException {
|
|
|
+ builder.field(INDEX_FIELD, index);
|
|
|
+ builder.field(TYPE_FIELD, type);
|
|
|
+ if (id != null) {
|
|
|
+ builder.field(ID_FIELD, id);
|
|
|
+ }
|
|
|
+ builder.startObject(CAUSE_FIELD);
|
|
|
+ ElasticsearchException.generateThrowableXContent(builder, params, cause);
|
|
|
+ builder.endObject();
|
|
|
+ builder.field(STATUS_FIELD, status.getStatus());
|
|
|
+ return builder;
|
|
|
+ }
|
|
|
+
|
|
|
+ public static BulkItemResponse.Failure fromXContent(XContentParser parser) {
|
|
|
+ return PARSER.apply(parser, null);
|
|
|
+ }
|
|
|
+
|
|
|
+ @Override
|
|
|
+ public String toString() {
|
|
|
+ return Strings.toString(this);
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ private int id;
|
|
|
+
|
|
|
+ private OpType opType;
|
|
|
+
|
|
|
+ private DocWriteResponse response;
|
|
|
+
|
|
|
+ private BulkItemResponse.Failure failure;
|
|
|
+
|
|
|
+ BulkItemResponse() {}
|
|
|
+
|
|
|
+ BulkItemResponse(ShardId shardId, StreamInput in) throws IOException {
|
|
|
+ id = in.readVInt();
|
|
|
+ opType = OpType.fromId(in.readByte());
|
|
|
+
|
|
|
+ byte type = in.readByte();
|
|
|
+ if (type == 0) {
|
|
|
+ response = new IndexResponse(shardId, in);
|
|
|
+ } else if (type == 1) {
|
|
|
+ response = new DeleteResponse(shardId, in);
|
|
|
+ } else if (type == 3) { // make 3 instead of 2, because 2 is already in use for 'no responses'
|
|
|
+ response = new UpdateResponse(shardId, in);
|
|
|
+ } else if (type != 2) {
|
|
|
+ throw new IllegalArgumentException("Unexpected type [" + type + "]");
|
|
|
+ }
|
|
|
+
|
|
|
+ if (in.readBoolean()) {
|
|
|
+ failure = new BulkItemResponse.Failure(in);
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ BulkItemResponse(StreamInput in) throws IOException {
|
|
|
+ id = in.readVInt();
|
|
|
+ opType = OpType.fromId(in.readByte());
|
|
|
+
|
|
|
+ byte type = in.readByte();
|
|
|
+ if (type == 0) {
|
|
|
+ response = new IndexResponse(in);
|
|
|
+ } else if (type == 1) {
|
|
|
+ response = new DeleteResponse(in);
|
|
|
+ } else if (type == 3) { // make 3 instead of 2, because 2 is already in use for 'no responses'
|
|
|
+ response = new UpdateResponse(in);
|
|
|
+ } else if (type != 2) {
|
|
|
+ throw new IllegalArgumentException("Unexpected type [" + type + "]");
|
|
|
+ }
|
|
|
+
|
|
|
+ if (in.readBoolean()) {
|
|
|
+ failure = new BulkItemResponse.Failure(in);
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ public BulkItemResponse(int id, OpType opType, DocWriteResponse response) {
|
|
|
+ this.id = id;
|
|
|
+ this.response = response;
|
|
|
+ this.opType = opType;
|
|
|
+ }
|
|
|
+
|
|
|
+ public BulkItemResponse(int id, OpType opType, BulkItemResponse.Failure failure) {
|
|
|
+ this.id = id;
|
|
|
+ this.opType = opType;
|
|
|
+ this.failure = failure;
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * The numeric order of the item matching the same request order in the bulk request.
|
|
|
+ */
|
|
|
+ public int getItemId() {
|
|
|
+ return id;
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * The operation type ("index", "create" or "delete").
|
|
|
+ */
|
|
|
+ public OpType getOpType() {
|
|
|
+ return this.opType;
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * The index name of the action.
|
|
|
+ */
|
|
|
+ public String getIndex() {
|
|
|
+ if (failure != null) {
|
|
|
+ return failure.getIndex();
|
|
|
+ }
|
|
|
+ return response.getIndex();
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * The type of the action.
|
|
|
+ */
|
|
|
+ public String getType() {
|
|
|
+ if (failure != null) {
|
|
|
+ return failure.getType();
|
|
|
+ }
|
|
|
+ return response.getType();
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * The id of the action.
|
|
|
+ */
|
|
|
+ public String getId() {
|
|
|
+ if (failure != null) {
|
|
|
+ return failure.getId();
|
|
|
+ }
|
|
|
+ return response.getId();
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * The version of the action.
|
|
|
+ */
|
|
|
+ public long getVersion() {
|
|
|
+ if (failure != null) {
|
|
|
+ return -1;
|
|
|
+ }
|
|
|
+ return response.getVersion();
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * The actual response ({@link IndexResponse} or {@link DeleteResponse}). {@code null} in
|
|
|
+ * case of failure.
|
|
|
+ */
|
|
|
+ public <T extends DocWriteResponse> T getResponse() {
|
|
|
+ return (T) response;
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * Is this a failed execution of an operation.
|
|
|
+ */
|
|
|
+ public boolean isFailed() {
|
|
|
+ return failure != null;
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * The failure message, {@code null} if it did not fail.
|
|
|
+ */
|
|
|
+ public String getFailureMessage() {
|
|
|
+ if (failure != null) {
|
|
|
+ return failure.getMessage();
|
|
|
+ }
|
|
|
+ return null;
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * The actual failure object if there was a failure.
|
|
|
+ */
|
|
|
+ public BulkItemResponse.Failure getFailure() {
|
|
|
+ return this.failure;
|
|
|
+ }
|
|
|
+
|
|
|
+ @Override
|
|
|
+ public void writeTo(StreamOutput out) throws IOException {
|
|
|
+ out.writeVInt(id);
|
|
|
+ out.writeByte(opType.getId());
|
|
|
+
|
|
|
+ if (response == null) {
|
|
|
+ out.writeByte((byte) 2);
|
|
|
+ } else {
|
|
|
+ writeResponseType(out);
|
|
|
+ response.writeTo(out);
|
|
|
+ }
|
|
|
+ if (failure == null) {
|
|
|
+ out.writeBoolean(false);
|
|
|
+ } else {
|
|
|
+ out.writeBoolean(true);
|
|
|
+ failure.writeTo(out);
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ public void writeThin(StreamOutput out) throws IOException {
|
|
|
+ out.writeVInt(id);
|
|
|
+ out.writeByte(opType.getId());
|
|
|
+
|
|
|
+ if (response == null) {
|
|
|
+ out.writeByte((byte) 2);
|
|
|
+ } else {
|
|
|
+ writeResponseType(out);
|
|
|
+ response.writeThin(out);
|
|
|
+ }
|
|
|
+ if (failure == null) {
|
|
|
+ out.writeBoolean(false);
|
|
|
+ } else {
|
|
|
+ out.writeBoolean(true);
|
|
|
+ failure.writeTo(out);
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ private void writeResponseType(StreamOutput out) throws IOException {
|
|
|
+ if (response instanceof IndexResponse) {
|
|
|
+ out.writeByte((byte) 0);
|
|
|
+ } else if (response instanceof DeleteResponse) {
|
|
|
+ out.writeByte((byte) 1);
|
|
|
+ } else if (response instanceof UpdateResponse) {
|
|
|
+ out.writeByte((byte) 3); // make 3 instead of 2, because 2 is already in use for 'no responses'
|
|
|
+ } else {
|
|
|
+ throw new IllegalStateException("Unexpected response type found [" + response.getClass() + "]");
|
|
|
+ }
|
|
|
+ }
|
|
|
+}
|
AE86