Java Code Examples for org.elasticsearch.action.index.IndexRequestBuilder#setRouting()

private void populateIndexRequestBuilder(IndexRequestBuilder builder, IndexOptions options) {
    if (options != null) {
        if (options.getId() != null) builder.setId(options.getId());
        if (options.getRouting() != null) builder.setRouting(options.getRouting());
        if (options.getParent() != null) builder.setParent(options.getParent());
        if (options.getOpType() != null)
            builder.setOpType(DocWriteRequest.OpType.valueOf(options.getOpType().name()));
        if (options.getWaitForActiveShard() != null)
            builder.setWaitForActiveShards(options.getWaitForActiveShard());
        if (options.getRefreshPolicy() != null)
            builder.setRefreshPolicy(WriteRequest.RefreshPolicy.valueOf(options.getRefreshPolicy().name()));
        if (options.getVersion() != null) builder.setVersion(options.getVersion());
        if (options.getVersionType() != null) builder.setVersionType(options.getVersionType());
        if (options.getTimeout() != null) builder.setTimeout(options.getTimeout());
    }
}
 

public Optional<BulkResult> indexData(ElasticDataPointer targetDataPointer, SearchHit[] hits) {
  BulkRequestBuilder bulkRequest = createBulkRequestBuilder();

  for (SearchHit hit : hits) {
    Map<String, Object> source = hit.getSource();

    IndexRequestBuilder requestBuilder = prepareIndex(targetDataPointer.getIndexName(), targetDataPointer
        .getTypeName(), hit.getId(), source, hit.getIndex());
    if (hit.getFields().get("_ttl") != null) {
      requestBuilder.setTTL(hit.getFields().get("_ttl").value());
    }
    if (hit.getFields().get("_routing") != null) {
      requestBuilder.setRouting(hit.getFields().get("_routing").value());
    }
    requestBuilder.setSource(source);
    bulkRequest.add(requestBuilder);
  }
  return executeBulk(hits.length, bulkRequest);
}
 

/**
 * 非同期でドキュメントを登録する.
 * @param index インデックス名
 * @param type タイプ名
 * @param id ドキュメントのid
 * @param routingId routingId
 * @param data データ
 * @param opType 操作タイプ
 * @param version version番号
 * @return 非同期応答
 */
public ActionFuture<IndexResponse> asyncIndex(String index,
        String type,
        String id,
        String routingId,
        Map<String, Object> data,
        OpType opType,
        long version) {
    IndexRequestBuilder req = esTransportClient.prepareIndex(index, type, id).setSource(data).setOpType(opType)
            .setConsistencyLevel(WriteConsistencyLevel.DEFAULT).setRefresh(true);
    if (routingFlag) {
        req = req.setRouting(routingId);
    }
    if (version > -1) {
        req.setVersion(version);
    }

    ActionFuture<IndexResponse> ret = req.execute();
    EsRequestLogInfo logInfo = new EsRequestLogInfo(index, type, id, routingId, data, opType.toString(),
            version);
    this.fireEvent(Event.afterCreate, logInfo);

    return ret;
}
 

/**
 * ルーティングIDに関係なくバルクでドキュメントを登録.
 * @param index インデックス名
 * @param bulkMap バルクドキュメント
 * @return ES応答
 */
public DcBulkResponse asyncBulkCreate(
        String index, Map<String, List<EsBulkRequest>> bulkMap) {
    BulkRequestBuilder bulkRequest = esTransportClient.prepareBulk();
    // ルーティングIDごとにバルク登録を行うと効率が悪いため、引数で渡されたEsBulkRequestは全て一括登録する。
    // また、バルク登録後にactionGet()すると同期実行となるため、ここでは実行しない。
    // このため、execute()のレスポンスを返却し、呼び出し側でactionGet()してからレスポンスチェック、リフレッシュすること。
    for (Entry<String, List<EsBulkRequest>> ents : bulkMap.entrySet()) {
        for (EsBulkRequest data : ents.getValue()) {
            IndexRequestBuilder req = esTransportClient.
                    prepareIndex(index, data.getType(), data.getId()).setSource(data.getSource());
            if (routingFlag) {
                req = req.setRouting(ents.getKey());
            }
            bulkRequest.add(req);
        }
    }
    DcBulkResponse response = DcBulkResponseImpl.getInstance(bulkRequest.execute().actionGet());
    return response;
}
 

