flink-issues mailing list archives

Site index · List index
Message view « Date » · « Thread »
Top « Date » · « Thread »
From yew1eb <...@git.apache.org>
Subject [GitHub] flink pull request #5374: [FLINK-8101][flink-connectors] Elasticsearch 5.3+ ...
Date Wed, 31 Jan 2018 12:03:23 GMT
Github user yew1eb commented on a diff in the pull request:

    https://github.com/apache/flink/pull/5374#discussion_r165032079
  
    --- Diff: flink-connectors/flink-connector-elasticsearch5.3/src/main/java/org/apache/flink/streaming/connectors/elasticsearch53/BulkProcessorIndexer.java
---
    @@ -0,0 +1,57 @@
    +/*
    + * 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.flink.streaming.connectors.elasticsearch53;
    +
    +import org.apache.flink.streaming.connectors.elasticsearch.RequestIndexer;
    +
    +import org.elasticsearch.action.ActionRequest;
    +import org.elasticsearch.action.DocWriteRequest;
    +import org.elasticsearch.action.bulk.BulkProcessor;
    +
    +import java.util.concurrent.atomic.AtomicLong;
    +
    +/**
    + * Implementation of a {@link RequestIndexer}, using a {@link BulkProcessor}.
    + * {@link ActionRequest ActionRequests} will be converted to {@link DocWriteRequest}
    + * and will be buffered before sending a bulk request to the Elasticsearch cluster.
    + */
    +public class BulkProcessorIndexer implements RequestIndexer {
    +
    +	private final BulkProcessor bulkProcessor;
    +	private final boolean flushOnCheckpoint;
    +	private final AtomicLong numPendingRequestsRef;
    +
    +	public BulkProcessorIndexer(BulkProcessor bulkProcessor,
    +								boolean flushOnCheckpoint,
    +								AtomicLong numPendingRequests) {
    +		this.bulkProcessor = bulkProcessor;
    +		this.flushOnCheckpoint = flushOnCheckpoint;
    +		this.numPendingRequestsRef = numPendingRequests;
    +	}
    +
    +	@Override
    +	public void add(ActionRequest... actionRequests) {
    +		for (ActionRequest actionRequest : actionRequests) {
    +			if (flushOnCheckpoint) {
    +				numPendingRequestsRef.getAndIncrement();
    +			}
    +			this.bulkProcessor.add((DocWriteRequest) actionRequest);
    --- End diff --
    
    Are you sure that `ActionRequest` type can be cast to `DocWriteRequest` type?


---

Mime
View raw message