SearchTest.java

package org.xbib.elasticsearch.helper;

import org.elasticsearch.action.bulk.BulkAction;
import org.elasticsearch.action.bulk.BulkRequestBuilder;
import org.elasticsearch.action.search.SearchRequestBuilder;
import org.elasticsearch.action.search.SearchResponse;
import org.elasticsearch.client.Client;
import org.elasticsearch.index.query.QueryBuilder;
import org.elasticsearch.index.query.QueryBuilders;
import org.elasticsearch.search.sort.SortOrder;
import org.junit.Test;
import org.xbib.elasticsearch.NodeTestUtils;

import static org.elasticsearch.client.Requests.indexRequest;
import static org.elasticsearch.client.Requests.refreshRequest;
import static org.elasticsearch.common.xcontent.XContentFactory.jsonBuilder;

public class SearchTest extends NodeTestUtils {

    @Test
    public void testSearch() throws Exception {
        Client client = client("1");
        long t0 = System.currentTimeMillis();
        BulkRequestBuilder builder = new BulkRequestBuilder(client, BulkAction.INSTANCE);
        for (int i = 0; i < 1000; i++) {
            builder.add(indexRequest()
                    .index("pages").type("row")
                    .source(jsonBuilder()
                            .startObject()
                            .field("user1", "kimchy")
                            .field("user2", "kimchy")
                            .field("user3", "kimchy")
                            .field("user4", "kimchy")
                            .field("user5", "kimchy")
                            .field("user6", "kimchy")
                            .field("user7", "kimchy")
                            .field("user8", "kimchy")
                            .field("user9", "kimchy")
                            .field("rowcount", i)
                            .field("rs", 1234)));
        }
        client.bulk(builder.request()).actionGet();

        client.admin().indices().refresh(refreshRequest()).actionGet();

        long t1 = System.currentTimeMillis();
        logger.info("t1-t0 = {}", t1-t0);

        for (int i = 0; i < 100; i++) {
            t1 = System.currentTimeMillis();
            QueryBuilder queryStringBuilder =
                    QueryBuilders.queryStringQuery("rs:" + 1234);
            SearchRequestBuilder requestBuilder = client.prepareSearch()
                    .setIndices("pages")
                    .setTypes("row")
                    .setQuery(queryStringBuilder)
                    .addSort("rowcount", SortOrder.DESC)
                    .setFrom(i*10).setSize(10);
            SearchResponse response = requestBuilder.execute().actionGet();
            long t2 = System.currentTimeMillis();
            logger.info("t2-t1 = {}", t2-t1);
        }
    }
}