1 files changed, 272 insertions, 0 deletions
diff --git a/mso-api-handlers/mso-requests-db/src/main/java/org/onap/so/requestsdb/client/RequestsDbClient.java b/mso-api-handlers/mso-requests-db/src/main/java/org/onap/so/requestsdb/client/RequestsDbClient.java
new file mode 100644
index 0000000000..8e291aa15b
--- /dev/null
+++ b/mso-api-handlers/mso-requests-db/src/main/java/org/onap/so/requestsdb/client/RequestsDbClient.java
@@ -0,0 +1,272 @@
+/*-
+ * ============LICENSE_START=======================================================
+ * ONAP - SO
+ * ================================================================================
+ * Copyright (C) 2017 AT&T Intellectual Property. All rights reserved.
+ * ================================================================================
+ * Licensed 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.
+ * ============LICENSE_END=========================================================
+ */
+
+package org.onap.so.requestsdb.client;
+
+import org.apache.http.HttpStatus;
+import org.onap.so.db.request.beans.InfraActiveRequests;
+import org.onap.so.db.request.beans.OperationalEnvDistributionStatus;
+import org.onap.so.db.request.beans.OperationStatus;
+import org.onap.so.db.request.beans.OperationalEnvServiceModelStatus;
+import org.onap.so.db.request.beans.WatchdogServiceModVerIdLookup;
+import org.onap.so.db.request.beans.ArchivedInfraRequests;
+import org.onap.so.db.request.beans.ResourceOperationStatus;
+import org.onap.so.db.request.beans.WatchdogComponentDistributionStatus;
+import org.onap.so.db.request.beans.WatchdogDistributionStatus;
+import org.onap.so.db.request.beans.SiteStatus;
+import org.onap.so.db.request.data.controller.InstanceNameDuplicateCheckRequest;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.beans.factory.annotation.Value;
+import org.springframework.context.annotation.Bean;
+import org.springframework.core.ParameterizedTypeReference;
+import org.springframework.http.HttpEntity;
+import org.springframework.http.HttpHeaders;
+import org.springframework.http.HttpMethod;
+import org.springframework.http.client.HttpComponentsClientHttpRequestFactory;
+import org.springframework.stereotype.Component;
+import org.springframework.web.client.HttpClientErrorException;
+import org.springframework.web.client.RestTemplate;
+import uk.co.blackpepper.bowman.Client;
+import uk.co.blackpepper.bowman.ClientFactory;
+import uk.co.blackpepper.bowman.Configuration;
+
+import javax.annotation.PostConstruct;
+import javax.ws.rs.core.UriBuilder;
+import java.net.URI;
+import java.util.List;
+import java.util.Map;
+import java.util.ArrayList;
+import java.util.HashMap;
+import java.util.Iterator;
+
+@Component
+public class RequestsDbClient {
+
+	private static final String SERVICE_ID = "SERVICE_ID";
+	private static final String OPERATION_ID = "OPERATION_ID";
+	private static final String OPERATIONAL_ENVIRONMENT_ID = "OPERATIONAL_ENVIRONMENT_ID";
+	private static final String SERVICE_MODEL_VERSION_ID = "SERVICE_MODEL_VERSION_ID";
+	private static final String REQUEST_ID = "REQUEST_ID";
+
+	private final Client<InfraActiveRequests> infraActiveRequestClient;
+	private final Client<OperationStatus> operationStatusClient;
+	private final Client<OperationalEnvDistributionStatus> distributionStatusClient;
+	private final Client<OperationalEnvServiceModelStatus> serviceModelStatusClient;
+
+	@Value("${mso.adapters.requestDb.endpoint}")
+	private String endpoint;
+
+	@Value("${mso.adapters.requestDb.auth}")
+	private String msoAdaptersAuth;
+
+	private String getOrchestrationFilterURI = "/infraActiveRequests/getOrchestrationFiltersFromInfraActive/";
+	private static final String OPERATION_STATUS_SEARCH = "/operationStatus/search";
+	private static final String OPERATIONAL_ENV_SERVICE_MODEL_STATUS_SEARCH = "/operationalEnvServiceModelStatus/search";
+
+	private String checkVnfIdStatus = "/infraActiveRequests/checkVnfIdStatus/";
+
+	private String infraActiveRequestURI = "/infraActiveRequests/";
+
+	private String checkInstanceNameDuplicate = "/infraActiveRequests/checkInstanceNameDuplicate";
+	
+	private String findOneByServiceIdAndOperationIdURI = "/findOneByServiceIdAndOperationId";
+	
+	private  String operationalEnvDistributionStatusURI = "/operationalEnvDistributionStatus/";
+
+	private String cloudOrchestrationFiltersFromInfraActive = "/infraActiveRequests/getCloudOrchestrationFiltersFromInfraActive";
+
+	private String findOneByOperationalEnvIdAndServiceModelVersionIdURI = "/findOneByOperationalEnvIdAndServiceModelVersionId";
+	
+	private String findAllByOperationalEnvIdAndRequestIdURI = "/findAllByOperationalEnvIdAndRequestId";
+
+	private HttpHeaders headers;
+
+	@Autowired
+	private RestTemplate restTemplate;
+	@Autowired
+	ClassURLMapper classURLMapper;
+	
+
+	@PostConstruct
+	public void init() {
+		getOrchestrationFilterURI = endpoint + getOrchestrationFilterURI;
+		infraActiveRequestURI = endpoint + infraActiveRequestURI;
+		checkVnfIdStatus = endpoint + checkVnfIdStatus;
+		checkInstanceNameDuplicate = endpoint + checkInstanceNameDuplicate;
+		cloudOrchestrationFiltersFromInfraActive = endpoint + cloudOrchestrationFiltersFromInfraActive;
+		findOneByServiceIdAndOperationIdURI = endpoint + OPERATION_STATUS_SEARCH + findOneByServiceIdAndOperationIdURI;
+		operationalEnvDistributionStatusURI = endpoint + operationalEnvDistributionStatusURI;
+		findOneByOperationalEnvIdAndServiceModelVersionIdURI = endpoint + OPERATIONAL_ENV_SERVICE_MODEL_STATUS_SEARCH + findOneByOperationalEnvIdAndServiceModelVersionIdURI;
+		findAllByOperationalEnvIdAndRequestIdURI = endpoint + OPERATIONAL_ENV_SERVICE_MODEL_STATUS_SEARCH + findAllByOperationalEnvIdAndRequestIdURI;
+		headers = new HttpHeaders();
+		headers.set("Authorization", msoAdaptersAuth);
+	}
+
+	public RequestsDbClient() {
+		ClientFactory clientFactory = Configuration.builder().setRestTemplateConfigurer(restTemplate -> restTemplate.getInterceptors().add((request, body, execution) -> {
+			request.getHeaders().add("Authorization", msoAdaptersAuth);
+			return execution.execute(request, body);
+		})).build().buildClientFactory();
+		infraActiveRequestClient = clientFactory.create(InfraActiveRequests.class);
+		operationStatusClient = clientFactory.create(OperationStatus.class);
+		distributionStatusClient = clientFactory.create(OperationalEnvDistributionStatus.class);
+		serviceModelStatusClient = clientFactory.create(OperationalEnvServiceModelStatus.class);
+	}
+	public List<InfraActiveRequests> getCloudOrchestrationFiltersFromInfraActive(Map<String, String> orchestrationMap){
+		URI uri = getUri(cloudOrchestrationFiltersFromInfraActive);
+		HttpEntity<Map> entity = new HttpEntity<>(orchestrationMap, headers);
+		try{
+			return restTemplate.exchange(uri, HttpMethod.POST, entity, new ParameterizedTypeReference<List<InfraActiveRequests>>() {}).getBody();
+		}catch(HttpClientErrorException e){
+			if(HttpStatus.SC_NOT_FOUND == e.getStatusCode().value()){
+				return null;
+			}
+			throw e;
+		}
+	}
+	public InfraActiveRequests getInfraActiveRequestbyRequestId(String requestId) {
+		return this.getSingleInfraActiveRequests(this.getUri(endpoint + "/infraActiveRequests/" + requestId));
+	}
+
+	public List<InfraActiveRequests> getOrchestrationFiltersFromInfraActive(Map<String, List<String>> orchestrationMap) {
+		URI uri = getUri(getOrchestrationFilterURI);
+		HttpEntity<Map<String, List<String>>> entity = new HttpEntity<>(orchestrationMap, headers);
+		return restTemplate.exchange(uri, HttpMethod.POST, entity, new ParameterizedTypeReference<List<InfraActiveRequests>>() {}).getBody();
+	}
+
+	public InfraActiveRequests checkVnfIdStatus(String operationalEnvironmentId) {
+		URI uri = getUri(checkVnfIdStatus + operationalEnvironmentId);
+		return restTemplate.exchange(uri, HttpMethod.GET, HttpEntity.EMPTY, InfraActiveRequests.class).getBody();
+	}
+	public InfraActiveRequests checkInstanceNameDuplicate(HashMap<String, String> instanceIdMap, String instanceName, String requestScope) {
+		URI uri = getUri(checkInstanceNameDuplicate);
+		HttpEntity<InstanceNameDuplicateCheckRequest> entity = new HttpEntity<>(new InstanceNameDuplicateCheckRequest(instanceIdMap, instanceName, requestScope), headers);
+		try{
+			return restTemplate.exchange(uri, HttpMethod.POST, entity, InfraActiveRequests.class).getBody();
+		}catch(HttpClientErrorException e){
+			if(HttpStatus.SC_NOT_FOUND == e.getStatusCode().value()){
+				return null;
+			}
+			throw e;
+		}
+
+	}
+	
+	public  OperationStatus getOneByServiceIdAndOperationId(String serviceId, String operationId){
+		return this.getSingleOperationStatus(UriBuilder.fromUri(findOneByServiceIdAndOperationIdURI)
+				.queryParam(SERVICE_ID,serviceId)
+				.queryParam(OPERATION_ID,operationId)
+				.build());
+	}
+
+	public OperationalEnvServiceModelStatus findOneByOperationalEnvIdAndServiceModelVersionId(String operationalEnvironmentId, String serviceModelVersionId) {
+		return this.getSingleOperationalEnvServiceModelStatus(UriBuilder.fromUri(findOneByOperationalEnvIdAndServiceModelVersionIdURI)
+				.queryParam(OPERATIONAL_ENVIRONMENT_ID,operationalEnvironmentId)
+				.queryParam(SERVICE_MODEL_VERSION_ID,serviceModelVersionId)
+				.build());
+	}
+
+	public List<OperationalEnvServiceModelStatus> getAllByOperationalEnvIdAndRequestId(String operationalEnvironmentId, String requestId){
+		return this.getMultipleOperationalEnvServiceModelStatus(UriBuilder.fromUri(findAllByOperationalEnvIdAndRequestIdURI)
+				.queryParam(OPERATIONAL_ENVIRONMENT_ID,operationalEnvironmentId)
+				.queryParam(REQUEST_ID,requestId)
+				.build());
+	}
+	
+	public OperationalEnvDistributionStatus getDistributionStatusById(String distributionId){
+		return this.getSingleOperationalEnvDistributionStatus(UriBuilder.fromUri(operationalEnvDistributionStatusURI+distributionId).build());
+	}
+	
+	private OperationalEnvServiceModelStatus getSingleOperationalEnvServiceModelStatus(URI uri){
+		return serviceModelStatusClient.get(uri);
+	}
+
+	private List<OperationalEnvServiceModelStatus> getMultipleOperationalEnvServiceModelStatus(URI uri){
+		Iterable <OperationalEnvServiceModelStatus> iterable = serviceModelStatusClient.getAll(uri);
+		List<OperationalEnvServiceModelStatus> serviceModelStatuses = new ArrayList<>();
+		Iterator<OperationalEnvServiceModelStatus> statusIterator = iterable.iterator();
+		statusIterator.forEachRemaining(serviceModelStatuses::add);
+		return serviceModelStatuses;
+	}
+
+	public void save(InfraActiveRequests infraActiveRequests) {
+		URI uri = getUri(infraActiveRequestURI);
+		HttpEntity<InfraActiveRequests> entity = new HttpEntity<>(infraActiveRequests, headers);
+		restTemplate.postForLocation(uri, entity);
+	}
+	
+	public <T> void save(T object){
+		URI uri = getUri(endpoint+classURLMapper.getURI(object.getClass()));
+		HttpEntity<T> entity = new HttpEntity<>(object, headers);
+		restTemplate.postForLocation(uri, entity);
+	}
+	
+	private OperationalEnvDistributionStatus getSingleOperationalEnvDistributionStatus(URI uri){
+		return distributionStatusClient.get(uri);
+	}
+
+	private InfraActiveRequests getSingleInfraActiveRequests(URI uri) {
+		return infraActiveRequestClient.get(uri);
+	}
+
+	public void updateInfraActiveRequests(InfraActiveRequests request) {		
+		infraActiveRequestClient.put(request);
+	}
+	
+	private OperationStatus getSingleOperationStatus(URI uri){
+		return operationStatusClient.get(uri);
+	}
+
+	private URI getUri(String uri) {
+		return URI.create(uri);
+	}
+
+	@Bean
+	public RestTemplate restTemplate() {
+		return new RestTemplate( new HttpComponentsClientHttpRequestFactory());
+	}
+
+	@Component
+	static class ClassURLMapper {
+		private static final Map <Class,String> classURLMap = new HashMap<>();
+
+		ClassURLMapper() {
+			classURLMap.put(ArchivedInfraRequests.class,"/archivedInfraRequests/");
+			classURLMap.put(InfraActiveRequests.class,"/infraActiveRequests/");
+			classURLMap.put(OperationalEnvDistributionStatus.class,"/operationalEnvDistributionStatus/");
+			classURLMap.put(OperationalEnvServiceModelStatus.class,"/operationalEnvServiceModelStatus/");
+			classURLMap.put(OperationStatus.class,"/operationStatus/");
+			classURLMap.put(ResourceOperationStatus.class,"/resourceOperationStatus/");
+			classURLMap.put(SiteStatus.class,"/siteStatus/");
+			classURLMap.put(WatchdogComponentDistributionStatus.class,"/watchdogComponentDistributionStatus/");
+			classURLMap.put(WatchdogDistributionStatus.class,"/watchdogDistributionStatus/");
+			classURLMap.put(WatchdogServiceModVerIdLookup.class,"/watchdogServiceModVerIdLookup/");
+		}
+
+		  <T> String getURI(Class<T> className) {
+			  Class actualClass = classURLMap.keySet()
+					  .stream()
+					  .filter(requestdbClass -> requestdbClass.isAssignableFrom(className))
+					  .findFirst()
+					  .get();
+			  return classURLMap.get(actualClass);
+		}
+	}
+}