package com.atlassian.cache.hazelcast;
   
   import java.util.SortedMap;
   import java.util.concurrent.TimeUnit;
   
   import javax.annotation.Nonnull;
   import javax.annotation.Nullable;
   
   import com.atlassian.cache.CacheSettings;
  import com.atlassian.cache.CacheStatisticsKey;
  import com.atlassian.cache.ManagedCache;
  import com.atlassian.util.concurrent.Supplier;
  
  import com.google.common.collect.ImmutableSortedMap;
  import com.hazelcast.config.MapConfig;
  
  import static java.util.concurrent.TimeUnit.SECONDS;
  
  /**
   * Hazelcast implementation of common {@link ManagedCache} contract
   */
  public abstract class ManagedCacheSupport implements ManagedCache
  {
      private final MapConfig config;
      private final boolean flushable;
      private final String name;
  
      public ManagedCacheSupport(String name, MapConfig config, CacheSettings settings)
      {
          this.config = config;
          this.flushable = settings.getFlushable(true);
          this.name = name;
      }
  
      @Nullable
      @Override
      public Long currentExpireAfterAccessMillis()
      {
          long maxIdle = SECONDS.toMillis(config.getMaxIdleSeconds());
          return (maxIdle > 0 ? maxIdle : null);
      }
  
      @Nullable
      @Override
      public Long currentExpireAfterWriteMillis()
      {
          long timeToLive = SECONDS.toMillis(config.getTimeToLiveSeconds());
          return (timeToLive > 0 ? timeToLive : null);
      }
  
      @Nullable
      @Override
      public Integer currentMaxEntries()
      {
          int maxSize = config.getMaxSizeConfig().getSize();
          return maxSize > 0 ? maxSize : null;
      }
  
      @Override
      public boolean isReplicateViaCopy()
      {
          return true;
      }
  
      @Nonnull
      @Override
      public String getName()
      {
          return name;
      }
  
      @Override
      public boolean isFlushable()
      {
          return flushable;
      }
  
      @Override
      public boolean isLocal()
      {
          return false;
      }
  
      @Override
      public boolean isReplicateAsynchronously()
      {
          return false;
      }
  
      @Override
      public boolean updateExpireAfterAccess(long expireAfter, @Nonnull TimeUnit timeUnit)
      {
          config.setMaxIdleSeconds((int) timeUnit.toSeconds(expireAfter));
          return true;
      }
  
      @Override
      public boolean updateExpireAfterWrite(long expireAfter, @Nonnull TimeUnit timeUnit)
      {
         config.setMaxIdleSeconds((int) timeUnit.toSeconds(expireAfter));
         return true;
     }
 
     @Override
     public boolean updateMaxEntries(int newValue)
     {
         config.getMaxSizeConfig().setSize(newValue);
         return true;
     }
 
     @Nonnull
     @Override
     public SortedMap<CacheStatisticsKey, Supplier<Long>> getStatistics()
     {
         return ImmutableSortedMap.of();
     }
 }