appc-dispatcher/appc-dispatcher-common/execution-queue-management-lib/src/main/java/org/onap/appc/executionqueue/impl/ExecutionQueueServiceImpl.java

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  * ONAP : APPC
   4  * ================================================================================
   5  * Copyright (C) 2017-2018 AT&T Intellectual Property. All rights reserved.
   6  * ================================================================================
   7  * Copyright (C) 2017 Amdocs
   8  * =============================================================================
   9  * Licensed under the Apache License, Version 2.0 (the "License");
  10  * you may not use this file except in compliance with the License.
  11  * You may obtain a copy of the License at
  12  *
  13  *      http://www.apache.org/licenses/LICENSE-2.0
  14  *
  15  * Unless required by applicable law or agreed to in writing, software
  16  * distributed under the License is distributed on an "AS IS" BASIS,
  17  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  18  * See the License for the specific language governing permissions and
  19  * limitations under the License.
  20  *
  21  * ============LICENSE_END=========================================================
  22  */
  23
  24 package org.onap.appc.executionqueue.impl;
  25
  26 import com.att.eelf.configuration.EELFLogger;
  27 import com.att.eelf.configuration.EELFManager;
  28 import org.onap.appc.exceptions.APPCException;
  29 import org.onap.appc.executionqueue.ExecutionQueueService;
  30 import org.onap.appc.executionqueue.MessageExpirationListener;
  31 import org.onap.appc.executionqueue.impl.object.QueueMessage;
  32
  33 import java.util.Calendar;
  34 import java.util.Date;
  35 import java.util.concurrent.TimeUnit;
  36
  37 public class ExecutionQueueServiceImpl<M extends Runnable> implements ExecutionQueueService<M> {
  38
  39     private final EELFLogger logger = EELFManager.getInstance().getLogger(ExecutionQueueServiceImpl.class);
  40
  41     private QueueManager queueManager;
  42
  43     public ExecutionQueueServiceImpl(){
  44
  45     }
  46
  47     @Override
  48     public void putMessage(M message) throws APPCException {
  49          this.putMessage(message,-1,null);
  50     }
  51
  52     /**
  53      * Injected by blueprint
  54      * @param queueManager
  55      */
  56     public void setQueueManager(QueueManager queueManager) {
  57         this.queueManager = queueManager;
  58     }
  59
  60     @Override
  61     public void putMessage(M message, long timeout, TimeUnit unit) throws APPCException{
  62         QueueMessage queueMessage;
  63
  64         try {
  65             Date expirationTime = calculateExpirationTime(timeout,unit);
  66             queueMessage = new QueueMessage(message,expirationTime);
  67             boolean enqueueTask = queueManager.enqueueTask(queueMessage);
  68             if(!enqueueTask){
  69                 throw new APPCException("failed to put message in queue");
  70             }
  71         } catch (Exception e) {
  72             logger.error("Error in putMessage method of ExecutionQueueServiceImpl" + e.getMessage());
  73             throw new APPCException(e);
  74         }
  75     }
  76
  77     private Date calculateExpirationTime(long timeToLive, TimeUnit unit) {
  78         Date expirationTime = null;
  79         if(timeToLive > 0){
  80             long currentTime = System.currentTimeMillis();
  81             Calendar cal = Calendar.getInstance();
  82             cal.setTimeInMillis(currentTime + unit.toMillis(timeToLive));
  83             expirationTime = cal.getTime();
  84         }
  85         return expirationTime;
  86     }
  87
  88 }