io.searchbox.core.DocumentResult Java Examples

@Override
public IEsItem saveIndex(IEsItem doc) throws Exception {
	Index.Builder builder = new Index.Builder(doc);
	if (doc.getIndex() != null) {
		builder.index(doc.getIndex());
	}
	if (doc.getType() != null) {
		builder.type(doc.getType());
	}
	if (doc.getId() != null) {
		builder.id(doc.getId());
	}

	DocumentResult result = _exec(builder.build());
	if (result != null) {
		doc.setIndex(result.getIndex());
		doc.setType(result.getType());
		doc.setId(result.getId());
		return doc;
	}
	return null;
}
 

@Test
public void indexPartIterationTest() throws IndexerNotAvailableException, AccountNotFoundException, IndexerRequestException {
    DocumentResult documentResult = new DocumentResult(new Gson());
    PartIteration partIteration = new PartIteration();
    Mockito.when(indexManager.executeUpdate(Matchers.any(Update.Builder.class)))
            .thenReturn(documentResult);
    indexerManagerBean.indexPartIteration(partIteration);
    Mockito.verify(indexManager,times(1)).executeUpdate(Matchers.any(Update.Builder.class));

}
 

@Test
public void indexDocumentIterationTest() throws IndexerNotAvailableException, AccountNotFoundException, IndexerRequestException {
    DocumentResult documentResult = new DocumentResult(new Gson());
    DocumentIteration documentIteration = new DocumentIteration();
    Mockito.when(indexManager.executeUpdate(Matchers.any(Update.Builder.class)))
        .thenReturn(documentResult);
    indexerManagerBean.indexDocumentIteration(documentIteration);
    Mockito.verify(indexManager,times(1)).executeUpdate(Matchers.any(Update.Builder.class));

}
 

@Override
public boolean delete(IEsItem doc) throws Exception {
	Delete.Builder builder = new Delete.Builder(doc.getId());
	if (doc.getIndex() != null) {
		builder.index(doc.getIndex());
	}
	if (doc.getType() != null) {
		builder.type(doc.getType());
	}
	DocumentResult result = _exec(builder.build());
	if (result != null) {
		return true;
	}
	return false;
}
 

@Override
public FailedItemSource createItem(AbstractDocumentTargetedAction<DocumentResult> failed) {
    BufferedIndex failedRequest = (BufferedIndex) failed;
    if (failedRequest.getSource() instanceof FailedItemSource) {
        LOG.trace("Reusing {}", FailedItemSource.class.getSimpleName());
        return (FailedItemSource) failedRequest.getSource();
    }
    return new FailedItemSource<>(failedRequest.getSource(), createInfo(failed));
}
 

@Override
public IEsItem get(IEsItem doc) throws Exception {
	Get.Builder builder = new Get.Builder(doc.getIndex(), doc.getId());
	if (doc.getType() != null) {
		builder.type(doc.getType());
	}
	DocumentResult result = _exec(builder.build());
	if (result != null) {
		return result.getSourceAsObject(doc.getClass());
	}
	return null;
}
 

@Override
public FailedItemSource createItem(AbstractDocumentTargetedAction<DocumentResult> failedItem) {
    FailedItemInfo failedItemInfo = createInfo(failedItem);
    return new FailedItemSource<>(
            new StringItemSource((String) itemIntrospector.getPayload(failedItem)), failedItemInfo
    );
}
 

@Test
public void intropectorCanAccessActionPayload() {

    // given
    BatchItemIntrospector<AbstractDocumentTargetedAction<DocumentResult>> introspector = new JestActionIntrospector();
    String testPayload = "testPayload";
    AbstractDocumentTargetedAction<DocumentResult> action = new Index.Builder(testPayload).build();

    // when
    String payload = (String) introspector.getPayload(action);

    // then
    Assert.assertEquals(testPayload, payload);
}
 

@Override
public String getPayload(AbstractDocumentTargetedAction<DocumentResult> introspected) {
    return (String) introspected.payload;
}
 

public AbstractDocumentTargetedAction<DocumentResult> createDefaultTestIndexRequest(ItemSource<ByteBuf> itemSource, String expectedIndex) {
    return new BufferedIndex.Builder(itemSource)
            .type(UUID.randomUUID().toString())
            .index(expectedIndex)
            .build();
}
 

public AbstractDocumentTargetedAction<DocumentResult> createDefaultTestIndexRequest(ItemSource<String> itemSource, String expectedIndex) {
    return new Index.Builder(itemSource.getSource())
            .type(UUID.randomUUID().toString())
            .index(expectedIndex)
            .build();
}
 

protected FailedItemOps<AbstractDocumentTargetedAction<DocumentResult>> failedItemOps() {
    return new BufferedHttpFailedItemOps();
}
 

@Override
public FailedItemInfo createInfo(AbstractDocumentTargetedAction<DocumentResult> failed) {
    return new FailedItemInfo(failed.getIndex());
}
 

@Override
public FailedItemInfo createInfo(
        AbstractDocumentTargetedAction<DocumentResult> failed
) {
    return new FailedItemInfo(failed.getIndex());
}
 

@Override
public DocumentResult createNewElasticSearchResult(String responseBody, int statusCode, String reasonPhrase, Gson gson) {
    throw new UnsupportedOperationException("BufferedIndex cannot handle String result. Use buffer-based API");
}
 

protected FailedItemOps<AbstractDocumentTargetedAction<DocumentResult>> failedItemOps() {
    return new JestHttpFailedItemOps();
}
 

/**
* 删除数据
* @param indexName
* @param typeName
* @param id
* @return
* @throws Exception
*/
public boolean delete(JestClient jestClient,String indexName, String typeName, String id) throws Exception {  
    DocumentResult dr = jestClient.execute(new Delete.Builder(id).index(indexName).type(typeName).build());  
    return dr.isSucceeded();  
}
 

/**
* 删除数据
* @param indexName
* @param typeName
* @param id
* @return
* @throws Exception
*/
public boolean delete(JestClient jestClient,String indexName, String typeName, String id) throws Exception {  
    DocumentResult dr = jestClient.execute(new Delete.Builder(id).index(indexName).type(typeName).build());  
    return dr.isSucceeded();  
}