blob: af9a51dee39121147a0a3534735f196cc54de4f5 [file] [log] [blame]
/**
* 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.Service;
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.UserManagedCacheBuilder;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
public class EhcacheManager implements Service {
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 : () -> { }
);
}
@Override
public synchronized void start() throws Exception {
refCount.retain();
}
@Override
public synchronized void stop() throws Exception {
refCount.release();
userCaches.values().forEach(UserManagedCache::close);
}
public <K, V> Cache<K, V> getCache(String name, Class<K> keyType, Class<V> valueType) throws Exception {
Cache<K, V> cache = cacheManager.getCache(name, keyType, valueType);
if (cache == null && configuration != null && configuration.isCreateCacheIfNotExist()) {
CacheConfiguration<K, V> cacheConfiguration = 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 (cacheConfiguration != null) {
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);
cache = Cache.class.cast(userCaches.computeIfAbsent(
name,
key -> UserManagedCacheBuilder.newUserManagedCacheBuilder(keyType, valueType).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;
}
}