examples/examples-onap-vcpe/src/test/java/org/onap/policy/apex/domains/onap/vcpe/AppcResponseCreator.java

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  *  Copyright (C) 2018 Ericsson. All rights reserved.
   4  *  Modifications Copyright (C) 2019-2020 Nordix Foundation.
   5  * ================================================================================
   6  * Licensed under the Apache License, Version 2.0 (the "License");
   7  * you may not use this file except in compliance with the License.
   8  * You may obtain a copy of the License at
   9  *
  10  *      http://www.apache.org/licenses/LICENSE-2.0
  11  *
  12  * Unless required by applicable law or agreed to in writing, software
  13  * distributed under the License is distributed on an "AS IS" BASIS,
  14  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  15  * See the License for the specific language governing permissions and
  16  * limitations under the License.
  17  *
  18  * SPDX-License-Identifier: Apache-2.0
  19  * ============LICENSE_END=========================================================
  20  */
  21
  22 package org.onap.policy.apex.domains.onap.vcpe;
  23
  24 import com.google.gson.Gson;
  25 import com.google.gson.GsonBuilder;
  26 import java.time.Instant;
  27 import java.util.Timer;
  28 import java.util.TimerTask;
  29 import java.util.concurrent.BlockingQueue;
  30 import org.onap.policy.appclcm.AppcLcmBody;
  31 import org.onap.policy.appclcm.AppcLcmDmaapWrapper;
  32 import org.onap.policy.appclcm.AppcLcmInput;
  33 import org.onap.policy.appclcm.AppcLcmOutput;
  34 import org.onap.policy.controlloop.util.Serialization;
  35
  36 /**
  37  * Respond to an APPC request with a given delay.
  38  */
  39 public class AppcResponseCreator {
  40     // The request from APPC
  41     private final String jsonRequestString;
  42
  43     // The queue for APPC responses
  44     private final BlockingQueue<String> appcResponseQueue;
  45
  46     // The timer task for response generation
  47     private final Timer appcTimer;
  48
  49     private static final Gson gson = new GsonBuilder()
  50             .registerTypeAdapter(Instant.class, new Serialization.GsonInstantAdapter()).setPrettyPrinting().create();
  51
  52     private static Integer nextResponseCode = 400;
  53
  54     /**
  55      * Respond to the given APPC request after the given amount of milliseconds.
  56      *
  57      * @param appcResponseQueue the queue into which to put the APPC response
  58      * @param jsonRequestString the request JSON string
  59      * @param milliSecondsToWait the number of milliseconds to wait
  60      */
  61     public AppcResponseCreator(BlockingQueue<String> appcResponseQueue, String jsonRequestString,
  62             long milliSecondsToWait) {
  63         this.jsonRequestString = jsonRequestString;
  64         this.appcResponseQueue = appcResponseQueue;
  65
  66         appcTimer = new Timer();
  67         appcTimer.schedule(new AppcTimerTask(), milliSecondsToWait);
  68     }
  69
  70     private class AppcTimerTask extends TimerTask {
  71         /**
  72          * {@inheritDoc}.
  73          */
  74         @Override
  75         public void run() {
  76
  77             AppcLcmDmaapWrapper requestWrapper = null;
  78             requestWrapper = gson.fromJson(jsonRequestString, AppcLcmDmaapWrapper.class);
  79
  80             AppcLcmInput request = requestWrapper.getBody().getInput();
  81
  82             AppcLcmOutput response = new AppcLcmOutput(request);
  83
  84             response.getStatus().setCode(nextResponseCode);
  85             if (nextResponseCode == 400) {
  86                 response.getStatus().setMessage("Restart Successful");
  87                 nextResponseCode = 100;
  88             } else if (nextResponseCode == 100) {
  89                 response.getStatus().setMessage("Restart Request Accepted");
  90                 nextResponseCode = 200;
  91             } else {
  92                 response.getStatus().setMessage("Error in Restart Operation");
  93                 nextResponseCode = 400;
  94             }
  95
  96             response.setPayload("");
  97
  98             AppcLcmDmaapWrapper responseWrapper = new AppcLcmDmaapWrapper();
  99             responseWrapper.setBody(new AppcLcmBody());
 100             responseWrapper.getBody().setOutput(response);
 101
 102             responseWrapper.setVersion(requestWrapper.getVersion());
 103             responseWrapper.setRpcName(requestWrapper.getRpcName());
 104             responseWrapper.setCorrelationId(requestWrapper.getCorrelationId());
 105             responseWrapper.setType(requestWrapper.getType());
 106
 107             appcResponseQueue.add(gson.toJson(responseWrapper));
 108         }
 109     }
 110 }