| /* |
| * Licensed to the Apache Software Foundation (ASF) under one or more |
| * contributor license agreements. See the NOTICE file distributed with |
| * this work for additional information regarding copyright ownership. |
| * The ASF licenses this file to You under the Apache License, Version 2.0 |
| * (the "License"); you may not use this file except in compliance with |
| * the License. You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| package org.apache.camel.component.elasticsearch; |
| |
| import java.io.IOException; |
| import java.util.Map; |
| |
| import org.apache.camel.component.extension.verifier.DefaultComponentVerifierExtension; |
| import org.apache.camel.component.extension.verifier.ResultBuilder; |
| import org.apache.camel.component.extension.verifier.ResultErrorBuilder; |
| import org.apache.camel.component.extension.verifier.ResultErrorHelper; |
| import org.apache.http.HttpHost; |
| import org.elasticsearch.client.RequestOptions; |
| import org.elasticsearch.client.RestClient; |
| import org.elasticsearch.client.RestClientBuilder; |
| import org.elasticsearch.client.RestHighLevelClient; |
| |
| public class ElasticsearchRestComponentVerifierExtension extends DefaultComponentVerifierExtension { |
| |
| public ElasticsearchRestComponentVerifierExtension() { |
| this("elasticsearch-rest"); |
| } |
| |
| public ElasticsearchRestComponentVerifierExtension(String scheme) { |
| super(scheme); |
| } |
| |
| // ********************************* |
| // Parameters validation |
| // ********************************* |
| |
| @Override |
| protected Result verifyParameters(Map<String, Object> parameters) { |
| |
| ResultBuilder builder = ResultBuilder.withStatusAndScope(Result.Status.OK, Scope.PARAMETERS).error(ResultErrorHelper.requiresOption("clusterName", parameters)) |
| .error(ResultErrorHelper.requiresOption("hostAddresses", parameters)); |
| // Validate using the catalog |
| |
| super.verifyParametersAgainstCatalog(builder, parameters); |
| |
| return builder.build(); |
| } |
| |
| // ********************************* |
| // Connectivity validation |
| // ********************************* |
| |
| @Override |
| protected Result verifyConnectivity(Map<String, Object> parameters) { |
| ResultBuilder builder = ResultBuilder.withStatusAndScope(Result.Status.OK, Scope.CONNECTIVITY); |
| |
| try { |
| ElasticsearchConfiguration configuration = setProperties(new ElasticsearchConfiguration(), parameters); |
| RestClientBuilder clientBuilder = RestClient.builder(configuration.getHostAddressesList().toArray(new HttpHost[0])); |
| RestHighLevelClient restHighLevelClient = new RestHighLevelClient(clientBuilder); |
| restHighLevelClient.ping(RequestOptions.DEFAULT); |
| } catch (IOException e) { |
| ResultErrorBuilder errorBuilder = ResultErrorBuilder.withCodeAndDescription(VerificationError.StandardCode.AUTHENTICATION, e.getMessage()) |
| .detail("elasticsearch_rest_exception_message", e.getMessage()).detail(VerificationError.ExceptionAttribute.EXCEPTION_CLASS, e.getClass().getName()) |
| .detail(VerificationError.ExceptionAttribute.EXCEPTION_INSTANCE, e); |
| |
| builder.error(errorBuilder.build()); |
| } catch (Exception e) { |
| builder.error(ResultErrorBuilder.withException(e).build()); |
| } |
| return builder.build(); |
| } |
| } |