1 files changed, 143 insertions, 0 deletions
diff --git a/src/main/java/org/onap/ves/openapi/manager/service/ClientCallback.java b/src/main/java/org/onap/ves/openapi/manager/service/ClientCallback.java
new file mode 100644
index 0000000..720ab8c
--- /dev/null
+++ b/src/main/java/org/onap/ves/openapi/manager/service/ClientCallback.java
@@ -0,0 +1,143 @@
+/*
+ * ============LICENSE_START=======================================================
+ * VES-OPENAPI-MANAGER
+ * ================================================================================
+ * Copyright (C) 2021 Nokia. 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.ves.openapi.manager.service;
+
+import lombok.extern.log4j.Log4j2;
+import org.onap.sdc.api.consumer.INotificationCallback;
+import org.onap.sdc.api.notification.IArtifactInfo;
+import org.onap.sdc.api.notification.INotificationData;
+import org.onap.sdc.api.notification.IResourceInstance;
+import org.onap.ves.openapi.manager.model.Artifact;
+import org.onap.ves.openapi.manager.model.ArtifactValidationResult;
+import org.onap.ves.openapi.manager.service.notification.ArtifactsCollectorStatusSender;
+import org.onap.ves.openapi.manager.service.notification.FinalStatusSender;
+import org.onap.ves.openapi.manager.service.notification.ValidationStatusSender;
+import org.onap.ves.openapi.manager.service.validation.ArtifactsValidator;
+
+import java.util.Collection;
+import java.util.List;
+import java.util.Map;
+import java.util.stream.Collectors;
+
+@Log4j2
+public class ClientCallback implements INotificationCallback {
+
+    public static final String SEPARATOR = "===================================";
+    private final List<ArtifactsValidator> validators;
+    private final ValidationStatusSender validationStatusSender;
+    private final ArtifactsCollectorStatusSender artifactsCollectorStatusSender;
+    private final ArtifactsCollector artifactsCollector;
+    private final FinalStatusSender finalStatusSender;
+
+    /**
+     * Constructor of ClientCallback
+     * @param validators List of objects implementing ArtifactValidator interface
+     * @param artifactsCollector ArtifactsCollector object which downloads artifacts contents
+     * @param artifactsCollectorStatusSender ArtifactsCollectorStatusSender which sends download results back to SDC
+     * @param validationStatusSender ValidationStatusSender which sends validation results to SDC
+     * @param finalStatusSender FinalStatusSender which sends final status of VES OpenApi Manager workflow
+     */
+    public ClientCallback(List<ArtifactsValidator> validators, ArtifactsCollector artifactsCollector,
+                          ArtifactsCollectorStatusSender artifactsCollectorStatusSender,
+                          ValidationStatusSender validationStatusSender, FinalStatusSender finalStatusSender) {
+        this.validators = validators;
+        this.validationStatusSender = validationStatusSender;
+        this.artifactsCollectorStatusSender = artifactsCollectorStatusSender;
+        this.artifactsCollector = artifactsCollector;
+        this.finalStatusSender = finalStatusSender;
+    }
+
+    /**
+     * Callback method of distribution listener. Each time a service distribution takes place it's executed.
+     * @param service Distributed service information.
+     */
+    @Override
+    public void activateCallback(INotificationData service) {
+        logServiceInfo(service);
+        String distributionID = service.getDistributionID();
+
+        List<Artifact> artifacts = downloadArtifacts(service);
+        List<ArtifactValidationResult> validationResults = validate(distributionID, artifacts);
+        sendFinalStatus(distributionID, validationResults);
+    }
+
+    private void logServiceInfo(INotificationData service) {
+        log.info(SEPARATOR);
+        log.info("Distributed service information");
+        log.info("Service UUID: {}", service.getServiceUUID());
+        log.info("Service name: {}", service.getServiceName());
+        List<IResourceInstance> resources = service.getResources();
+        log.info("Service resources:");
+        resources.forEach(resource -> {
+            log.info(" - Resource: {}", resource.getResourceName());
+            log.info("   Artifacts:");
+            resource.getArtifacts().forEach(artifact -> log.info("   - Name: {}", artifact.getArtifactName()));
+        });
+        log.info(SEPARATOR);
+    }
+
+    private List<Artifact> downloadArtifacts(INotificationData service) {
+        List<IArtifactInfo> artifactInfos = getArtifactInfos(service);
+        List<Artifact> artifacts = artifactsCollector.pullArtifacts(artifactInfos);
+        sendDownloadStatuses(service, artifactInfos, artifacts);
+        return artifacts;
+    }
+
+    private void sendDownloadStatuses(INotificationData service, List<IArtifactInfo> artifactInfos, List<Artifact> artifacts) {
+        Map<Boolean, List<Artifact>> artifactsDownloadResults = artifacts.stream()
+                .collect(Collectors.partitioningBy(this::isContentDownloaded));
+        artifactsDownloadResults.get(Boolean.TRUE).forEach(artifact ->
+                artifactsCollectorStatusSender.sendDownloadOk(service.getDistributionID(), artifactInfos));
+        artifactsDownloadResults.get(Boolean.FALSE).forEach(artifact ->
+                artifactsCollectorStatusSender.sendDownloadError(service.getDistributionID(), artifactInfos));
+    }
+
+    private boolean isContentDownloaded(Artifact artifact) {
+        return artifact.getContent() != null && artifact.getContent().length > 0;
+    }
+
+    private List<IArtifactInfo> getArtifactInfos(INotificationData service) {
+        return service.getResources().stream()
+                .map(IResourceInstance::getArtifacts)
+                .flatMap(Collection::stream)
+                .collect(Collectors.toList());
+    }
+
+    private List<ArtifactValidationResult> validate(String distributionID, List<Artifact> artifacts) {
+        return validators.stream()
+                .map(validator -> {
+                    List<ArtifactValidationResult> validationResults = validator.validate(artifacts);
+                    validationStatusSender.send(distributionID, validationResults);
+                    return validationResults;
+                })
+                .flatMap(Collection::stream)
+                .collect(Collectors.toList());
+    }
+
+    private void sendFinalStatus(String distributionID, List<ArtifactValidationResult> validationResults) {
+        boolean areAllArtifactsValid = validationResults.stream().allMatch(ArtifactValidationResult::isValid);
+        if (areAllArtifactsValid) {
+            finalStatusSender.sendFinalStatusOk(distributionID);
+        } else {
+            finalStatusSender.sendFinalStatusError(distributionID);
+        }
+    }
+}