/**
 * 非同期でドキュメントを登録する.
 * @param index インデックス名
 * @param type タイプ名
 * @param id ドキュメントのid
 * @param routingId routingId
 * @param data データ
 * @param opType 操作タイプ
 * @param version version番号
 * @return 非同期応答
 */
public ActionFuture<IndexResponse> asyncIndex(String index,
        String type,
        String id,
        String routingId,
        Map<String, Object> data,
        OpType opType,
        long version) {
    IndexRequestBuilder req = esTransportClient.prepareIndex(index, type, id).setSource(data).setOpType(opType)
            .setConsistencyLevel(WriteConsistencyLevel.DEFAULT).setRefresh(true);
    if (routingFlag) {
        req = req.setRouting(routingId);
    }
    if (version > -1) {
        req.setVersion(version);
    }

    ActionFuture<IndexResponse> ret = req.execute();
    EsRequestLogInfo logInfo = new EsRequestLogInfo(index, type, id, routingId, data, opType.toString(),
            version);
    this.fireEvent(Event.afterCreate, logInfo);

    return ret;
}
 

/**
 * ルーティングIDに関係なくバルクでドキュメントを登録.
 * @param index インデックス名
 * @param bulkMap バルクドキュメント
 * @return ES応答
 */
public DcBulkResponse asyncBulkCreate(
        String index, Map<String, List<EsBulkRequest>> bulkMap) {
    BulkRequestBuilder bulkRequest = esTransportClient.prepareBulk();
    // ルーティングIDごとにバルク登録を行うと効率が悪いため、引数で渡されたEsBulkRequestは全て一括登録する。
    // また、バルク登録後にactionGet()すると同期実行となるため、ここでは実行しない。
    // このため、execute()のレスポンスを返却し、呼び出し側でactionGet()してからレスポンスチェック、リフレッシュすること。
    for (Entry<String, List<EsBulkRequest>> ents : bulkMap.entrySet()) {
        for (EsBulkRequest data : ents.getValue()) {
            IndexRequestBuilder req = esTransportClient.
                    prepareIndex(index, data.getType(), data.getId()).setSource(data.getSource());
            if (routingFlag) {
                req = req.setRouting(ents.getKey());
            }
            bulkRequest.add(req);
        }
    }
    DcBulkResponse response = DcBulkResponseImpl.getInstance(bulkRequest.execute().actionGet());
    return response;
}
 

/**
 * 插入数据
 *
 * @param mapping 配置对象
 * @param pkVal 主键值
 * @param esFieldData 数据Map
 */
public void insert(ESMapping mapping, Object pkVal, Map<String, Object> esFieldData) {
    if (mapping.get_id() != null) {
        String parentVal = (String) esFieldData.remove("$parent_routing");
        if (mapping.isUpsert()) {
            UpdateRequestBuilder updateRequestBuilder = transportClient
                .prepareUpdate(mapping.get_index(), mapping.get_type(), pkVal.toString())
                .setDoc(esFieldData)
                .setDocAsUpsert(true);
            if (StringUtils.isNotEmpty(parentVal)) {
                updateRequestBuilder.setRouting(parentVal);
            }
            getBulk().add(updateRequestBuilder);
        } else {
            IndexRequestBuilder indexRequestBuilder = transportClient
                .prepareIndex(mapping.get_index(), mapping.get_type(), pkVal.toString())
                .setSource(esFieldData);
            if (StringUtils.isNotEmpty(parentVal)) {
                indexRequestBuilder.setRouting(parentVal);
            }
            getBulk().add(indexRequestBuilder);
        }
        commitBulk();
    } else {
        SearchResponse response = transportClient.prepareSearch(mapping.get_index())
            .setTypes(mapping.get_type())
            .setQuery(QueryBuilders.termQuery(mapping.getPk(), pkVal))
            .setSize(10000)
            .get();
        for (SearchHit hit : response.getHits()) {
            getBulk().add(transportClient.prepareUpdate(mapping.get_index(), mapping.get_type(), hit.getId())
                .setDoc(esFieldData));
            commitBulk();
        }
    }

}
 

