content: /*
   * Licensed to the Apache Software Foundation (ASF) under one or more
   * contributor license agreements.  See the NOTICE file distributed with
   * this work for additional information regarding copyright ownership.
   * The ASF licenses this file to You under the Apache License, Version 2.0
   * (the "License"); you may not use this file except in compliance with
   * the License.  You may obtain a copy of the License at
   *
   *      http://www.apache.org/licenses/LICENSE-2.0
   *
   * Unless required by applicable law or agreed to in writing, software
   * distributed under the License is distributed on an "AS IS" BASIS,
   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
   * See the License for the specific language governing permissions and
   * limitations under the License.
   */
  package org.apache.camel.component.ehcache;
  
  import java.util.concurrent.ConcurrentHashMap;
  import java.util.concurrent.ConcurrentMap;
  
  import org.apache.camel.RuntimeCamelException;
  import org.apache.camel.support.service.ServiceSupport;
  import org.apache.camel.util.ObjectHelper;
  import org.apache.camel.util.ReferenceCount;
  import org.ehcache.Cache;
  import org.ehcache.CacheManager;
  import org.ehcache.UserManagedCache;
  import org.ehcache.config.CacheConfiguration;
  import org.ehcache.config.builders.CacheConfigurationBuilder;
  import org.ehcache.config.builders.UserManagedCacheBuilder;
  import org.ehcache.spi.service.ServiceConfiguration;
  import org.slf4j.Logger;
  import org.slf4j.LoggerFactory;
  
  public class EhcacheManager extends ServiceSupport {
      private static final Logger LOGGER = LoggerFactory.getLogger(EhcacheManager.class);
  
      private final EhcacheConfiguration configuration;
      private final CacheManager cacheManager;
      private final ConcurrentMap<String, UserManagedCache<?, ?>> userCaches;
      private final ReferenceCount refCount;
  
      public EhcacheManager(CacheManager cacheManager, boolean managed, EhcacheConfiguration configuration) {
          this.cacheManager = ObjectHelper.notNull(cacheManager, "cacheManager");
          this.userCaches = new ConcurrentHashMap<>();
          this.configuration = configuration;
          this.refCount = ReferenceCount.on(
                  managed ? cacheManager::init : () -> {
                  },
                  managed ? cacheManager::close : () -> {
                  });
      }
  
      protected void incRef() {
          refCount.retain();
      }
  
      protected void decRef() {
          refCount.release();
      }
  
      @Override
      protected void doShutdown() throws Exception {
          if (userCaches != null && !userCaches.isEmpty()) {
              for (UserManagedCache cache : userCaches.values()) {
                  try {
                      cache.close();
                  } catch (Exception e) {
                      // ignore
                  }
              }
              userCaches.clear();
          }
      }
  
      @SuppressWarnings("unchecked")
      public <K, V> Cache<K, V> getCache(String name, Class<K> keyType, Class<V> valueType) throws Exception {
          CacheConfiguration<K, V> cacheConfiguration = null;
          if (configuration != null) {
              if (configuration.hasConfiguration(name)) {
                  LOGGER.debug("Using custom cache configuration for cache {}", name);
                  cacheConfiguration = CacheConfiguration.class.cast(configuration.getConfigurations().get(name));
              } else if (configuration.hasConfiguration()) {
                  LOGGER.debug("Using global cache configuration for cache {}", name);
                  cacheConfiguration = CacheConfiguration.class.cast(configuration.getConfiguration());
              }
          }
          if (keyType == null && valueType == null) {
              if (cacheConfiguration != null) {
                  keyType = cacheConfiguration.getKeyType();
                  valueType = cacheConfiguration.getValueType();
              } else {
                  keyType = (Class<K>) Object.class;
                  valueType = (Class<V>) Object.class;
              }
          }
          Cache<K, V> cache = cacheManager.getCache(name, keyType, valueType);
          if (cache == null && configuration != null && configuration.isCreateCacheIfNotExist()) {
              if (cacheConfiguration != null) {
                  if (keyType != cacheConfiguration.getKeyType() || valueType != cacheConfiguration.getValueType()) {
                      LOGGER.info("Mismatch keyType / valueType configuration for cache {}", name);
                      CacheConfigurationBuilder builder = CacheConfigurationBuilder
                              .newCacheConfigurationBuilder(keyType, valueType, cacheConfiguration.getResourcePools())
                              .withClassLoader(cacheConfiguration.getClassLoader())
                              .withEvictionAdvisor(cacheConfiguration.getEvictionAdvisor())
                              .withExpiry(cacheConfiguration.getExpiryPolicy());
                      for (ServiceConfiguration<?, ?> serviceConfig : cacheConfiguration.getServiceConfigurations()) {
                          builder = builder.withService(serviceConfig);
                      }
                      cacheConfiguration = builder.build();
                  }
                  cache = cacheManager.createCache(name, cacheConfiguration);
              } else {
                  // If a cache configuration is not provided, create a User Managed
                  // Cache
                  LOGGER.debug("Using a UserManagedCache for cache {} as no configuration has been found", name);
                  Class<K> kt = keyType;
                  Class<V> vt = valueType;
                  cache = Cache.class.cast(userCaches.computeIfAbsent(
                          name,
                          key -> UserManagedCacheBuilder.newUserManagedCacheBuilder(kt, vt).build(true)));
              }
          }
  
          if (cache == null) {
              throw new RuntimeCamelException("Unable to retrieve the cache " + name + " from cache manager " + cacheManager);
          }
  
          return cache;
      }
  
      CacheManager getCacheManager() {
          return this.cacheManager;
      }
  
      ReferenceCount getReferenceCount() {
          return refCount;
      }
  }
  