Skip to content

Commit 14b5853

Browse files
dunwudunwu
committed
feat: 更新 elasticsearch 示例
1 parent a1d04b3 commit 14b5853

File tree

18 files changed

+1012
-261
lines changed

18 files changed

+1012
-261
lines changed

‎codes/javadb/elasticsearch/elasticsearch6/pom.xml

+33-17
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,14 @@
11
<?xml version="1.0" encoding="UTF-8"?>
22
<project xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xmlns="http://maven.apache.org/POM/4.0.0"
3-
xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
3+
xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
44
<modelVersion>4.0.0</modelVersion>
55

6+
<parent>o
7+
<groupId>org.springframework.boot</groupId>
8+
<artifactId>spring-boot-starter-parent</artifactId>
9+
<version>2.7.7</version>
10+
</parent>
11+
612
<groupId>io.github.dunwu</groupId>
713
<artifactId>javadb-elasticsearch6</artifactId>
814
<version>1.0.0</version>
@@ -17,37 +23,47 @@
1723
</properties>
1824

1925
<dependencies>
26+
<dependency>
27+
<groupId>org.springframework.boot</groupId>
28+
<artifactId>spring-boot-starter-aop</artifactId>
29+
</dependency>
30+
<dependency>
31+
<groupId>org.springframework.boot</groupId>
32+
<artifactId>spring-boot-starter-web</artifactId>
33+
</dependency>
2034
<dependency>
2135
<groupId>org.elasticsearch.client</groupId>
2236
<artifactId>elasticsearch-rest-high-level-client</artifactId>
23-
<version>6.4.3</version>
2437
</dependency>
2538
<dependency>
2639
<groupId>org.projectlombok</groupId>
2740
<artifactId>lombok</artifactId>
28-
<version>1.18.22</version>
2941
</dependency>
3042
<dependency>
3143
<groupId>cn.hutool</groupId>
3244
<artifactId>hutool-all</artifactId>
33-
<version>5.7.20</version>
34-
</dependency>
35-
<dependency>
36-
<groupId>com.fasterxml.jackson.core</groupId>
37-
<artifactId>jackson-databind</artifactId>
38-
<version>2.15.2</version>
45+
<version>5.8.8</version>
3946
</dependency>
4047

4148
<dependency>
42-
<groupId>ch.qos.logback</groupId>
43-
<artifactId>logback-classic</artifactId>
44-
<version>1.2.10</version>
45-
</dependency>
46-
<dependency>
47-
<groupId>org.apache.logging.log4j</groupId>
48-
<artifactId>log4j-to-slf4j</artifactId>
49-
<version>2.17.1</version>
49+
<groupId>org.springframework.boot</groupId>
50+
<artifactId>spring-boot-starter-test</artifactId>
51+
<scope>test</scope>
5052
</dependency>
5153
</dependencies>
54+
<dependencyManagement>
55+
<dependencies>
56+
<dependency>
57+
<groupId>org.elasticsearch.client</groupId>
58+
<artifactId>elasticsearch-rest-high-level-client</artifactId>
59+
<version>6.4.3</version>
60+
</dependency>
61+
<dependency>
62+
<groupId>org.elasticsearch</groupId>
63+
<artifactId>elasticsearch</artifactId>
64+
<version>6.4.3</version>
65+
</dependency>
66+
</dependencies>
67+
</dependencyManagement>
5268

5369
</project>

‎codes/javadb/elasticsearch/elasticsearch6/src/main/java/io/github/dunwu/javadb/elasticsearch/Demo.java

+5-6
Original file line numberDiff line numberDiff line change
@@ -2,29 +2,28 @@
22

33
import io.github.dunwu.javadb.elasticsearch.entity.User;
44
import io.github.dunwu.javadb.elasticsearch.mapper.UserEsMapper;
5-
import io.github.dunwu.javadb.elasticsearch.util.ElasticsearchUtil;
6-
import org.elasticsearch.client.RestHighLevelClient;
75

86
import java.io.IOException;
97
import java.util.Arrays;
108
import java.util.List;
119

