/*
* 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.map;
import com.hazelcast.concurrent.lock.
LockService;
import com.hazelcast.spi.
DefaultObjectNamespace;
import com.hazelcast.spi.
NodeEngine;
import com.hazelcast.util.
ConcurrencyUtil;
import com.hazelcast.util.
ConstructorFunction;
import java.util.concurrent.
ConcurrentHashMap;
import java.util.concurrent.
ConcurrentMap;
public class
PartitionContainer {
private final
MapService mapService;
private final int
partitionId;
private final
ConcurrentMap<
String,
RecordStore>
maps = new
ConcurrentHashMap<
String,
RecordStore>(1000);
public
PartitionContainer(final
MapService mapService, final int
partitionId) {
this.
mapService =
mapService;
this.
partitionId =
partitionId;
}
private final
ConstructorFunction<
String,
RecordStore>
recordStoreConstructor
= new
ConstructorFunction<
String,
RecordStore>() {
public
RecordStore createNew(
String name) {
return new
DefaultRecordStore(
name,
mapService,
partitionId);
}
};
public
ConcurrentMap<
String,
RecordStore>
getMaps() {
return
maps;
}
public int
getPartitionId() {
return
partitionId;
}
public
MapService getMapService() {
return
mapService;
}
public
RecordStore getRecordStore(
String name) {
return
ConcurrencyUtil.
getOrPutSynchronized(
maps,
name, this,
recordStoreConstructor);
}
public
RecordStore getExistingRecordStore(
String mapName) {
return
maps.
get(
mapName);
}
void
destroyMap(
String name) {
RecordStore recordStore =
maps.
remove(
name);
if (
recordStore != null) {
recordStore.
clearPartition();
} else {
clearLockStore(
name);
}
}
private void
clearLockStore(
String name) {
final
NodeEngine nodeEngine =
mapService.
getNodeEngine();
final
LockService lockService =
nodeEngine.
getSharedService(
LockService.
SERVICE_NAME);
if (
lockService != null) {
final
DefaultObjectNamespace namespace = new
DefaultObjectNamespace(
MapService.
SERVICE_NAME,
name);
lockService.
clearLockStore(
partitionId,
namespace);
}
}
void
clear() {
for (
RecordStore recordStore :
maps.
values()) {
recordStore.
clearPartition();
}
maps.
clear();
}
}