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
|
/*
* SonarQube
* Copyright (C) 2009-2025 SonarSource SA
* mailto:info AT sonarsource DOT com
*
* This program is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public
* License as published by the Free Software Foundation; either
* version 3 of the License, or (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public License
* along with this program; if not, write to the Free Software Foundation,
* Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
*/
package org.sonarqube.ws.tester;
import java.io.IOException;
import java.net.InetAddress;
import okhttp3.MediaType;
import okhttp3.OkHttpClient;
import okhttp3.Request;
import okhttp3.RequestBody;
import okhttp3.Response;
import static org.assertj.core.api.Assertions.assertThat;
/**
* Helper to directly access Elasticsearch. It requires the HTTP port
* to be open.
*/
public class Elasticsearch {
private final int httpPort;
public Elasticsearch(int httpPort) {
this.httpPort = httpPort;
}
/**
* Forbid indexing requests on the specified index. Index becomes read-only.
*/
public void lockWrites(String index) throws IOException {
putIndexSetting(httpPort, index, "blocks.write", "true");
}
/**
* Enable indexing requests on the specified index.
* @see #lockWrites(String)
*/
public void unlockWrites(String index) throws IOException {
putIndexSetting(httpPort, index, "blocks.write", "false");
}
public void makeYellow() throws IOException {
putIndexSetting(httpPort, "issues", "number_of_replicas", "5");
}
public void makeGreen() throws IOException {
putIndexSetting(httpPort, "issues", "number_of_replicas", "0");
}
private static void putIndexSetting(int searchHttpPort, String index, String key, String value) throws IOException {
Request.Builder request = new Request.Builder()
.url(baseUrl(searchHttpPort) + index + "/_settings")
.put(RequestBody.create(MediaType.parse("application/json"), "{" +
" \"index\" : {" +
" \"" + key + "\" : \"" + value + "\"" +
" }" +
"}"));
OkHttpClient okClient = new OkHttpClient.Builder().build();
try (Response response = okClient.newCall(request.build()).execute()) {
assertThat(response.isSuccessful()).isTrue();
}
}
private static String baseUrl(int searchHttpPort) {
return "http://" + InetAddress.getLoopbackAddress().getHostAddress() + ":" + searchHttpPort + "/";
}
}
|
