ElasticSearch使用代码

package elasticsearch01;

import static org.junit.Assert.*;

import java.util.HashMap;
import java.util.List;
import java.util.Map;

import org.elasticsearch.action.bulk.BulkItemResponse;
import org.elasticsearch.action.bulk.BulkRequestBuilder;
import org.elasticsearch.action.bulk.BulkResponse;
import org.elasticsearch.action.count.CountResponse;
import org.elasticsearch.action.delete.DeleteRequest;
import org.elasticsearch.action.delete.DeleteResponse;
import org.elasticsearch.action.deletebyquery.DeleteByQueryResponse;
import org.elasticsearch.action.get.GetResponse;
import org.elasticsearch.action.index.IndexRequest;
import org.elasticsearch.action.index.IndexResponse;
import org.elasticsearch.action.search.SearchRequestBuilder;
import org.elasticsearch.action.search.SearchResponse;
import org.elasticsearch.action.search.SearchType;
import org.elasticsearch.action.update.UpdateRequest;
import org.elasticsearch.action.update.UpdateResponse;
import org.elasticsearch.client.transport.TransportClient;
import org.elasticsearch.cluster.node.DiscoveryNode;
import org.elasticsearch.common.collect.ImmutableList;
import org.elasticsearch.common.settings.ImmutableSettings;
import org.elasticsearch.common.settings.Settings;
import org.elasticsearch.common.text.Text;
import org.elasticsearch.common.transport.InetSocketTransportAddress;
import org.elasticsearch.common.transport.TransportAddress;
import org.elasticsearch.common.xcontent.XContentBuilder;
import org.elasticsearch.common.xcontent.XContentFactory;
import org.elasticsearch.index.query.FilterBuilders;
import org.elasticsearch.index.query.QueryBuilders;
import org.elasticsearch.search.SearchHit;
import org.elasticsearch.search.SearchHits;
import org.elasticsearch.search.aggregations.Aggregation;
import org.elasticsearch.search.aggregations.AggregationBuilders;
import org.elasticsearch.search.aggregations.bucket.terms.Terms;
import org.elasticsearch.search.aggregations.bucket.terms.Terms.Bucket;
import org.elasticsearch.search.aggregations.metrics.sum.Sum;
import org.elasticsearch.search.highlight.HighlightField;
import org.elasticsearch.search.sort.SortOrder;
import org.junit.Before;
import org.junit.Test;

import com.fasterxml.jackson.databind.ObjectMapper;

public class EsTest {

    String index = "crxy";
    String type = "emp";
    TransportClient transportClient;
    /**
     * 相当于初始化方法，在执行其他测试类之前会首先被调用
     * @throws Exception
     */
    @Before
    public void before() throws Exception {
        transportClient = new TransportClient();
        TransportAddress transportAddress = new InetSocketTransportAddress("192.168.1.170", 9300);
        transportClient.addTransportAddress(transportAddress);
    }

    /**
     * 自己写测试类的时候可以用这个
     * @throws Exception
     */
    @Test
    public void test1() throws Exception {

        GetResponse response = transportClient.prepareGet(index , type , "1").execute().actionGet();
        String sourceAsString = response.getSourceAsString();
        System.out.println(sourceAsString);
    }

    /**
     * 工作中使用-1
     * @throws Exception
     */
    @Test
    public void test2() throws Exception {
        Settings settings = ImmutableSettings.settingsBuilder().put("cluster.name", "elasticsearch").build();
        TransportClient transportClient = new TransportClient(settings);
        TransportAddress transportAddress = new InetSocketTransportAddress("192.168.1.170", 9300);
        TransportAddress transportAddress1 = new InetSocketTransportAddress("192.168.1.171", 9300);
        transportClient.addTransportAddresses(transportAddress,transportAddress1);
        GetResponse response = transportClient.prepareGet(index , type , "1").execute().actionGet();
        String sourceAsString = response.getSourceAsString();
        System.out.println(sourceAsString);
    }

