private BulkRequestBuilder initRequest() { BulkRequestBuilder bulkRequest = client.prepareBulk(); bulkRequest.setConsistencyLevel( WriteConsistencyLevel.fromString( config.getWriteConsistencyLevel() ) ); bulkRequest.setRefresh( config.isForcedRefresh() ); return bulkRequest; }
@Override public void writeTo(StreamOutput out) throws IOException { super.writeTo(out); out.writeByte(consistencyLevel.id()); out.writeVInt(requests.size()); for (ActionRequest request : requests) { if (request instanceof IndexRequest) { out.writeByte((byte) 0); } else if (request instanceof DeleteRequest) { out.writeByte((byte) 1); } else if (request instanceof UpdateRequest) { out.writeByte((byte) 2); } request.writeTo(out); } out.writeBoolean(refresh); timeout.writeTo(out); } }
@Override public void readFrom(StreamInput in) throws IOException { super.readFrom(in); consistencyLevel = WriteConsistencyLevel.fromId(in.readByte()); int size = in.readVInt(); for (int i = 0; i < size; i++) { byte type = in.readByte(); if (type == 0) { IndexRequest request = new IndexRequest(); request.readFrom(in); requests.add(request); } else if (type == 1) { DeleteRequest request = new DeleteRequest(); request.readFrom(in); requests.add(request); } else if (type == 2) { UpdateRequest request = new UpdateRequest(); request.readFrom(in); requests.add(request); } } refresh = in.readBoolean(); timeout = TimeValue.readTimeValue(in); }
@Override public JsonObject toJson() { JsonObject json = super.toJson(); if (getConsistencyLevel() != null) { json.put(FIELD_CONSISTENCY_LEVEL, getConsistencyLevel().toString().toLowerCase()); } if (getTimeout() != null) json.put(FIELD_TIMEOUT, getTimeout()); return json; }
@Override public void writeTo(StreamOutput out) throws IOException { super.writeTo(out); searchRequest.writeTo(out); out.writeBoolean(abortOnVersionConflict); out.writeVInt(size); out.writeBoolean(refresh); timeout.writeTo(out); out.writeByte(consistency.id()); retryBackoffInitialTime.writeTo(out); out.writeVInt(maxRetries); if (out.getVersion().onOrAfter(Version.V_2_4_0)) { out.writeFloat(requestsPerSecond); } }
@Override public void readFrom(StreamInput in) throws IOException { super.readFrom(in); if (in.readBoolean()) { shardId = ShardId.readShardId(in); } else { shardId = null; } consistencyLevel = WriteConsistencyLevel.fromId(in.readByte()); timeout = TimeValue.readTimeValue(in); index = in.readString(); canHaveDuplicates = in.readBoolean(); // no need to serialize threaded* parameters, since they only matter locally if (in.getVersion().onOrAfter(Version.V_2_4_0)) { routedBasedOnClusterVersion = in.readVLong(); } }
protected AbstractWriteOptions(JsonObject json) { super(json); timeout = json.getString(FIELD_TIMEOUT); String s = json.getString(FIELD_CONSISTENCY_LEVEL); if (s != null) consistencyLevel = WriteConsistencyLevel.fromString(s); }
@Override public void writeTo(StreamOutput out) throws IOException { super.writeTo(out); searchRequest.writeTo(out); out.writeBoolean(abortOnVersionConflict); out.writeVInt(size); out.writeBoolean(refresh); timeout.writeTo(out); out.writeByte(consistency.id()); retryBackoffInitialTime.writeTo(out); out.writeVInt(maxRetries); if (out.getVersion().onOrAfter(Version.V_2_4_0)) { out.writeFloat(requestsPerSecond); } }
@Override public void readFrom(StreamInput in) throws IOException { super.readFrom(in); searchRequest = new SearchRequest(); searchRequest.readFrom(in); abortOnVersionConflict = in.readBoolean(); size = in.readVInt(); refresh = in.readBoolean(); timeout = TimeValue.readTimeValue(in); consistency = WriteConsistencyLevel.fromId(in.readByte()); retryBackoffInitialTime = TimeValue.readTimeValue(in); maxRetries = in.readVInt(); if (in.getVersion().onOrAfter(Version.V_2_4_0)) { requestsPerSecond = in.readFloat(); } }
public UpdateRequestBuilder<JsonInput, JsonOutput> consistencyLevel(String consistencyLevel) { request.consistencyLevel(WriteConsistencyLevel.fromString(consistencyLevel)); return this; }
@Override public void writeTo(StreamOutput out) throws IOException { super.writeTo(out); if (shardId != null) { out.writeBoolean(true); shardId.writeTo(out); } else { out.writeBoolean(false); } out.writeByte(consistencyLevel.id()); timeout.writeTo(out); out.writeString(index); out.writeBoolean(canHaveDuplicates); if (out.getVersion().onOrAfter(Version.V_2_4_0)) { out.writeVLong(routedBasedOnClusterVersion); } }
@Override public void readFrom(StreamInput in) throws IOException { super.readFrom(in); searchRequest = new SearchRequest(); searchRequest.readFrom(in); abortOnVersionConflict = in.readBoolean(); size = in.readVInt(); refresh = in.readBoolean(); timeout = TimeValue.readTimeValue(in); consistency = WriteConsistencyLevel.fromId(in.readByte()); retryBackoffInitialTime = TimeValue.readTimeValue(in); maxRetries = in.readVInt(); if (in.getVersion().onOrAfter(Version.V_2_4_0)) { requestsPerSecond = in.readFloat(); } }
public IndexRequestBuilder<JsonInput, JsonOutput> consistencyLevel(String consistencyLevel) { request.consistencyLevel(WriteConsistencyLevel.fromString(consistencyLevel)); return this; }
@Override public void writeTo(StreamOutput out) throws IOException { super.writeTo(out); out.writeByte(consistencyLevel.id()); out.writeString(type); out.writeString(id);
@Override public void readFrom(StreamInput in) throws IOException { super.readFrom(in); consistencyLevel = WriteConsistencyLevel.fromId(in.readByte()); type = in.readString(); id = in.readString();
public DeleteRequestBuilder<JsonInput, JsonOutput> consistencyLevel(String consistencyLevel) { request.consistencyLevel(WriteConsistencyLevel.fromString(consistencyLevel)); return this; }
public DeleteByQueryRequestBuilder<JsonInput, JsonOutput> consistencyLevel(String consistencyLevel) { request.consistencyLevel(WriteConsistencyLevel.fromString(consistencyLevel)); return this; }
public static void parseCommon(AbstractBulkByScrollRequest<?> internalRequest, RestRequest request) { internalRequest.setRefresh(request.paramAsBoolean("refresh", internalRequest.isRefresh())); internalRequest.setTimeout(request.paramAsTime("timeout", internalRequest.getTimeout())); String consistency = request.param("consistency"); if (consistency != null) { internalRequest.setConsistency(WriteConsistencyLevel.fromString(consistency)); } }
public static void parseCommon(AbstractBulkByScrollRequest<?> internalRequest, RestRequest request) { internalRequest.setRefresh(request.paramAsBoolean("refresh", internalRequest.isRefresh())); internalRequest.setTimeout(request.paramAsTime("timeout", internalRequest.getTimeout())); String consistency = request.param("consistency"); if (consistency != null) { internalRequest.setConsistency(WriteConsistencyLevel.fromString(consistency)); } }
protected TransportReplicationAction(Settings settings, String actionName, TransportService transportService, ClusterService clusterService, IndicesService indicesService, ThreadPool threadPool, ShardStateAction shardStateAction, MappingUpdatedAction mappingUpdatedAction, ActionFilters actionFilters, IndexNameExpressionResolver indexNameExpressionResolver, Class<Request> request, Class<ReplicaRequest> replicaRequest, String executor) { super(settings, actionName, threadPool, actionFilters, indexNameExpressionResolver, transportService.getTaskManager()); this.transportService = transportService; this.clusterService = clusterService; this.indicesService = indicesService; this.shardStateAction = shardStateAction; this.mappingUpdatedAction = mappingUpdatedAction; this.transportPrimaryAction = actionName + "[p]"; this.transportReplicaAction = actionName + "[r]"; this.executor = executor; this.checkWriteConsistency = checkWriteConsistency(); transportService.registerRequestHandler(actionName, request, ThreadPool.Names.SAME, new OperationTransportHandler()); transportService.registerRequestHandler(transportPrimaryAction, request, executor, new PrimaryOperationTransportHandler()); // we must never reject on because of thread pool capacity on replicas transportService.registerRequestHandler(transportReplicaAction, replicaRequest, executor, true, true, new ReplicaOperationTransportHandler()); this.transportOptions = transportOptions(); this.defaultWriteConsistencyLevel = WriteConsistencyLevel.fromString(settings.get("action.write_consistency", "quorum")); }