2 * ============LICENSE_START=======================================================
3 * Copyright (C) 2020 Nordix Foundation.
4 * Modifications Copyright (C) 2020 AT&T Intellectual Property. All rights reserved.
5 * Modifications Copyright (C) 2020 Bell Canada. All rights reserved.
6 * ================================================================================
7 * Licensed under the Apache License, Version 2.0 (the "License");
8 * you may not use this file except in compliance with the License.
9 * You may obtain a copy of the License at
11 * http://www.apache.org/licenses/LICENSE-2.0
13 * Unless required by applicable law or agreed to in writing, software
14 * distributed under the License is distributed on an "AS IS" BASIS,
15 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
16 * See the License for the specific language governing permissions and
17 * limitations under the License.
19 * SPDX-License-Identifier: Apache-2.0
20 * ============LICENSE_END=========================================================
23 package org.onap.policy.simulators;
25 import com.google.protobuf.InvalidProtocolBufferException;
26 import com.google.protobuf.util.JsonFormat;
27 import io.grpc.Server;
28 import io.grpc.netty.NettyServerBuilder;
29 import io.grpc.stub.StreamObserver;
30 import java.io.IOException;
31 import java.net.InetSocketAddress;
32 import java.util.concurrent.TimeUnit;
33 import java.util.concurrent.atomic.AtomicInteger;
35 import org.apache.commons.lang3.StringUtils;
36 import org.onap.ccsdk.cds.controllerblueprints.common.api.ActionIdentifiers;
37 import org.onap.ccsdk.cds.controllerblueprints.processing.api.BluePrintProcessingServiceGrpc.BluePrintProcessingServiceImplBase;
38 import org.onap.ccsdk.cds.controllerblueprints.processing.api.ExecutionServiceInput;
39 import org.onap.ccsdk.cds.controllerblueprints.processing.api.ExecutionServiceOutput;
40 import org.onap.ccsdk.cds.controllerblueprints.processing.api.ExecutionServiceOutput.Builder;
41 import org.onap.policy.common.utils.resources.ResourceUtils;
43 public class CdsSimulator {
45 private final int port;
47 private final Server server;
49 private final String resourceLocation;
51 private AtomicInteger countOfEvents = new AtomicInteger(1);
54 * Constructs the object, but does not start it.
56 * @param host host name of the server
57 * @param port port of the server
59 public CdsSimulator(String host, int port) {
60 this(host, port, "org/onap/policy/simulators/cds/", 0, 0);
64 * Constructs the object, but does not start it.
66 * @param host host name of the server
67 * @param port port of the server
68 * @param countOfSuccesfulEvents number of successive successful events
69 * @param requestedResponseDelayMs time for the request to be processed
71 public CdsSimulator(String host, int port, String resourceLocation, int countOfSuccesfulEvents,
72 long requestedResponseDelayMs) {
74 this.resourceLocation = resourceLocation;
76 BluePrintProcessingServiceImplBase testCdsBlueprintServerImpl = new BluePrintProcessingServiceImplBase() {
79 public StreamObserver<ExecutionServiceInput> process(
80 final StreamObserver<ExecutionServiceOutput> responseObserver) {
82 return new StreamObserver<ExecutionServiceInput>() {
85 public void onNext(final ExecutionServiceInput executionServiceInput) {
87 String responseString = getResponseString(executionServiceInput, countOfSuccesfulEvents);
88 Builder builder = ExecutionServiceOutput.newBuilder();
89 JsonFormat.parser().ignoringUnknownFields().merge(responseString, builder);
90 TimeUnit.MILLISECONDS.sleep(requestedResponseDelayMs);
91 responseObserver.onNext(builder.build());
92 } catch (InvalidProtocolBufferException e) {
93 throw new SimulatorRuntimeException("Cannot convert ExecutionServiceOutput output", e);
94 } catch (IOException e) {
95 throw new SimulatorRuntimeException("Cannot read ExecutionServiceOutput from file", e);
96 } catch (InterruptedException e) {
97 throw new SimulatorRuntimeException("Execution Interrupted", e);
102 public void onError(final Throwable throwable) {
103 responseObserver.onError(throwable);
107 public void onCompleted() {
108 responseObserver.onCompleted();
114 server = NettyServerBuilder.forAddress(new InetSocketAddress(host, port)).addService(testCdsBlueprintServerImpl)
118 public void start() throws IOException {
127 * Constructs the ResponseString on the basis of request.
129 * @param executionServiceInput service input
130 * @param countOfSuccesfulEvents number of successive successful events
131 * @return responseString
133 public String getResponseString(ExecutionServiceInput executionServiceInput, int countOfSuccesfulEvents) {
134 String resourceName = "DefaultResponseEvent";
135 if (!StringUtils.isBlank(executionServiceInput.getActionIdentifiers().getActionName())) {
136 ActionIdentifiers actionIdentifiers = executionServiceInput.getActionIdentifiers();
137 resourceName = actionIdentifiers.getBlueprintName() + "-" + actionIdentifiers.getActionName();
139 if (countOfSuccesfulEvents > 0 && countOfEvents.getAndIncrement() % countOfSuccesfulEvents == 0) {
140 // generating the failure response
141 resourceName = resourceName + "-error.json";
143 resourceName = resourceName + ".json";
145 String responseString = ResourceUtils.getResourceAsString(resourceLocation + resourceName);
146 if (responseString == null) {
147 responseString = ResourceUtils.getResourceAsString(resourceLocation
148 + "DefaultResponseEvent.json");
150 return responseString;