Как получить от BulkItemResponse на соответствующий запрос

Question

Я использую Elasticsearch's bulk requests в Java, и я пытаюсь справиться с ситуациями, когда какая-то ошибка происходит:

BulkResponse bulkResponse = bulkRequest.get(); 
if (bulkResponse.hasFailures()) { 
    for (BulkItemResponse response : bulkResponse) { 
     if (response.isFailed() 
     || response.getResponse().getShardInfo().getFailed() > 0) { 
      //Find the corresponding request and resend it 
     } 
    } 
} 

После нахождения ответа с ошибкой, я хочу повторно отправить запрос, поскольку в моем случае ошибки могут быть кратковременными, и в большинстве случаев повторная попытка может решить проблему.

Так что мой вопрос: как добраться от BulkItemResponse до исходного Request, который привел его? Есть ли способ лучше, чем полагаться на порядок запросов и ответов?

Answer 1

Нет, у вас нет этого AFAIK. Вам нужно каким-то образом сохранить все элементы, добавленные вами в массив, а в случае ошибки используйте идентификатор, исходящий от response.getId(), чтобы связать это с исходными данными.

Так что-то подобное:

HashMap<String, Object> myData = new HashMap<>(); 

BulkRequestBuilder brb = BulkAction.INSTANCE.newRequestBuilder(client); 

myData.put(myObject.getId(), myObject); 
brb.add(new IndexRequest("index", "type", myObject.getId()).source(myObject.toJson())); 

// Other actions 
BulkResponse response = client.bulk(brb.request()).get(); 
response.forEach(bulkItemResponse -> { 
    if (bulkItemResponse.isFailed()) { 
     Object objectToSendAgain = myData.get(bulkItemResponse.getId()); 
     // Do what is needed with this object 
    } 
}); 

myData = null; 

Я надеюсь, что это помогает.