/*
* Copyright (c) 2008-2013, Hazelcast, Inc. All Rights Reserved.
*
* Licensed 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 com.hazelcast.spring.cache;
import com.hazelcast.core.DistributedObject;
import com.hazelcast.core.HazelcastInstance;
import com.hazelcast.core.IMap;
import org.springframework.beans.factory.InitializingBean;
import org.springframework.cache.Cache;
import org.springframework.cache.CacheManager;
import java.util.Collection;
import java.util.Collections;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ConcurrentMap;
/**
* @author mdogan 4/3/12
*/
public class HazelcastCacheManager implements CacheManager, InitializingBean {
private final ConcurrentMap<String, Cache> caches = new ConcurrentHashMap<String, Cache>();
private HazelcastInstance hazelcastInstance;
public HazelcastCacheManager() {
}
public HazelcastCacheManager(HazelcastInstance hazelcastInstance) {
this.hazelcastInstance = hazelcastInstance;
}
public void afterPropertiesSet() throws Exception {
final Collection<DistributedObject> distributedObjects = hazelcastInstance.getDistributedObjects();
for (DistributedObject distributedObject : distributedObjects) {
if (distributedObject instanceof IMap) {
final IMap<Object, Object> map = (IMap) distributedObject;
caches.put(map.getName(), new HazelcastCache(map));
}
}
}
public Cache getCache(String name) {
Cache cache = caches.get(name);
if (cache == null) {
final IMap<Object, Object> map = hazelcastInstance.getMap(name);
cache = new HazelcastCache(map);
final Cache currentCache = caches.putIfAbsent(name, cache);
if (currentCache != null) {
cache = currentCache;
}
}
return cache;
}
public void setHazelcastInstance(final HazelcastInstance hazelcastInstance) {
this.hazelcastInstance = hazelcastInstance;
}
public Collection<String> getCacheNames() {
return Collections.unmodifiableCollection(caches.keySet());
}
}