    /**
     * 工作中使用-2
     * @throws Exception
     */
    @Test
    public void test3() throws Exception {
        Settings settings = ImmutableSettings.settingsBuilder()
                .put("cluster.name", "elasticsearch")
                .put("client.transport.sniff", true)//自动嗅探机制，可以自动链接集群中的其他节点
                .build();
        TransportClient transportClient = new TransportClient(settings);
        TransportAddress transportAddress = new InetSocketTransportAddress("192.168.1.170", 9300);
        TransportAddress transportAddress1 = new InetSocketTransportAddress("192.168.1.171", 9300);
        transportClient.addTransportAddresses(transportAddress,transportAddress1);

        ImmutableList<DiscoveryNode> connectedNodes = transportClient.connectedNodes();
        for (DiscoveryNode discoveryNode : connectedNodes) {
            System.out.println(discoveryNode.getHostAddress());
        }

    }

    /**
     * index -json格式
     * @throws Exception
     */
    @Test
    public void test4() throws Exception {
        String source = "{\"name\":\"mm\",\"age\":\"19\"}";
        IndexResponse response = transportClient.prepareIndex(index, type, "2").setSource(source).execute().actionGet();
        String id = response.getId();
        System.out.println(id);
    }

    /**
     * index - map
     * @throws Exception
     */
    @Test
    public void test5() throws Exception {
        HashMap<String, Object> source = new HashMap<String, Object>();
        source.put("name", "ww");
        source.put("age", 20);

        IndexResponse response = transportClient.prepareIndex(index, type).setSource(source).execute().actionGet();
        String id = response.getId();
        System.out.println(id);

    }

    /**
     * index - 对象
     * @throws Exception
     */
    @Test
    public void test6() throws Exception {
        User user = new User();
        user.setAge(29);
        user.setName("nn");

        ObjectMapper objectMapper = new ObjectMapper();
        IndexResponse response = transportClient.prepareIndex(index, type, "4").setSource(objectMapper.writeValueAsString(user)).execute().actionGet();
        String id = response.getId();
        System.out.println(id);

    }

    /**
     * index -eshelp
     * @throws Exception
     */
    @Test
    public void test7() throws Exception {
        XContentBuilder endObject = XContentFactory.jsonBuilder().startObject().field("name", "lk").field("age", 28).endObject();
        IndexResponse response = transportClient.prepareIndex(index, type, "5").setSource(endObject).get();
        String id = response.getId();
        System.out.println(id);
    }

    /**
     * get
     * @throws Exception
     */
    @Test
    public void test8() throws Exception {
        GetResponse response = transportClient.prepareGet(index, type, "5").get();
        System.out.println(response.getSourceAsString());
    }

    /**
     * update
     * @throws Exception
     */
    @Test
    public void test9() throws Exception {
        XContentBuilder endObject = XContentFactory.jsonBuilder().startObject().field("name","zs").endObject();

        UpdateResponse response = transportClient.prepareUpdate(index, type, "5").setDoc(endObject).get();
        System.out.println(response.getVersion());

    }

    /**
     * upsert
     * @throws Exception
     */
    @Test
    public void test10() throws Exception {
        UpdateRequest request = new UpdateRequest();
        request.index(index);
        request.type(type);
        request.id("6");

        XContentBuilder endObject = XContentFactory.jsonBuilder().startObject().field("name", "aa").endObject();
        request.doc(endObject);

        XContentBuilder endObject2 = XContentFactory.jsonBuilder().startObject().field("name", "crxy").field("age", 10).endObject();
        request.upsert(endObject2);

        UpdateResponse response = transportClient.update(request ).get();

        System.out.println(response.getVersion());
    }

    /**
     * 删除
     * @throws Exception
     */
    @Test
    public void test11() throws Exception {
        DeleteResponse response = transportClient.prepareDelete(index, type, "6").get();
        System.out.println(response.getVersion());
    }

    /**
     * 删除
     * @throws Exception
     */
    @Test
    public void test11_1() throws Exception {
        DeleteByQueryResponse response = transportClient.prepareDeleteByQuery(index).setQuery(QueryBuilders.matchAllQuery()).get();
    }

