com.hazelcast.map.impl.operation.PostJoinMapOperation Maven / Gradle / Ivy
/*
* Copyright (c) 2008-2016, 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.impl.operation;
import com.hazelcast.map.MapInterceptor;
import com.hazelcast.map.impl.InterceptorRegistry;
import com.hazelcast.map.impl.MapContainer;
import com.hazelcast.map.impl.MapService;
import com.hazelcast.map.impl.MapServiceContext;
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.Indexes;
import com.hazelcast.spi.AbstractOperation;
import java.io.IOException;
import java.util.AbstractMap;
import java.util.HashMap;
import java.util.LinkedList;
import java.util.List;
import java.util.Map;
public class PostJoinMapOperation extends AbstractOperation {
private List indexInfoList = new LinkedList();
private List interceptorInfoList = new LinkedList();
@Override
public String getServiceName() {
return MapService.SERVICE_NAME;
}
public void addMapIndex(MapContainer mapContainer) {
final Indexes indexes = mapContainer.getIndexes();
if (indexes.hasIndex()) {
MapIndexInfo mapIndexInfo = new MapIndexInfo(mapContainer.getName());
for (Index index : indexes.getIndexes()) {
mapIndexInfo.addIndexInfo(index.getAttributeName(), index.isOrdered());
}
indexInfoList.add(mapIndexInfo);
}
}
public void addMapInterceptors(MapContainer mapContainer) {
InterceptorRegistry interceptorRegistry = mapContainer.getInterceptorRegistry();
List interceptorList = interceptorRegistry.getInterceptors();
Map interceptorMap = interceptorRegistry.getId2InterceptorMap();
Map revMap = new HashMap();
for (Map.Entry 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 {
private String mapName;
private final List> interceptors = new LinkedList>();
InterceptorInfo(String mapName) {
this.mapName = mapName;
}
InterceptorInfo() {
}
void addInterceptor(String id, MapInterceptor interceptor) {
interceptors.add(new AbstractMap.SimpleImmutableEntry(id, interceptor));
}
@Override
public void writeData(ObjectDataOutput out) throws IOException {
out.writeUTF(mapName);
out.writeInt(interceptors.size());
for (Map.Entry 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(id, interceptor));
}
}
}
@Override
public void run() throws Exception {
MapService mapService = getService();
MapServiceContext mapServiceContext = mapService.getMapServiceContext();
for (MapIndexInfo mapIndex : indexInfoList) {
final MapContainer mapContainer = mapServiceContext.getMapContainer(mapIndex.mapName);
final Indexes indexes = mapContainer.getIndexes();
for (MapIndexInfo.IndexInfo indexInfo : mapIndex.lsIndexes) {
indexes.addOrGetIndex(indexInfo.attributeName, indexInfo.ordered);
}
}
for (InterceptorInfo interceptorInfo : interceptorInfoList) {
final MapContainer mapContainer = mapServiceContext.getMapContainer(interceptorInfo.mapName);
InterceptorRegistry interceptorRegistry = mapContainer.getInterceptorRegistry();
Map interceptorMap = interceptorRegistry.getId2InterceptorMap();
List> entryList = interceptorInfo.interceptors;
for (Map.Entry entry : entryList) {
if (!interceptorMap.containsKey(entry.getKey())) {
interceptorRegistry.register(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);
}
}
static class MapIndexInfo implements DataSerializable {
private String mapName;
private List lsIndexes = new LinkedList();
public MapIndexInfo(String mapName) {
this.mapName = mapName;
}
public MapIndexInfo() {
}
static class IndexInfo implements DataSerializable {
private String attributeName;
private boolean ordered;
IndexInfo() {
}
IndexInfo(String attributeName, boolean ordered) {
this.attributeName = attributeName;
this.ordered = ordered;
}
@Override
public void writeData(ObjectDataOutput out) throws IOException {
out.writeUTF(attributeName);
out.writeBoolean(ordered);
}
@Override
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));
}
@Override
public void writeData(ObjectDataOutput out) throws IOException {
out.writeUTF(mapName);
out.writeInt(lsIndexes.size());
for (MapIndexInfo.IndexInfo indexInfo : lsIndexes) {
indexInfo.writeData(out);
}
}
@Override
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);
}
}
}
}