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         try {
  52             Date expirationTime = calculateExpirationTime(timeout,unit);
  53             QueueManager queueManager = QueueManager.getInstance();
  54             boolean enqueueTask = queueManager.enqueueTask(new QueueMessage<M>(message,expirationTime));
  55             if(!enqueueTask){
  56                 throw new APPCException("failed to put message in queue");
  57             }
  58         } catch (Exception e) {
  59             logger.error("Error in putMessage method of ExecutionQueueServiceImpl" + e.getMessage());
  60             throw new APPCException(e);
  61         }
  62     }
  63
  64     @Override
  65     public void registerMessageExpirationListener(MessageExpirationListener listener) {
  66         QueueManager.getInstance().setListener(listener);
  67     }
  68
  69     private Date calculateExpirationTime(long timeToLive, TimeUnit unit) {
  70         Date expirationTime = null;
  71         if(timeToLive > 0){
  72             long currentTime = System.currentTimeMillis();
  73             Calendar cal = Calendar.getInstance();
  74             cal.setTimeInMillis(currentTime + unit.toMillis(timeToLive));
  75             expirationTime = cal.getTime();
  76         }
  77         return expirationTime;
  78     }
  79
  80 }