@Override
protected IndexRequestBuilder completeIndexRequestBuilder(IndexRequestBuilder requestBuilder,
                                                          String resourceName, EntityContentProducer ecp,
                                                          boolean includeContent)
        throws IOException{
    return requestBuilder.setRouting(null);
}
 

/**
 * バルクリクエストのINDEXリクエストを作成する.
 * @param index インデックス名
 * @param routingId ルーティングID
 * @param data バルクドキュメント情報
 * @return 作成したINDEXリクエスト
 */
private IndexRequestBuilder createIndexRequest(String index, String routingId, EsBulkRequest data) {
    IndexRequestBuilder request = esTransportClient.
            prepareIndex(index, data.getType(), data.getId()).setSource(data.getSource());
    if (routingFlag) {
        request = request.setRouting(routingId);
    }
    return request;
}
 

/**
 * バルクリクエストのINDEXリクエストを作成する.
 * @param index インデックス名
 * @param routingId ルーティングID
 * @param data バルクドキュメント情報
 * @return 作成したINDEXリクエスト
 */
private IndexRequestBuilder createIndexRequest(String index, String routingId, EsBulkRequest data) {
    IndexRequestBuilder request = esTransportClient.
            prepareIndex(index, data.getType(), data.getId()).setSource(data.getSource());
    if (routingFlag) {
        request = request.setRouting(routingId);
    }
    return request;
}
 

/**
 * add based on supplied parameters.
 * @param indexName indexName
 * @param type type
 * @param id id
 * @param routing routing
 * @param ts ts
 * @param json json
 */
public void add(String indexName, String type, String id, String parent, String routing, String ts, String json) {

  // make sure that these are not null
  Objects.requireNonNull(indexName);
  Objects.requireNonNull(type);
  Objects.requireNonNull(json);

  IndexRequestBuilder indexRequestBuilder = manager.client()
      .prepareIndex(indexName, type)
      .setSource(json);

  // / They didn't specify an ID, so we will create one for them.
  if (id != null) {
    indexRequestBuilder.setId(id);
  }
  if (ts != null) {
    indexRequestBuilder.setTimestamp(ts);
  }
  if (parent != null) {
    indexRequestBuilder.setParent(parent);
  }
  if (routing != null) {
    indexRequestBuilder.setRouting(routing);
  }
  add(indexRequestBuilder.request());
}
 

@Override
protected IndexRequestBuilder completeIndexRequestBuilder(IndexRequestBuilder requestBuilder,
                                                          String resourceName, EntityContentProducer ecp,
                                                          boolean includeContent)
        throws IOException{
    return requestBuilder.setRouting(null);
}
 

@Override
protected IndexRequestBuilder completeIndexRequestBuilder(IndexRequestBuilder requestBuilder, String resourceName,
                                                          EntityContentProducer ecp, boolean includeContent) {
    return requestBuilder.setRouting(ecp.getSiteId(resourceName));
}
 

@Override
protected IndexRequestBuilder completeIndexRequestBuilder(IndexRequestBuilder requestBuilder, String resourceName,
                                                          EntityContentProducer ecp, boolean includeContent) {
    return requestBuilder.setRouting(ecp.getSiteId(resourceName));
}