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.time.Instant;
  25 import java.util.Calendar;
  26 import java.util.Date;
  27 import java.util.concurrent.TimeUnit;
  28
  29 import org.openecomp.appc.exceptions.APPCException;
  30 import org.openecomp.appc.executionqueue.ExecutionQueueService;
  31 import org.openecomp.appc.executionqueue.MessageExpirationListener;
  32 import org.openecomp.appc.executionqueue.impl.object.QueueMessage;
  33
  34 import com.att.eelf.configuration.EELFLogger;
  35 import com.att.eelf.configuration.EELFManager;
  36
  37 public class ExecutionQueueServiceImpl<M extends Runnable> implements ExecutionQueueService<M> {
  38
  39     private static final EELFLogger logger =
  40             EELFManager.getInstance().getLogger(ExecutionQueueServiceImpl.class);
  41
  42     ExecutionQueueServiceImpl(){
  43
  44     }
  45
  46     @Override
  47     public void putMessage(M message) throws APPCException {
  48          this.putMessage(message,-1,null);
  49     }
  50
  51     @Override
  52     public void putMessage(M message, long timeout, TimeUnit unit) throws APPCException{
  53         try {
  54             Instant expirationTime = calculateExpirationTime(timeout,unit);
  55             QueueManager queueManager = QueueManager.getInstance();
  56             boolean enqueueTask = queueManager.enqueueTask(new QueueMessage<M>(message,expirationTime));
  57             if(!enqueueTask){
  58                 throw new APPCException("failed to put message in queue");
  59             }
  60         } catch (Exception e) {
  61             logger.error("Error in putMessage method of ExecutionQueueServiceImpl" + e.getMessage());
  62             throw new APPCException(e);
  63         }
  64     }
  65
  66     @Override
  67     public void registerMessageExpirationListener(MessageExpirationListener listener) {
  68         QueueManager.getInstance().setListener(listener);
  69     }
  70
  71     private Instant calculateExpirationTime(long timeToLive, TimeUnit unit) {
  72         if (timeToLive > 0 && unit != null) {
  73             // as of Java 8, there is no built-in conversion method from
  74             // TimeUnit to ChronoUnit; do it manually
  75             return Instant.now().plusMillis(unit.toMillis(timeToLive));
  76         } else {
  77             // never expires
  78             return Instant.MAX;
  79         }
  80     }
  81
  82 }