1210
public class Demo {
1311

14-
private static final String HOSTS = "127.0.0.1:9200";
15-
private static final RestHighLevelClient restHighLevelClient = ElasticsearchUtil.newRestHighLevelClient(HOSTS);
12+
private static final String env = "test";
13+
private static final ElasticsearchTemplate elasticsearchTemplate
14+
= ElasticsearchFactory.newElasticsearchTemplate(env);
1615

1716
public static void main(String[] args) throws IOException, InterruptedException {
1817

19-
UserEsMapper mapper = new UserEsMapper(restHighLevelClient);
18+
UserEsMapper mapper = new UserEsMapper(elasticsearchTemplate);
2019

2120
System.out.println("索引是否存在：" + mapper.isIndexExists());
2221

2322
User jack = User.builder().id(1L).username("jack").age(18).build();
2423
User tom = User.builder().id(2L).username("tom").age(20).build();
2524
List<User> users = Arrays.asList(jack, tom);
2625

27-
System.out.println("批量插入：" + mapper.batchInsert(users));
26+
System.out.println("批量插入：" + mapper.batchSave(users));
2827
System.out.println("根据ID查询：" + mapper.getById("1").toString());
2928
System.out.println("根据ID查询：" + mapper.pojoById("2").toString());
3029
System.out.println("根据ID批量查询：" + mapper.pojoListByIds(Arrays.asList("1", "2")).toString());

‎codes/javadb/elasticsearch/elasticsearch6/src/main/java/io/github/dunwu/javadb/elasticsearch/ElasticsearchFactory.java

+166
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,166 @@
1+
package io.github.dunwu.javadb.elasticsearch;
2+
3+
import cn.hutool.core.collection.CollectionUtil;
4+
import cn.hutool.core.util.ArrayUtil;
5+
import cn.hutool.core.util.StrUtil;
6+
import lombok.extern.slf4j.Slf4j;
7+
import org.apache.http.HttpHost;
8+
import org.elasticsearch.client.RestClient;
9+
import org.elasticsearch.client.RestClientBuilder;
10+
import org.elasticsearch.client.RestHighLevelClient;
11+
12+
import java.util.List;
13+
import java.util.stream.Collectors;
14+
15+
/**
16+
* Elasticsearch 客户端实例工厂
17+
*
18+
* @author <a href="mailto:forbreak@163.com">Zhang Peng</a>
19+
* @date 2024-02-07
20+
*/
21+
@Slf4j
22+
public class ElasticsearchFactory {
23+
24+
public static int CONNECT_TIMEOUT_MILLIS = 1000;
25+
26+
public static int SOCKET_TIMEOUT_MILLIS = 30000;
27+
28+
public static int CONNECTION_REQUEST_TIMEOUT_MILLIS = 500;
29+
30+
public static int MAX_CONN_TOTAL = 30;
31+
32+
public static int MAX_CONN_PER_ROUTE = 10;
33+
34+
public static RestClient newRestClient() {
35+
// 从配置中心读取环境变量
36+
String env = "test";
37+
return newRestClient(env);
38+
}
39+
40+
public static RestClient newRestClient(String env) {
41+
String hosts = getDefaultEsAddress(env);
42+
return newRestClient(toHttpHostList(hosts));
43+
}
44+
45+
public static RestClient newRestClient(HttpHost[] httpHosts) {
46+
RestClientBuilder builder = getRestClientBuilder(httpHosts);
47+
if (builder == null) {
48+
return null;
49+
}
50+
try {
51+
return builder.build();
52+
} catch (Exception e) {
53+
log.error("【ES】connect failed.", e);
54+
return null;
55+
}
56+
}
57+
58+
public static RestHighLevelClient newRestHighLevelClient() {
59+
// 从配置中心读取环境变量
60+
String env = "test";
61+
return newRestHighLevelClient(env);
62+
}
63+
64+
public static RestHighLevelClient newRestHighLevelClient(String env) {
65+
String hosts = getDefaultEsAddress(env);
66+
return newRestHighLevelClient(toHttpHostList(hosts));
67+
}
68+
69+
public static RestHighLevelClient newRestHighLevelClient(HttpHost[] httpHosts) {
70+
RestClientBuilder builder = getRestClientBuilder(httpHosts);
71+
if (builder == null) {
72+
return null;
73+
}
74+
try {
75+
return new RestHighLevelClient(builder);
76+
} catch (Exception e) {
77+
log.error("【ES】connect failed.", e);
78+
return null;
79+
}
80+
}
81+
82+
public static ElasticsearchTemplate newElasticsearchTemplate() {
83+
// 从配置中心读取环境变量
84+
String env = "test";
85+
return newElasticsearchTemplate(env);
86+
}
87+
88+
public static ElasticsearchTemplate newElasticsearchTemplate(String env) {
89+
String hosts = getDefaultEsAddress(env);
90+
return newElasticsearchTemplate(toHttpHostList(hosts));
91+
}
92+
93+
public static ElasticsearchTemplate newElasticsearchTemplate(HttpHost[] httpHosts) {
94+
RestHighLevelClient client = newRestHighLevelClient(httpHosts);
95+
if (client == null) {
96+
return null;
97+
}
98+
return new ElasticsearchTemplate(client);
99+
}
100+
101+
public static ElasticsearchTemplate newElasticsearchTemplate(RestHighLevelClient client) {
102+
if (client == null) {
103+
return null;
104+
}
105+
return new ElasticsearchTemplate(client);
106+
}
107+
108+
public static RestClientBuilder getRestClientBuilder(HttpHost[] httpHosts) {
109+
if (ArrayUtil.isEmpty(httpHosts)) {
110+
log.error("【ES】connect failed. hosts are empty.");
111+
return null;
112+
}
113+
RestClientBuilder restClientBuilder = RestClient.builder(httpHosts);
114+
restClientBuilder.setRequestConfigCallback(builder -> {
115+
builder.setConnectTimeout(CONNECT_TIMEOUT_MILLIS);
116+
builder.setSocketTimeout(SOCKET_TIMEOUT_MILLIS);
117+
builder.setConnectionRequestTimeout(CONNECTION_REQUEST_TIMEOUT_MILLIS);
118+
return builder;
119+
});
120+
restClientBuilder.setHttpClientConfigCallback(builder -> {
121+
builder.setMaxConnTotal(MAX_CONN_TOTAL);
122+
builder.setMaxConnPerRoute(MAX_CONN_PER_ROUTE);
123+
return builder;
124+
});
125+
return restClientBuilder;
126+
}
127+
128+
private static HttpHost[] toHttpHostList(String hosts) {
129+
if (StrUtil.isBlank(hosts)) {
130+
return null;
131+
}
132+
List<String> strList = StrUtil.split(hosts, ",");
133+
List<HttpHost> list = strList.stream().map(str -> {
134+
List<String> params = StrUtil.split(str, ":");
135+
return new HttpHost(params.get(0), Integer.parseInt(params.get(1)), "http");
136+
}).collect(Collectors.toList());
137+
if (CollectionUtil.isEmpty(list)) {
138+
return new HttpHost[0];
139+
}
140+
return list.toArray(new HttpHost[0]);
141+
}
142+
143+
public static String getDefaultEsAddress() {
144+
// 从配置中心读取环境变量
145+
String env = "test";
146+
return getDefaultEsAddress(env);
147+
}
148+
149+
private static String getDefaultEsAddress(String env) {
150+
String defaultAddress;
151+
switch (env) {
152+
case "prd":
153+
defaultAddress = "127.0.0.1:9200,127.0.0.2:9200,127.0.0.3:9200";
154+
break;
155+
case "pre":
156+
defaultAddress = "127.0.0.1:9200";
157+
break;
158+
case "test":
159+
default:
160+
defaultAddress = "127.0.0.1:9200";
161+
break;
162+
}
163+
return defaultAddress;
164+
}
165+
166+
}

0 commit comments

Comments
 (0)