sidecar/fproxy/src/main/java/org/onap/aaf/fproxy/cache/InMemoryCredentialCache.java

   1 /**
   2  * ============LICENSE_START=======================================================
   3  * org.onap.aaf
   4  * ================================================================================
   5  * Copyright © 2018 European Software Marketing Ltd.
   6  * ================================================================================
   7  * Licensed under the Apache License, Version 2.0 (the "License");
   8  * you may not use this file except in compliance with the License.
   9  * You may obtain a copy of the License at
  10  *
  11  *       http://www.apache.org/licenses/LICENSE-2.0
  12  *
  13  * Unless required by applicable law or agreed to in writing, software
  14  * distributed under the License is distributed on an "AS IS" BASIS,
  15  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  16  * See the License for the specific language governing permissions and
  17  * limitations under the License.
  18  * ============LICENSE_END=========================================================
  19  */
  20 package org.onap.aaf.fproxy.cache;
  21
  22 import java.lang.ref.SoftReference;
  23 import java.util.Optional;
  24 import java.util.concurrent.ConcurrentHashMap;
  25 import java.util.concurrent.DelayQueue;
  26 import java.util.concurrent.Delayed;
  27 import java.util.concurrent.TimeUnit;
  28 import org.onap.aaf.fproxy.data.CredentialCacheData;
  29
  30 public class InMemoryCredentialCache implements CredentialCache {
  31
  32     private final ConcurrentHashMap<String, SoftReference<CredentialCacheData>> cache = new ConcurrentHashMap<>();
  33     private final DelayQueue<DelayedCacheObject> cleaningUpQueue = new DelayQueue<>();
  34
  35     public InMemoryCredentialCache() {
  36         Thread cleanerThread = new Thread(() -> {
  37             while (!Thread.currentThread().isInterrupted()) {
  38                 try {
  39                     DelayedCacheObject delayedCacheObject = cleaningUpQueue.take();
  40                     cache.remove(delayedCacheObject.getKey(), delayedCacheObject.getReference());
  41                 } catch (InterruptedException e) {
  42                     Thread.currentThread().interrupt();
  43                 }
  44             }
  45         });
  46         cleanerThread.setDaemon(true);
  47         cleanerThread.start();
  48     }
  49
  50     @Override
  51     public void add(String key, CredentialCacheData value, long periodInMillis) {
  52         if (key == null) {
  53             return;
  54         }
  55         if (value == null) {
  56             cache.remove(key);
  57         } else {
  58             long expiryTime = System.currentTimeMillis() + periodInMillis;
  59             SoftReference<CredentialCacheData> reference = new SoftReference<>(value);
  60             cache.put(key, reference);
  61             cleaningUpQueue.put(new DelayedCacheObject(key, reference, expiryTime));
  62         }
  63     }
  64
  65     @Override
  66     public void remove(String key) {
  67         cache.remove(key);
  68     }
  69
  70     @Override
  71     public CredentialCacheData get(String key) {
  72         return Optional.ofNullable(cache.get(key)).map(SoftReference::get).orElse(null);
  73     }
  74
  75     @Override
  76     public void clear() {
  77         cache.clear();
  78     }
  79
  80     @Override
  81     public long size() {
  82         return cache.size();
  83     }
  84
  85     private static class DelayedCacheObject implements Delayed {
  86
  87         private final String key;
  88         private final SoftReference<CredentialCacheData> reference;
  89         private final long expiryTime;
  90
  91         public DelayedCacheObject(String key, SoftReference<CredentialCacheData> reference, long expiryTime) {
  92             super();
  93             this.key = key;
  94             this.reference = reference;
  95             this.expiryTime = expiryTime;
  96         }
  97
  98         @Override
  99         public long getDelay(TimeUnit unit) {
 100             return unit.convert(expiryTime - System.currentTimeMillis(), TimeUnit.MILLISECONDS);
 101         }
 102
 103         @Override
 104         public int compareTo(Delayed o) {
 105             return Long.compare(expiryTime, ((DelayedCacheObject) o).expiryTime);
 106         }
 107
 108         public String getKey() {
 109             return key;
 110         }
 111
 112         public SoftReference<CredentialCacheData> getReference() {
 113             return reference;
 114         }
 115     }
 116 }