胖不了小陆

信你所信，为你所见

0%

Spring Boot (十二)与 Elasticsearch 整合

发表于 分类于

Springboot 与 elasticsearch 整合demo

springboot 整合elasticsearch5.6.4 版本的

一、pom.xml

pom.xml 主要片段

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
<properties>
	<project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
	<project.reporting.outputEncoding>UTF-8</project.reporting.outputEncoding>
	<java.version>1.8</java.version>
	<elasticsearch.version>5.6.4</elasticsearch.version>
</properties>

<!-- 接口注解需要用到 -->
 <dependency>
	<groupId>org.springframework.boot</groupId>
	<artifactId>spring-boot-starter-web</artifactId>
</dependency>
<!-- 重要 -->
<dependency>
	<groupId>org.elasticsearch.client</groupId>
	<artifactId>transport</artifactId>
	<version>${elasticsearch.version}</version>
</dependency>
<!-- 需要 -->
<dependency>
	<groupId>org.apache.logging.log4j</groupId>
	<artifactId>log4j-core</artifactId>
</dependency>

二、application.properties 内容

1
2
3
4
5
# 你集群的名字
elasticsearch.cluster.name=people
# 地址，多个用逗号隔开，注意端口，9300 是transportService 的端口。
elasticsearch.host=127.0.0.1:9300,127.0.0.1:9301

三、自定义Elasticsearch 配置类

这个配置类是为了获取TransportClient 

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
package top.lrshuai.es.config;

import org.elasticsearch.client.transport.TransportClient;
import org.elasticsearch.common.settings.Settings;
import org.elasticsearch.common.transport.InetSocketTransportAddress;
import org.elasticsearch.transport.client.PreBuiltTransportClient;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

import java.net.InetAddress;
import java.net.UnknownHostException;

@Configuration
public class ElasticsearchConfig {

	@Value("${elasticsearch.cluster.name}")
	private String clusterName;

	@Value("${elasticsearch.host}")
	private String host;

	@Bean
	public TransportClient transportClient() throws UnknownHostException {
		// 设置集群名称
		Settings settings = Settings.builder().put("cluster.name", clusterName)
				.build();
		TransportClient transportClient = new PreBuiltTransportClient(settings);
		String[] nodes = host.split(",");
		for (String node : nodes) {
			if (node.length() > 0) {
				String[] hostPort = node.split(":");
				transportClient.addTransportAddress(
						new InetSocketTransportAddress(
						InetAddress.getByName(hostPort[0]),
						Integer.parseInt(hostPort[1])));
			}
		}
		return transportClient;
	}

}

四、自定义查询的接口与实现

下面是我的一个例子，可以参考下，然后按照自己的需求定制查询接口

1、PersonDao 接口

1
2
3
4
5
6
7
8
9
10
11
12
13
package top.lrshuai.es.dao;


import top.lrshuai.es.entity.Person;

public interface PersonDao {
	public String save(Person person);
	public String update(Person person);
	public String deltele(String id);
	public Object find(String id);
	public Object query(Person person);
}

2、PersonDaoImpl 接口的实现类

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
package top.lrshuai.es.dao.impl;

import java.io.IOException;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.concurrent.ExecutionException;

import org.apache.log4j.Logger;
import org.elasticsearch.action.delete.DeleteResponse;
import org.elasticsearch.action.get.GetResponse;
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.common.xcontent.XContentBuilder;
import org.elasticsearch.common.xcontent.XContentFactory;
import org.elasticsearch.index.query.BoolQueryBuilder;
import org.elasticsearch.index.query.QueryBuilders;
import org.elasticsearch.index.query.RangeQueryBuilder;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import top.lrshuai.es.dao.PersonDao;
import top.lrshuai.es.entity.Person;

@Component
public class PersonDaoImpl implements PersonDao{

	@Autowired
    private TransportClient transportClient;
	
	private Logger log = Logger.getLogger(getClass());
	//索引名称（数据库名）
	private String index = "test";
	
	//类型名称（表名）
	private String type = "person";
	
	/**
	 * 保存
	 */
	@Override
	public String save(Person person) {
		 try {
			 XContentBuilder builder = XContentFactory.jsonBuilder().startObject();
			builder.field("name", person.getName());
			builder.field("age", person.getAge());
			builder.field("sex", person.getSex());
			builder.field("birthday", person.getBirthday());
			builder.field("introduce", person.getIntroduce());
			builder.endObject();
			IndexResponse response = this.transportClient.prepareIndex(index, type)
					.setSource(builder).get();
			return response.getId();
		} catch (IOException e) {
			e.printStackTrace();
			log.error(e.getMessage(), e);
		}
		return null;
	}
	
