All Downloads are FREE. Search and download functionalities are using the official Maven repository.
Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
com.hazelcast.cache.impl.AbstractCacheProxyBase Maven / Gradle / Ivy
/*
* Copyright (c) 2008-2015, 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.cache.impl;
import com.hazelcast.cache.impl.operation.CacheDestroyOperation;
import com.hazelcast.config.CacheConfig;
import com.hazelcast.core.ExecutionCallback;
import com.hazelcast.core.HazelcastInstanceNotActiveException;
import com.hazelcast.nio.Address;
import com.hazelcast.nio.IOUtil;
import com.hazelcast.nio.serialization.Data;
import com.hazelcast.nio.serialization.SerializationService;
import com.hazelcast.partition.InternalPartitionService;
import com.hazelcast.spi.ExecutionService;
import com.hazelcast.spi.InternalCompletableFuture;
import com.hazelcast.spi.NodeEngine;
import com.hazelcast.spi.Operation;
import com.hazelcast.spi.OperationFactory;
import com.hazelcast.spi.OperationService;
import com.hazelcast.util.executor.CompletableFutureTask;
import java.io.Closeable;
import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;
import java.util.Set;
import java.util.concurrent.CopyOnWriteArrayList;
import java.util.concurrent.Future;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.atomic.AtomicBoolean;
import javax.cache.CacheException;
import javax.cache.configuration.Factory;
import javax.cache.integration.CacheLoader;
import javax.cache.integration.CompletionListener;
import static com.hazelcast.cache.impl.CacheProxyUtil.validateResults;
/**
* Abstract class providing cache open/close operations and {@link NodeEngine}, {@link CacheService} and
* {@link SerializationService} accessor which will be used by implementation of {@link com.hazelcast.cache.ICache}
* in server or embedded mode.
*
* @param the type of key.
* @param the type of value.
* @see com.hazelcast.cache.impl.CacheProxy
*/
abstract class AbstractCacheProxyBase {
static final int TIMEOUT = 10;
protected final CacheConfig cacheConfig;
//this will represent the name from the user perspective
protected final String name;
protected final String nameWithPrefix;
protected final ICacheService cacheService;
protected final SerializationService serializationService;
protected final CacheOperationProvider operationProvider;
private final NodeEngine nodeEngine;
private final CopyOnWriteArrayList loadAllTasks = new CopyOnWriteArrayList();
private CacheLoader cacheLoader;
private final AtomicBoolean isClosed = new AtomicBoolean(false);
private final AtomicBoolean isDestroyed = new AtomicBoolean(false);
protected AbstractCacheProxyBase(CacheConfig cacheConfig, NodeEngine nodeEngine, ICacheService cacheService) {
this.name = cacheConfig.getName();
this.nameWithPrefix = cacheConfig.getNameWithPrefix();
this.cacheConfig = cacheConfig;
this.nodeEngine = nodeEngine;
this.cacheService = cacheService;
this.serializationService = nodeEngine.getSerializationService();
this.operationProvider =
cacheService.getCacheOperationProvider(nameWithPrefix, cacheConfig.getInMemoryFormat());
init();
}
private void init() {
if (cacheConfig.getCacheLoaderFactory() != null) {
final Factory> cacheLoaderFactory = cacheConfig.getCacheLoaderFactory();
cacheLoader = cacheLoaderFactory.create();
} else {
cacheLoader = null;
}
}
//region close&destroy
protected void ensureOpen() {
if (isClosed()) {
throw new IllegalStateException("Cache operations can not be performed. The cache closed");
}
}
public void close() {
if (!isClosed.compareAndSet(false, true)) {
return;
}
Exception caughtException = null;
for (Future f : loadAllTasks) {
try {
f.get(TIMEOUT, TimeUnit.SECONDS);
} catch (Exception e) {
if (caughtException == null) {
caughtException = e;
}
getNodeEngine().getLogger(getClass()).warning("Problem while waiting for loadAll tasks to complete", e);
}
}
loadAllTasks.clear();
//close the configured CacheLoader
closeCacheLoader();
closeListeners();
if (caughtException != null) {
throw new CacheException("Problem while waiting for loadAll tasks to complete", caughtException);
}
}
public void destroy() {
close();
if (!isDestroyed.compareAndSet(false, true)) {
return;
}
isClosed.set(true);
Operation operation = new CacheDestroyOperation(cacheConfig.getNameWithPrefix());
int partitionId = getNodeEngine().getPartitionService().getPartitionId(getDistributedObjectName());
OperationService operationService = getNodeEngine().getOperationService();
InternalCompletableFuture f = operationService.invokeOnPartition(CacheService.SERVICE_NAME, operation, partitionId);
//todo What happens in exception case? Cache doesn't get destroyed
f.getSafely();
cacheService.destroyCache(getDistributedObjectName(), true, null);
f.getSafely();
}
public boolean isClosed() {
return isClosed.get();
}
public boolean isDestroyed() {
return isDestroyed.get();
}
public void open() {
if (isDestroyed.get()) {
throw new IllegalStateException("Cache is already destroyed! Cannot be reopened");
}
if (!isClosed.compareAndSet(true, false)) {
return;
}
init();
}
protected abstract void closeListeners();
//endregion close&destroy
//region DISTRIBUTED OBJECT
protected String getDistributedObjectName() {
return nameWithPrefix;
}
protected String getServiceName() {
return CacheService.SERVICE_NAME;
}
protected ICacheService getService() {
return cacheService;
}
protected NodeEngine getNodeEngine() {
if (nodeEngine == null || !nodeEngine.isActive()) {
throw new HazelcastInstanceNotActiveException();
}
return nodeEngine;
}
//endregion DISTRIBUTED OBJECT
//region CacheLoader
protected void validateCacheLoader(CompletionListener completionListener) {
if (cacheLoader == null && completionListener != null) {
completionListener.onCompletion();
}
}
protected void closeCacheLoader() {
//close the configured CacheLoader
if (cacheLoader instanceof Closeable) {
IOUtil.closeResource((Closeable) cacheLoader);
}
}
protected void submitLoadAllTask(LoadAllTask loadAllTask) {
final ExecutionService executionService = nodeEngine.getExecutionService();
final CompletableFutureTask future = (CompletableFutureTask) executionService
.submit("loadAll-" + nameWithPrefix, loadAllTask);
loadAllTasks.add(future);
future.andThen(new ExecutionCallback() {
@Override
public void onResponse(Object response) {
loadAllTasks.remove(future);
}
@Override
public void onFailure(Throwable t) {
loadAllTasks.remove(future);
getNodeEngine().getLogger(getClass()).warning("Problem in loadAll task", t);
}
});
}
protected final class LoadAllTask
implements Runnable {
private final CompletionListener completionListener;
private final CacheOperationProvider operationProvider;
private final Set keysData;
private final boolean replaceExistingValues;
public LoadAllTask(CacheOperationProvider operationProvider,
Set keysData, boolean replaceExistingValues, CompletionListener completionListener) {
this.operationProvider = operationProvider;
this.keysData = keysData;
this.replaceExistingValues = replaceExistingValues;
this.completionListener = completionListener;
}
@Override
public void run() {
try {
OperationService operationService = getNodeEngine().getOperationService();
OperationFactory operationFactory;
InternalPartitionService partitionService = getNodeEngine().getPartitionService();
Map> memberPartitionsMap = partitionService.getMemberPartitionsMap();
Map results = new HashMap();
for (Entry> memberPartitions : memberPartitionsMap.entrySet()) {
Set partitions = new HashSet(memberPartitions.getValue());
Set ownerKeys = filterOwnerKeys(partitionService, partitions);
operationFactory = operationProvider.createLoadAllOperationFactory(ownerKeys, replaceExistingValues);
Map memberResults;
memberResults = operationService.invokeOnPartitions(getServiceName(), operationFactory, partitions);
results.putAll(memberResults);
}
validateResults(results);
if (completionListener != null) {
completionListener.onCompletion();
}
} catch (Exception e) {
if (completionListener != null) {
completionListener.onException(e);
}
}
}
private Set filterOwnerKeys(InternalPartitionService partitionService, Set partitions) {
Set ownerKeys = new HashSet();
for (Data key: keysData) {
int keyPartitionId = partitionService.getPartitionId(key);
if (partitions.contains(keyPartitionId)) {
ownerKeys.add(key);
}
}
return ownerKeys;
}
}
//endregion CacheLoader
}
