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

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