path: root/mso-catalog-db/src/main/java/org/onap/so/db/catalog/client/CatalogDbClient.java

/*-
 * ============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.db.catalog.client;

import org.onap.so.db.catalog.beans.BuildingBlockDetail;
import org.onap.so.db.catalog.beans.CollectionNetworkResourceCustomization;
import org.onap.so.db.catalog.beans.CollectionResourceInstanceGroupCustomization;
import org.onap.so.db.catalog.beans.InstanceGroup;
import org.onap.so.db.catalog.beans.NetworkCollectionResourceCustomization;
import org.onap.so.db.catalog.beans.OrchestrationAction;
import org.onap.so.db.catalog.beans.OrchestrationStatus;
import org.onap.so.db.catalog.beans.OrchestrationStatusStateTransitionDirective;
import org.onap.so.db.catalog.beans.ResourceType;
import org.onap.so.db.catalog.beans.Service;
import org.onap.so.db.catalog.beans.VfModuleCustomization;
import org.onap.so.db.catalog.beans.VnfcInstanceGroupCustomization;
import org.onap.so.db.catalog.beans.CloudSite;
import org.onap.so.db.catalog.beans.CloudIdentity;
import org.onap.so.db.catalog.beans.CloudifyManager;
import org.onap.so.db.catalog.beans.ServiceRecipe;
import org.onap.so.db.catalog.beans.macro.NorthBoundRequest;
import org.onap.so.db.catalog.beans.macro.OrchestrationFlow;
import org.onap.so.db.catalog.beans.macro.RainyDayHandlerStatus;
import org.onap.so.logging.jaxrs.filter.jersey.SpringClientFilter;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.http.client.BufferingClientHttpRequestFactory;
import org.springframework.http.client.ClientHttpRequestFactory;
import org.springframework.http.client.HttpComponentsClientHttpRequestFactory;
import org.springframework.stereotype.Component;
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.ArrayList;
import java.util.Iterator;
import java.util.List;

@Component("CatalogDbClient")
public class CatalogDbClient {

	private static final String SERVICE_RECIPE_SEARCH = "/serviceRecipe/search";
	private static final String SERVICE_MODEL_UUID = "SERVICE_MODEL_UUID";
	private static final String ACTION = "ACTION";
	private static final String MODEL_NAME = "MODEL_NAME";
	private static final String SERVICE_SEARCH = "/service/search";
	private static final String MODEL_VERSION = "MODEL_VERSION";
	private static final String MODEL_INVARIANT_UUID = "MODEL_INVARIANT_UUID";
	private String findFirstByModelNameURI = "/findFirstByModelNameOrderByModelVersionDesc";
	private String findFirstByServiceModelUUIDAndActionURI = "/findFirstByServiceModelUUIDAndAction";
	private String findByModelVersionAndModelInvariantUUIDURI = "/findByModelVersionAndModelInvariantUUID";
	
	private Client<Service> serviceClient;

	private Client<VfModuleCustomization> vfModuleCustomizationClient;

	private Client<OrchestrationFlow> orchestrationClient;

	private Client<NorthBoundRequest> northBoundRequestClient;

	private Client<RainyDayHandlerStatus> rainyDayHandlerStatusClient;

	private Client<BuildingBlockDetail> buildingBlockDetailClient;

	private Client<OrchestrationStatusStateTransitionDirective> orchestrationStatusStateTransitionDirectiveClient;

	private Client<VnfcInstanceGroupCustomization> vnfcInstanceGroupCustomizationClient;

	private Client<CollectionResourceInstanceGroupCustomization> collectionResourceInstanceGroupCustomizationClient;

	private Client<InstanceGroup> instanceGroupClient;
	
	private Client<NetworkCollectionResourceCustomization> networkCollectionResourceCustomizationClient;
	
	private Client<CollectionNetworkResourceCustomization> collectionNetworkResourceCustomizationClient;

	private Client<ServiceRecipe> serviceRecipeClient;

	private Client<CloudSite> cloudSiteClient;

	private Client<CloudIdentity> cloudIdentityClient;

	private Client<CloudifyManager> cloudifyManagerClient;

	@Value("${mso.catalog.db.spring.endpoint}")
	private String endpoint;

	@Value("${mso.db.auth}")
	private String msoAdaptersAuth;

	@PostConstruct
	public void init(){
		findFirstByModelNameURI = endpoint + SERVICE_SEARCH + findFirstByModelNameURI;
		findByModelVersionAndModelInvariantUUIDURI = endpoint + SERVICE_SEARCH + findByModelVersionAndModelInvariantUUIDURI;
		findFirstByServiceModelUUIDAndActionURI = endpoint + SERVICE_RECIPE_SEARCH + findFirstByServiceModelUUIDAndActionURI; 
	}

	public CatalogDbClient() {
		ClientHttpRequestFactory factory = new BufferingClientHttpRequestFactory(new HttpComponentsClientHttpRequestFactory());
		
		ClientFactory clientFactory = Configuration.builder().setClientHttpRequestFactory(factory).setRestTemplateConfigurer(restTemplate -> {
			restTemplate.getInterceptors().add((new SpringClientFilter()));
			
			restTemplate.getInterceptors().add((request, body, execution) -> {

				request.getHeaders().add("Authorization", msoAdaptersAuth);
				return execution.execute(request, body);
			});
		}).build().buildClientFactory();
		serviceClient = clientFactory.create(Service.class);
		orchestrationClient = clientFactory.create(OrchestrationFlow.class);
		vfModuleCustomizationClient = clientFactory.create(VfModuleCustomization.class);
		northBoundRequestClient = clientFactory.create(NorthBoundRequest.class);
		rainyDayHandlerStatusClient = clientFactory.create(RainyDayHandlerStatus.class);
		buildingBlockDetailClient = clientFactory.create(BuildingBlockDetail.class);
		orchestrationStatusStateTransitionDirectiveClient = clientFactory
				.create(OrchestrationStatusStateTransitionDirective.class);
		vnfcInstanceGroupCustomizationClient = clientFactory.create(VnfcInstanceGroupCustomization.class);
		collectionResourceInstanceGroupCustomizationClient = clientFactory
				.create(CollectionResourceInstanceGroupCustomization.class);
		instanceGroupClient = clientFactory.create(InstanceGroup.class);
		networkCollectionResourceCustomizationClient = clientFactory.create(NetworkCollectionResourceCustomization.class);
		collectionNetworkResourceCustomizationClient = clientFactory.create(CollectionNetworkResourceCustomization.class);
		cloudSiteClient = clientFactory.create(CloudSite.class);
		cloudIdentityClient = clientFactory.create(CloudIdentity.class);
		cloudifyManagerClient = clientFactory.create(CloudifyManager.class);
		serviceRecipeClient = clientFactory.create(ServiceRecipe.class);
	}
	
	public NetworkCollectionResourceCustomization getNetworkCollectionResourceCustomizationByID(String modelCustomizationUUID) {
		NetworkCollectionResourceCustomization networkCollectionResourceCustomization = 
				this.getSingleNetworkCollectionResourceCustomization(UriBuilder.fromUri(endpoint + "/networkCollectionResourceCustomization/" + modelCustomizationUUID).build());
		if (networkCollectionResourceCustomization != null) {
			networkCollectionResourceCustomization.setModelCustomizationUUID(modelCustomizationUUID);
		}
		return networkCollectionResourceCustomization;
	}

	private NetworkCollectionResourceCustomization getSingleNetworkCollectionResourceCustomization(URI uri) {
		return networkCollectionResourceCustomizationClient.get(uri);
	}

	public Service getServiceByID(String modelUUID) {
		Service service = this.getSingleService(UriBuilder.fromUri(endpoint + "/service/" + modelUUID).build());
		if (service != null) {
			service.setModelUUID(modelUUID);
		}
		return service;
	}

	public BuildingBlockDetail getBuildingBlockDetail(String buildingBlockName) {
		BuildingBlockDetail buildingBlockDetail = buildingBlockDetailClient
				.get(UriBuilder.fromUri(endpoint + "/buildingBlockDetail/search/findOneByBuildingBlockName")
						.queryParam("buildingBlockName", buildingBlockName).build());
		if (buildingBlockDetail != null) {
			buildingBlockDetail.setBuildingBlockName(buildingBlockName);
		}
		return buildingBlockDetail;
	}

	public CollectionNetworkResourceCustomization getCollectionNetworkResourceCustomizationByID(String modelCustomizationUUID) {
		CollectionNetworkResourceCustomization collectionNetworkResourceCustomization = 
				this.getSingleCollectionNetworkResourceCustomization(
						UriBuilder.fromUri(endpoint + "/collectionNetworkResourceCustomization/" + modelCustomizationUUID).build());
		if (collectionNetworkResourceCustomization != null) {
			collectionNetworkResourceCustomization.setModelCustomizationUUID(modelCustomizationUUID);
		}
		return collectionNetworkResourceCustomization;
	}

	public InstanceGroup getInstanceGroupByModelUUID(String modelUUID) {
		InstanceGroup instanceGroup = this
				.getSingleInstanceGroup(UriBuilder.fromUri(endpoint + "/instanceGroup/" + modelUUID).build());
		if (instanceGroup != null) {
			instanceGroup.setModelUUID(modelUUID);
		}
		return instanceGroup;
	}

	public OrchestrationStatusStateTransitionDirective getOrchestrationStatusStateTransitionDirective(
			ResourceType resourceType, OrchestrationStatus orchestrationStatus, OrchestrationAction targetAction) {
		return orchestrationStatusStateTransitionDirectiveClient.get(UriBuilder
				.fromUri(
						endpoint + "/orchestrationStatusStateTransitionDirective/search/findOneByResourceTypeAndOrchestrationStatusAndTargetAction")
				.queryParam("resourceType", resourceType.name())
				.queryParam("orchestrationStatus", orchestrationStatus.name())
				.queryParam("targetAction", targetAction.name()).build());
	}

	public List<OrchestrationFlow> getOrchestrationFlowByAction(String action) {
		return this
				.getMultipleOrchestrationFlows(UriBuilder.fromUri(endpoint + "/orchestrationFlow/search/findByAction")
						.queryParam("COMPOSITE_ACTION", action).build());
	}

	public List<OrchestrationFlow> getAllOrchestrationFlows() {
		return this.getMultipleOrchestrationFlows(UriBuilder.fromUri(endpoint + "/orchestrationFlow/").build());
	}

	private List<OrchestrationFlow> getMultipleOrchestrationFlows(URI uri) {
		Iterable<OrchestrationFlow> orchIterator = orchestrationClient.getAll(uri);
		List<OrchestrationFlow> orchList = new ArrayList<>();
		Iterator<OrchestrationFlow> it = orchIterator.iterator();
		it.forEachRemaining(orchList::add);
		return orchList;
	}

	public List<VnfcInstanceGroupCustomization> getVnfcInstanceGroupsByVnfResourceCust(String modelCustomizationUUID) {
		return this.getMultipleVnfcInstanceGroupCustomizations(
				UriBuilder.fromUri(endpoint + "/vnfcInstanceGroupCustomization/search/findByModelCustomizationUUID")
						.queryParam("MODEL_CUSTOMIZATION_UUID", modelCustomizationUUID).build());
	}

	public List<CollectionResourceInstanceGroupCustomization> getCollectionResourceInstanceGroupCustomizationByModelCustUUID(
			String modelCustomizationUUID) {
		return this.getMultipleCollectionResourceInstanceGroupCustomizations(UriBuilder
				.fromUri(endpoint + "/collectionResourceInstanceGroupCustomization/search/findByModelCustomizationUUID")
				.queryParam("MODEL_CUSTOMIZATION_UUID", modelCustomizationUUID).build());
	}

	private List<CollectionResourceInstanceGroupCustomization> getMultipleCollectionResourceInstanceGroupCustomizations(
			URI uri) {
		Iterable<CollectionResourceInstanceGroupCustomization> collectionInstanceGroupCustIter = collectionResourceInstanceGroupCustomizationClient
				.getAll(uri);
		List<CollectionResourceInstanceGroupCustomization> collectionInstanceGroupCustList = new ArrayList<>();
		Iterator<CollectionResourceInstanceGroupCustomization> it = collectionInstanceGroupCustIter.iterator();
		it.forEachRemaining(collectionInstanceGroupCustList::add);
		return collectionInstanceGroupCustList;
	}

	private List<VnfcInstanceGroupCustomization> getMultipleVnfcInstanceGroupCustomizations(URI uri) {
		Iterable<VnfcInstanceGroupCustomization> vnfcIterator = vnfcInstanceGroupCustomizationClient.getAll(uri);
		List<VnfcInstanceGroupCustomization> vnfcList = new ArrayList<>();
		Iterator<VnfcInstanceGroupCustomization> it = vnfcIterator.iterator();
		it.forEachRemaining(vnfcList::add);
		return vnfcList;
	}

	public VfModuleCustomization getVfModuleCustomizationByModelCuztomizationUUID(String modelCustomizationUUID) {
		VfModuleCustomization vfModuleCust = this.getSingleVfModuleCustomization(
				UriBuilder.fromUri(endpoint + "/vfModuleCustomization/" + modelCustomizationUUID).build());
		if (vfModuleCust != null) {
			vfModuleCust.setModelCustomizationUUID(modelCustomizationUUID);
		}
		return vfModuleCust;
	}

	public NorthBoundRequest getNorthBoundRequestByActionAndIsALaCarteAndRequestScope(String requestAction,
			String resourceName, boolean aLaCarte) {
		return this.getSingleNorthBoundRequest(UriBuilder
				.fromUri(endpoint + "/northbound_request_ref_lookup/search/findOneByActionAndRequestScopeAndIsAlacarte")
				.queryParam("ACTION", requestAction).queryParam("REQUEST_SCOPE", resourceName)
				.queryParam("IS_ALACARTE", aLaCarte).build());
	}

	public RainyDayHandlerStatus getRainyDayHandlerStatusByFlowNameAndServiceTypeAndVnfTypeAndErrorCodeAndWorkStep(
			String flowName, String serviceType, String vnfType, String errorCode, String workStep) {
		return this.getSingleRainyDayHandlerStatus(UriBuilder
				.fromUri(
						endpoint + "/rainy_day_handler_macro/search/findOneByFlowNameAndServiceTypeAndVnfTypeAndErrorCodeAndWorkStep")
				.queryParam("FLOW_NAME", flowName).queryParam("SERVICE_TYPE", serviceType)
				.queryParam("VNF_TYPE", vnfType).queryParam("ERROR_CODE", errorCode).queryParam("WORK_STEP", workStep)
				.build());
	}
	
	public  ServiceRecipe getFirstByServiceModelUUIDAndAction(String modelUUID, String action){
		return this.getSingleServiceRecipe(UriBuilder.fromUri(findFirstByServiceModelUUIDAndActionURI)
				.queryParam(SERVICE_MODEL_UUID,modelUUID)
				.queryParam(ACTION,action)
				.build());
	}
	
	public Service getFirstByModelNameOrderByModelVersionDesc(String modelName){
		return this.getSingleService(UriBuilder.fromUri(findFirstByModelNameURI)
				.queryParam(MODEL_NAME,modelName)
				.build());
	}
	
	
	private CollectionNetworkResourceCustomization getSingleCollectionNetworkResourceCustomization(URI uri) {
		return collectionNetworkResourceCustomizationClient.get(uri);
	}

	public CloudifyManager getCloudifyManager(String id) {
		return this.getSingleCloudifyManager(UriBuilder.fromUri(endpoint+"/cloudifyManager/"+id).build());
	}
	
	public CloudSite getCloudSite(String id){
		return this.getSinglCloudSite(UriBuilder.fromUri(endpoint+"/cloudSite/"+id).build());
	}
	
	public CloudIdentity getCloudIdentity(String id){
		return this.getSingleCloudIdentity(UriBuilder.fromUri(endpoint+"/cloudIdentity/"+id).build());
	}
	
	public CloudSite getCloudSiteByClliAndAicVersion (String clli, String cloudVersion){
		return this.getSinglCloudSite(UriBuilder.fromUri(endpoint+"/cloudSite/search/findByClliAndCloudVersion")
		.queryParam("CLLI",clli).queryParam("CLOUD_VERSION",cloudVersion)
		.build());
	}

	private InstanceGroup getSingleInstanceGroup(URI uri) {
		return instanceGroupClient.get(uri);
	}

	private Service getSingleService(URI uri) {
		return serviceClient.get(uri);
	}

	private VfModuleCustomization getSingleVfModuleCustomization(URI uri) {
		return vfModuleCustomizationClient.get(uri);
	}

	private NorthBoundRequest getSingleNorthBoundRequest(URI uri) {
		return northBoundRequestClient.get(uri);
	}

	private RainyDayHandlerStatus getSingleRainyDayHandlerStatus(URI uri) {
		return rainyDayHandlerStatusClient.get(uri);
	}
	
	private ServiceRecipe getSingleServiceRecipe(URI uri){
		return serviceRecipeClient.get(uri);
	}

	protected CloudSite getSinglCloudSite(URI uri) {
		return cloudSiteClient.get(uri);
	}

	protected CloudIdentity getSingleCloudIdentity(URI uri) {
		return cloudIdentityClient.get(uri);
	}

	protected CloudifyManager getSingleCloudifyManager(URI uri) {
		return cloudifyManagerClient.get(uri);
	}

	public Service getServiceByModelVersionAndModelInvariantUUID(String modelVersion, String modelInvariantUUID) {
		return this.getSingleService(
				UriBuilder.fromUri(findByModelVersionAndModelInvariantUUIDURI)
						.queryParam(MODEL_VERSION, modelVersion)
						.queryParam(MODEL_INVARIANT_UUID, modelInvariantUUID).build());
	}

	//USED FOR TEST ONLY
	public void setPortToEndpoint(String port) {
		endpoint = endpoint + port;
	}
	
	//USED FOR TEST ONLY
	public void removePortFromEndpoint() {
		endpoint = endpoint.substring(0, endpoint.lastIndexOf(':') + 1);
	}
}