	/**
	 * 更新
	 */
	@Override
	public String update(Person person) {
		UpdateRequest request = new UpdateRequest(index, type, person.getId());
        try {
            XContentBuilder builder = XContentFactory.jsonBuilder().startObject();
            if (person.getName() != null) {
                builder.field("name", person.getName());
            }
            if (person.getSex() != null) {
            	builder.field("sex", person.getSex());
            }
            if (person.getIntroduce() != null) {
            	builder.field("introduce", person.getIntroduce());
            }
            if (person.getBirthday() != null) {
            	builder.field("birthday", person.getBirthday());
            }
            if (person.getAge() > 0) {
            	builder.field("age", person.getAge());
            }
            builder.endObject();
            request.doc(builder);
            UpdateResponse response = transportClient.update(request).get();
            return response.getId();
        } catch (IOException | InterruptedException | ExecutionException e) {
            log.error(e.getMessage(), e);
        }
        return null;
	}
	
	@Override
	public String deltele(String id) {
		DeleteResponse response = transportClient.prepareDelete(index, type, id).get();
		return response.getId();
	}
	
	@Override
	public Object find(String id) {
		GetResponse response =transportClient.prepareGet(index, type, id).get();
		System.out.println("response="+response);
		Map<String, Object> result = response.getSource();
		if(result != null) {
			result.put("_id", response.getId());
		}
		return result;
	}
	
	@Override
	public Object query(Person person) {
		List<Map<String,Object>> result =  new ArrayList<>();;
		try {
			 BoolQueryBuilder boolBuilder = QueryBuilders.boolQuery();
	        if (person.getName() != null) {
//	            boolBuilder.must(QueryBuilders.matchQuery("name", person.getName()));
	            boolBuilder.should(QueryBuilders.matchQuery("name", person.getName()));
	        }
	        if (person.getIntroduce() != null) {
//	            boolBuilder.must(QueryBuilders.matchQuery("introduce", person.getIntroduce()));
	            boolBuilder.should(QueryBuilders.matchQuery("introduce", person.getIntroduce()));
	        }
	        
	        //range 查询范围，大于age,小于age+10
	        if(person.getAge() > 0) {
	        	RangeQueryBuilder rangeQuery = QueryBuilders.rangeQuery("age");
	        	rangeQuery.from(person.getAge());
	        	rangeQuery.to(person.getAge()+10);
	        	boolBuilder.filter(rangeQuery);
	        }
	        SearchRequestBuilder builder = transportClient.prepareSearch(index)
	                    .setTypes(type)
	                    .setSearchType(SearchType.QUERY_THEN_FETCH)
	                    .setQuery(boolBuilder)
	                    .setFrom(0)
	                    .setSize(10);
			// 高亮
			//HighlightBuilder hBuilder = new HighlightBuilder();
			//hBuilder.preTags("<h2>");
			//hBuilder.postTags("</h2>");	
			//hBuilder.field("question"); 
			//高亮的字段
			//builder.highlighter(hBuilder);
	        log.info(String.valueOf(builder));
	        SearchResponse response = builder.get();
	        response.getHits().forEach((s)->result.add(s.getSource()));
		} catch (Exception e) {
			e.printStackTrace();
			log.error(e.getMessage(), e);
		}
		return result;
	}
	
}

五测试类

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
package top.lrshuai.es;

import java.util.Date;

import org.junit.Test;
import org.junit.runner.RunWith;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.test.context.junit4.SpringRunner;

import top.lrshuai.es.entity.Person;
import top.lrshuai.es.service.PersonService;

@RunWith(SpringRunner.class)
@SpringBootTest
public class ApplicationTests {

	@Autowired
	private PersonService personService;
	
	@Test
	public void testSavePerson() {
		String name = "帅大叔";
		String introduce = "宇宙超级无敌帅";
		Person person = new Person(name, 23, "男", new Date(), introduce);
		System.out.println(personService.savePerson(person));
	}
	
	@Test
	public void testUpdatePerson() {
		String name = "靓女";
		int age = 24;
		String sex = "女";
		String introduce = "这是一个非常非常非常非常漂亮的女孩。";
		Date birthday = new Date();
		Person person = new Person(name, age, sex, birthday, introduce);
		person.setId("mjupFmABhhkOZSWoch9i");
		personService.updatePerson(person);
	}
	
	@Test
	public void testFindPerson() {
		String id = "mjupFmABhhkOZSWoch9i";
		System.out.println(personService.findPerson(id));
	}
	
	@Test
	public void testDelPerson() {
		String id = "mjupFmABhhkOZSWoch9i";
		System.out.println(personService.delPerson(id));
	}
	
	@Test
	public void testQueryPerson() {
		Person person = new Person();
		person.setName("帅");
		person.setIntroduce("人");
//		person.setAge(27);
		Object obj  = personService.queryPerson(person);
		System.out.println(obj);
	}

}

示例代码：Github

您的打赏，是我创作的动力！不给钱？那我只能靠想象力充饥了。