/*
* 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.operation;
import com.hazelcast.map.MapContainer;
import com.hazelcast.map.MapInterceptor;
import com.hazelcast.map.MapService;
import com.hazelcast.nio.ObjectDataInput;
import com.hazelcast.nio.ObjectDataOutput;
import com.hazelcast.nio.serialization.DataSerializable;
import com.hazelcast.query.impl.Index;
import com.hazelcast.query.impl.IndexService;
import com.hazelcast.spi.AbstractOperation;
import java.io.IOException;
import java.util.*;
public class PostJoinMapOperation extends AbstractOperation {
private List<MapIndexInfo> indexInfoList = new LinkedList<MapIndexInfo>();
private List<InterceptorInfo> interceptorInfoList = new LinkedList<InterceptorInfo>();
@Override
public String getServiceName() {
return MapService.SERVICE_NAME;
}
public void addMapIndex(MapContainer mapContainer) {
final IndexService indexService = mapContainer.getIndexService();
if (indexService.hasIndex()) {
MapIndexInfo mapIndexInfo = new MapIndexInfo(mapContainer.getName());
for (Index index : indexService.getIndexes()) {
mapIndexInfo.addIndexInfo(index.getAttributeName(), index.isOrdered());
}
indexInfoList.add(mapIndexInfo);
}
}
public void addMapInterceptors(MapContainer mapContainer) {
List<MapInterceptor> interceptorList = mapContainer.getInterceptors();
Map<String, MapInterceptor> interceptorMap = mapContainer.getInterceptorMap();
Map<MapInterceptor, String> revMap = new HashMap<MapInterceptor, String>();
for (Map.Entry<String, MapInterceptor> entry : interceptorMap.entrySet()) {
revMap.put(entry.getValue(), entry.getKey());
}
InterceptorInfo interceptorInfo = new InterceptorInfo(mapContainer.getName());
for (MapInterceptor interceptor : interceptorList) {
interceptorInfo.addInterceptor(revMap.get(interceptor), interceptor);
}
interceptorInfoList.add(interceptorInfo);
}
static class InterceptorInfo implements DataSerializable {
String mapName;
final List<Map.Entry<String, MapInterceptor>> interceptors = new LinkedList<Map.Entry<String, MapInterceptor>>();
InterceptorInfo(String mapName) {
this.mapName = mapName;
}
InterceptorInfo() {
}
void addInterceptor(String id, MapInterceptor interceptor) {
interceptors.add(new AbstractMap.SimpleImmutableEntry<String, MapInterceptor>(id, interceptor));
}
@Override
public void writeData(ObjectDataOutput out) throws IOException {
out.writeUTF(mapName);
out.writeInt(interceptors.size());
for (Map.Entry<String, MapInterceptor> entry : interceptors) {
out.writeUTF(entry.getKey());
out.writeObject(entry.getValue());
}
}
@Override
public void readData(ObjectDataInput in) throws IOException {
mapName = in.readUTF();
int size = in.readInt();
for (int i = 0; i < size; i++) {
String id = in.readUTF();
MapInterceptor interceptor = in.readObject();
interceptors.add(new AbstractMap.SimpleImmutableEntry<String, MapInterceptor>(id, interceptor));
}
}
}
static class MapIndexInfo implements DataSerializable {
String mapName;
private List<MapIndexInfo.IndexInfo> lsIndexes = new LinkedList<MapIndexInfo.IndexInfo>();
public MapIndexInfo(String mapName) {
this.mapName = mapName;
}
public MapIndexInfo() {
}
static class IndexInfo implements DataSerializable {
String attributeName;
boolean ordered;
IndexInfo() {
}
IndexInfo(String attributeName, boolean ordered) {
this.attributeName = attributeName;
this.ordered = ordered;
}
public void writeData(ObjectDataOutput out) throws IOException {
out.writeUTF(attributeName);
out.writeBoolean(ordered);
}
public void readData(ObjectDataInput in) throws IOException {
attributeName = in.readUTF();
ordered = in.readBoolean();
}
}
public void addIndexInfo(String attributeName, boolean ordered) {
lsIndexes.add(new MapIndexInfo.IndexInfo(attributeName, ordered));
}
public void writeData(ObjectDataOutput out) throws IOException {
out.writeUTF(mapName);
out.writeInt(lsIndexes.size());
for (MapIndexInfo.IndexInfo indexInfo : lsIndexes) {
indexInfo.writeData(out);
}
}
public void readData(ObjectDataInput in) throws IOException {
mapName = in.readUTF();
int size = in.readInt();
for (int i = 0; i < size; i++) {
MapIndexInfo.IndexInfo indexInfo = new MapIndexInfo.IndexInfo();
indexInfo.readData(in);
lsIndexes.add(indexInfo);
}
}
}
@Override
public void run() throws Exception {
MapService mapService = getService();
for (MapIndexInfo mapIndex : indexInfoList) {
final MapContainer mapContainer = mapService.getMapContainer(mapIndex.mapName);
final IndexService indexService = mapContainer.getIndexService();
for (MapIndexInfo.IndexInfo indexInfo : mapIndex.lsIndexes) {
indexService.addOrGetIndex(indexInfo.attributeName, indexInfo.ordered);
}
}
for (InterceptorInfo interceptorInfo : interceptorInfoList) {
final MapContainer mapContainer = mapService.getMapContainer(interceptorInfo.mapName);
Map<String, MapInterceptor> interceptorMap = mapContainer.getInterceptorMap();
List<Map.Entry<String, MapInterceptor>> entryList = interceptorInfo.interceptors;
for (Map.Entry<String, MapInterceptor> entry : entryList) {
if (!interceptorMap.containsKey(entry.getKey())) {
mapContainer.addInterceptor(entry.getKey(), entry.getValue());
}
}
}
}
@Override
protected void writeInternal(ObjectDataOutput out) throws IOException {
super.writeInternal(out);
out.writeInt(indexInfoList.size());
for (MapIndexInfo mapIndex : indexInfoList) {
mapIndex.writeData(out);
}
out.writeInt(interceptorInfoList.size());
for (InterceptorInfo interceptorInfo : interceptorInfoList) {
interceptorInfo.writeData(out);
}
}
@Override
protected void readInternal(ObjectDataInput in) throws IOException {
super.readInternal(in);
int size = in.readInt();
for (int i = 0; i < size; i++) {
MapIndexInfo mapIndexInfo = new MapIndexInfo();
mapIndexInfo.readData(in);
indexInfoList.add(mapIndexInfo);
}
int size2 = in.readInt();
for (int i = 0; i < size2; i++) {
InterceptorInfo info = new InterceptorInfo();
info.readData(in);
interceptorInfoList.add(info);
}
}
}