1 files changed, 107 insertions, 0 deletions
diff --git a/prh-app-server/src/main/java/org/onap/dcaegen2/services/prh/tasks/AaiQueryTaskImpl.java b/prh-app-server/src/main/java/org/onap/dcaegen2/services/prh/tasks/AaiQueryTaskImpl.java
new file mode 100644
index 00000000..b30d199f
--- /dev/null
+++ b/prh-app-server/src/main/java/org/onap/dcaegen2/services/prh/tasks/AaiQueryTaskImpl.java
@@ -0,0 +1,107 @@
+/*
+ * ============LICENSE_START=======================================================
+ * PNF-REGISTRATION-HANDLER
+ * ================================================================================
+ * Copyright (C) 2018 NOKIA 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.dcaegen2.services.prh.tasks;
+
+import org.onap.dcaegen2.services.prh.model.*;
+import org.onap.dcaegen2.services.sdk.rest.services.aai.client.service.http.AaiHttpClient;
+import org.onap.dcaegen2.services.sdk.rest.services.model.AaiModel;
+import org.onap.dcaegen2.services.sdk.rest.services.model.AaiServiceInstanceQueryModel;
+import org.onap.dcaegen2.services.sdk.rest.services.model.ImmutableAaiServiceInstanceQueryModel;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.stereotype.Component;
+import reactor.core.publisher.Mono;
+
+import java.util.List;
+import java.util.Optional;
+import java.util.stream.Stream;
+
+@Component
+public class AaiQueryTaskImpl implements AaiQueryTask {
+    public final static String ACTIVE_STATUS = "Active";
+    public final static String RELATED_TO = "service-instance";
+    public final static String CUSTOMER = "customer.global-customer-id";
+    public final static String SERVICE_TYPE = "service-subscription.service-type";
+    public final static String SERVICE_INSTANCE_ID = "service-instance.service-instance-id";
+
+    private final AaiHttpClient<AaiModel, AaiPnfResultModel> getPnfModelClient;
+    private final AaiHttpClient<AaiServiceInstanceQueryModel, AaiServiceInstanceResultModel> getServiceClient;
+
+    @Autowired
+    public AaiQueryTaskImpl(
+            final AaiHttpClient<AaiModel, AaiPnfResultModel> getPnfModelClient,
+            final AaiHttpClient<AaiServiceInstanceQueryModel, AaiServiceInstanceResultModel> getServiceClient) {
+        this.getPnfModelClient = getPnfModelClient;
+        this.getServiceClient = getServiceClient;
+    }
+
+    @Override
+    public Mono<Boolean> execute(AaiModel aaiModel) {
+        return getPnfModelClient
+                .getAaiResponse(aaiModel)
+                .flatMap(this::checkIfPnfHasRelationToService)
+                .flatMap(getServiceClient::getAaiResponse)
+                .map(this::checkIfRelatedServiceInstanceIsActive)
+                .defaultIfEmpty(false);
+    }
+
+    private Mono<AaiServiceInstanceQueryModel> checkIfPnfHasRelationToService(final AaiPnfResultModel model) {
+        return Mono
+                .justOrEmpty(model.getRelationshipList())
+                .map(this::findRelatedTo)
+                .flatMap(Mono::justOrEmpty)
+                .map(RelationshipDict::getRelationshipData)
+                .flatMap(x -> {
+                    final Optional<String> customer = findValue(x, CUSTOMER);
+                    final Optional<String> serviceType = findValue(x, SERVICE_TYPE);
+                    final Optional<String> serviceInstanceId= findValue(x, SERVICE_INSTANCE_ID);
+
+                    return customer.isPresent() && serviceType.isPresent() && serviceInstanceId.isPresent()
+                            ? Mono.just(ImmutableAaiServiceInstanceQueryModel
+                            .builder()
+                            .customerId(customer.get())
+                            .serviceType(serviceType.get())
+                            .serviceInstanceId(serviceInstanceId.get())
+                            .build())
+                            : Mono.empty();
+                });
+    }
+
+    private Boolean checkIfRelatedServiceInstanceIsActive(final AaiServiceInstanceResultModel model) {
+        return ACTIVE_STATUS.equalsIgnoreCase(model.getOrchestrationStatus());
+    }
+
+    private Optional<RelationshipDict> findRelatedTo(final Relationship data) {
+        return Optional.ofNullable(data.getRelationship())
+                .map(Stream::of)
+                .orElseGet(Stream::empty)
+                .flatMap(List::stream)
+                .filter(x -> RELATED_TO.equals(x.getRelatedTo()))
+                .findFirst();
+    }
+
+    private Optional<String> findValue(final List<RelationshipData> data, final String key) {
+        return data
+                .stream()
+                .filter(y -> key.equals(y.getRelationshipKey()))
+                .findFirst()
+                .map(RelationshipData::getRelationshipValue);
+    }
+}