import org.apache.http.HttpHost;
import org.apache.http.auth.AuthScope;
import org.apache.http.auth.UsernamePasswordCredentials;
import org.apache.http.client.CredentialsProvider;
import org.apache.http.impl.client.BasicCredentialsProvider;
import org.apache.http.impl.nio.client.HttpAsyncClientBuilder;
import org.apache.log4j.BasicConfigurator;
import org.apache.log4j.Logger;
import org.elasticsearch.ElasticsearchException;
import org.elasticsearch.action.DocWriteResponse;
import org.elasticsearch.action.index.IndexRequest;
import org.elasticsearch.action.index.IndexResponse;
import org.elasticsearch.action.get.GetRequest;
import org.elasticsearch.action.get.GetResponse;
import org.elasticsearch.client.RestClient;
import org.elasticsearch.client.RestClientBuilder;
import org.elasticsearch.client.RestHighLevelClient;
import org.elasticsearch.rest.RestStatus;
import java.util.Date;
import java.util.HashMap;
import java.util.Map;
public class test_es_sdk {
private static Logger logger = Logger.getLogger(test_es_sdk.class);
public static void main(String[]args){
BasicConfigurator.configure();
// Set verification information by entering the username and password
final CredentialsProvider credentialsProvider = new BasicCredentialsProvider();
credentialsProvider.setCredentials(AuthScope.ANY,
new UsernamePasswordCredentials("user", "passwd"));
// Initialize RestClient. For hostName and port, enter the private VIP address and port of the cluster respectively
RestClientBuilder builder = RestClient.builder(new HttpHost("xx.xx.xx.xx", 9200, "http"));
// Set authentication information
builder.setHttpClientConfigCallback(new RestClientBuilder.HttpClientConfigCallback() {
@Override
public HttpAsyncClientBuilder customizeHttpClient(HttpAsyncClientBuilder httpClientBuilder) {
return httpClientBuilder.setDefaultCredentialsProvider(credentialsProvider);
}
});
// Set the timeout period
builder.setMaxRetryTimeoutMillis(10000);
// Construct the High Level Client based on the Low Level Client
RestHighLevelClient client = new RestHighLevelClient(builder);
// Index the document
Map<String, Object> jsonMap = new HashMap<String, Object>();
jsonMap.put("user", "bellen");
jsonMap.put("name", new Date());
jsonMap.put("message", "trying out Elasticsearch");
IndexRequest indexRequest = new IndexRequest("posts", "doc", "1")
.source(jsonMap);
try {
// Get the response result
IndexResponse indexResponse = client.index(indexRequest);
String index = indexResponse.getIndex();
String type = indexResponse.getType();
String id = indexResponse.getId();
long version = indexResponse.getVersion();
if (indexResponse.getResult() == DocWriteResponse.Result.CREATED) {
logger.info("doc indexed, index: "+ index +", type:"+ type +",id:"+ id+",version:"+version);
} else if (indexResponse.getResult() == DocWriteResponse.Result.UPDATED) {
logger.info("doc updated, index: "+ index +", type:"+ type +",id:"+ id+",version:"+version);
}
}catch(ElasticsearchException e) {
if (e.status() == RestStatus.CONFLICT) {
logger.error("version conflict");
}
}catch(Exception e){
logger.error("execute index api failed, "+ e.toString());
}
// Query the document
GetRequest getRequest = new GetRequest(
"posts",
"doc",
"1");
try {
// Get the response result
GetResponse getResponse = client.get(getRequest);
String index = getResponse.getIndex();
String type = getResponse.getType();
String id = getResponse.getId();
if (getResponse.isExists()) {
long version = getResponse.getVersion();
String sourceAsString = getResponse.getSourceAsString();
logger.info("get doc, index: "+ index +", type:"+ type +",id:"+ id+",version:"+version +", source:"+ sourceAsString);
}
}catch (ElasticsearchException e) {
if (e.status() == RestStatus.NOT_FOUND) {
logger.warn("doc not found");
}
}
catch(Exception e){
logger.error("execute get api failed, "+ e.toString());
}
// Close the client
try {
client.close();
}catch (Exception e){
logger.error("close rest client exception:"+ e.toString());
}
}
}
Was this page helpful?