vnfs/TestVNF/netconfserver/src/main/java/com/ericsson/testvnf/server/requestqueue/RequestQueueHandler.java

   1 /*
   2  * ============LICENSE_START=======================================================
   3  *  Copyright (C) 2016-2018 Ericsson. All rights reserved.
   4  * ================================================================================
   5  * Licensed under the Apache License, Version 2.0 (the "License");
   6  * you may not use this file except in compliance with the License.
   7  * You may obtain a copy of the License at
   8  *
   9  *      http://www.apache.org/licenses/LICENSE-2.0
  10  *
  11  * Unless required by applicable law or agreed to in writing, software
  12  * distributed under the License is distributed on an "AS IS" BASIS,
  13  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  14  * See the License for the specific language governing permissions and
  15  * limitations under the License.
  16  *
  17  * SPDX-License-Identifier: Apache-2.0
  18  * ============LICENSE_END=========================================================
  19  */
  20
  21 package com.ericsson.testvnf.server.requestqueue;
  22
  23 import java.util.Queue;
  24 import java.util.concurrent.LinkedBlockingQueue;
  25 import org.apache.commons.logging.Log;
  26 import org.apache.commons.logging.LogFactory;
  27
  28 import com.ericsson.testvnf.server.models.NetconfMessage;
  29
  30 /**
  31  * RequestQueueHandler manages incoming netconf requests from clients
  32  *
  33  * @author Ajith Sreekumar
  34  *
  35  */
  36 public class RequestQueueHandler {
  37
  38         private static final Log log = LogFactory.getLog(RequestQueueHandler.class);
  39
  40         private volatile Queue<NetconfMessage> requestQueue;
  41
  42         public RequestQueueHandler() {
  43                 requestQueue = new LinkedBlockingQueue<>();
  44         }
  45
  46         // wait until an element is added to queue, once added return it
  47         public NetconfMessage waitAndGetMessageFromQueue() throws InterruptedException {
  48                 NetconfMessage netconfMessage = null;
  49                 synchronized (requestQueue) {
  50                         while (true) {
  51                                 log.info("Waiting for message to be added to queue..");
  52                                 if (!requestQueue.isEmpty()) {
  53                                         netconfMessage = requestQueue.element();
  54                                         requestQueue.remove(requestQueue.element());
  55                                 }
  56                                 if (netconfMessage != null) {
  57                                         log.info("Message received in queue is taken for processing.");
  58                                         break;
  59                                 }
  60                                 requestQueue.wait();
  61                         }
  62                 }
  63                 return netconfMessage;
  64         }
  65
  66         //add message into the queue
  67         public void addToQueue(NetconfMessage netconfMessage) {
  68                 synchronized (requestQueue) {
  69                         log.info("Received a new message in queue.");
  70                         requestQueue.add(netconfMessage);
  71                         requestQueue.notifyAll();
  72                 }
  73         }
  74
  75 }