/**
* 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.cache;
import net.sf.ehcache.Cache;
import net.sf.ehcache.CacheManager;
import net.sf.ehcache.Ehcache;
import net.sf.ehcache.event.CacheEventListener;
import org.apache.camel.Component;
import org.apache.camel.Consumer;
import org.apache.camel.Exchange;
import org.apache.camel.Message;
import org.apache.camel.Processor;
import org.apache.camel.Producer;
import org.apache.camel.impl.DefaultEndpoint;
import org.apache.camel.impl.DefaultExchange;
import org.apache.camel.impl.DefaultMessage;
import org.apache.camel.util.ObjectHelper;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
public class CacheEndpoint extends DefaultEndpoint {
private static final transient Logger LOG = LoggerFactory.getLogger(CacheEndpoint.class);
private CacheConfiguration config;
private CacheManagerFactory cacheManagerFactory;
public CacheEndpoint() {
}
public CacheEndpoint(String endpointUri, Component component,
CacheConfiguration config, CacheManagerFactory cacheManagerFactory) {
super(endpointUri, component);
this.config = config;
this.cacheManagerFactory = cacheManagerFactory;
}
public Consumer createConsumer(Processor processor) throws Exception {
ObjectHelper.notNull(config, "config");
ObjectHelper.notNull(cacheManagerFactory, "cacheManagerFactory");
return new CacheConsumer(this, processor, config);
}
public Producer createProducer() throws Exception {
ObjectHelper.notNull(config, "config");
ObjectHelper.notNull(cacheManagerFactory, "cacheManagerFactory");
return new CacheProducer(this, config);
}
public boolean isSingleton() {
return true;
}
public CacheConfiguration getConfig() {
return config;
}
public void setConfig(CacheConfiguration config) {
this.config = config;
}
public CacheManagerFactory getCacheManagerFactory() {
return cacheManagerFactory;
}
public void setCacheManagerFactory(CacheManagerFactory cacheManagerFactory) {
this.cacheManagerFactory = cacheManagerFactory;
}
public Exchange createCacheExchange(String operation, String key,
Object value) {
Exchange exchange = new DefaultExchange(this.getCamelContext(),
getExchangePattern());
Message message = new DefaultMessage();
message.setHeader(CacheConstants.CACHE_OPERATION, operation);
message.setHeader(CacheConstants.CACHE_KEY, key);
message.setBody(value);
exchange.setIn(message);
return exchange;
}
/**
* Returns {@link Cache} instance or create new one if not exists.
*
* @return {@link Cache}
*/
public Ehcache initializeCache() {
CacheManager cacheManager = getCacheManagerFactory().getInstance();
Cache cache;
if (cacheManager.cacheExists(config.getCacheName())) {
if (LOG.isTraceEnabled()) {
LOG.trace("Found an existing cache: {}", config.getCacheName());
LOG.trace("Cache {} currently contains {} elements",
config.getCacheName(),
cacheManager.getCache(config.getCacheName()).getSize());
}
cache = cacheManager.getCache(config.getCacheName());
} else {
cache = new Cache(config.getCacheName(),
config.getMaxElementsInMemory(),
config.getMemoryStoreEvictionPolicy(),
config.isOverflowToDisk(),
config.getDiskStorePath(),
config.isEternal(),
config.getTimeToLiveSeconds(),
config.getTimeToIdleSeconds(),
config.isDiskPersistent(),
config.getDiskExpiryThreadIntervalSeconds(),
null);
for (CacheEventListener listener : config.getEventListenerRegistry().getEventListeners()) {
cache.getCacheEventNotificationService().registerListener(listener);
}
for (CacheLoaderWrapper loader : config.getCacheLoaderRegistry().getCacheLoaders()) {
loader.init(cache);
cache.registerCacheLoader(loader);
}
cacheManager.addCache(cache);
if (LOG.isDebugEnabled()) {
LOG.debug("Added a new cache: " + cache.getName());
}
}
return cache;
}
@Override
public void stop() {
CacheManager cacheManager = getCacheManagerFactory().getInstance();
cacheManager.removeCache(config.getCacheName());
}
}