    /**
     * 查询索引库中数据的总量,类似于sql中的select *
     * @throws Exception
     */
    @Test
    public void test12() throws Exception {
        CountResponse response = transportClient.prepareCount(index).get();
        System.out.println(response.getCount());
    }

    /**
     * 批量操作 bulk
     * @throws Exception
     */
    @Test
    public void test13() throws Exception {
        BulkRequestBuilder bulkBuilder = transportClient.prepareBulk();
        IndexRequest indexrequest = new IndexRequest(index,type,"6");
        XContentBuilder endObject = XContentFactory.jsonBuilder().startObject().field("name", "sss").field("age111", 001).endObject();
        indexrequest.source(endObject);
        //TODO---

        bulkBuilder.add(indexrequest );
        DeleteRequest deleteRequest = new DeleteRequest(index,type,"5");
        bulkBuilder.add(deleteRequest);

        BulkResponse bulkResponse = bulkBuilder.get();

        if(bulkResponse.hasFailures()){
            System.out.println("执行失败：");
            BulkItemResponse[] items = bulkResponse.getItems();
            for (BulkItemResponse bulkItemResponse : items) {
                String failureMessage = bulkItemResponse.getFailureMessage();
                System.out.println(failureMessage);
            }
        }else{
            System.out.println("正常执行");
        }

    }
    /**
     * 查询，排序，分页，高亮，过滤
     * lt:小于
     * lte：小于等于
     * gt：大于
     * gte：大于等于
     * @throws Exception
     */
    @Test
    public void test14() throws Exception {
        SearchResponse searchResponse = transportClient.prepareSearch(index)
                            .setTypes(type)
                            .setSearchType(SearchType.QUERY_THEN_FETCH)
                            .setQuery(QueryBuilders.matchQuery("name", "zs"))
                            //.setPostFilter(FilterBuilders.rangeFilter("age").gt(20).lte(28))
                            .setFrom(0)
                            .setSize(10)
                            .addHighlightedField("name")
                            .setHighlighterPreTags("<font color='red'>")
                            .setHighlighterPostTags("</font>")
                            .addSort("age", SortOrder.DESC)
                            .get();

        SearchHits hits = searchResponse.getHits();
        long totalHits = hits.getTotalHits();
        System.out.println("总数："+totalHits);

        for (SearchHit searchHit : hits) {
            Map<String, HighlightField> highlightFields = searchHit.getHighlightFields();
            HighlightField highlightField = highlightFields.get("name");
            Text[] fragments = highlightField.fragments();
            System.out.println(searchHit.getSourceAsString());
            for (Text text : fragments) {
                System.out.println("高亮内容"+text);
            }
        }
    }

    /**
     * 类似于这个select count(*),name from table group by name;
     * @throws Exception
     */
    @Test
    public void test15() throws Exception {
        SearchResponse searchResponse = transportClient.prepareSearch(index)
        .setTypes(type)
        .addAggregation(AggregationBuilders.terms("nameterm").field("name").size(0)).get();

        Terms terms = searchResponse.getAggregations().get("nameterm");

        List<Bucket> buckets = terms.getBuckets();
        for (Bucket bucket : buckets) {
            System.out.println(bucket.getKey()+"--->"+bucket.getDocCount());
        }
    }

    /**
     * 类似于select sum(age),name from table group by name;
     * @throws Exception
     */
    @Test
    public void test16() throws Exception {
        SearchResponse searchResponse = transportClient.prepareSearch(index)
        .setTypes(type)
        .addAggregation(AggregationBuilders.terms("nameterm").field("name").
                    subAggregation(AggregationBuilders.sum("agesum").field("age")).size(0)).get();

        Terms terms = searchResponse.getAggregations().get("nameterm");

        List<Bucket> buckets = terms.getBuckets();
        for (Bucket bucket : buckets) {
            Sum sum = bucket.getAggregations().get("agesum");

            System.out.println(bucket.getKey()+"--->"+sum.getValue());
        }
    }

}