com.hazelcast.map.impl.operation.RemoveFromLoadAllOperation Maven / Gradle / Ivy
/*
* Copyright (c) 2008-2020, 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.internal.nio.IOUtil;
import com.hazelcast.map.IMap;
import com.hazelcast.map.impl.MapDataSerializerHook;
import com.hazelcast.map.impl.recordstore.Storage;
import com.hazelcast.nio.ObjectDataInput;
import com.hazelcast.nio.ObjectDataOutput;
import com.hazelcast.internal.serialization.Data;
import com.hazelcast.spi.impl.operationservice.PartitionAwareOperation;
import com.hazelcast.spi.impl.operationservice.MutatingOperation;
import java.io.IOException;
import java.util.ArrayList;
import java.util.Collection;
import java.util.Collections;
import java.util.List;
/**
* Removes keys that are contained in the {@link com.hazelcast.map.impl.recordstore.RecordStore}
* from the provided list, leaving only keys which are not contained in the record store.
*
* This operation is expected to be invoked locally as the key collection is mutated
* when running the operation and the operation does not have a response object.
*
* @see IMap#loadAll(java.util.Set, boolean)
*/
public class RemoveFromLoadAllOperation extends MapOperation implements PartitionAwareOperation, MutatingOperation {
private List keys;
public RemoveFromLoadAllOperation() {
keys = Collections.emptyList();
}
public RemoveFromLoadAllOperation(String name, List keys) {
super(name);
this.keys = keys;
}
@Override
protected void runInternal() {
removeExistingKeys(keys);
}
/**
* Removes keys from the provided collection which
* are contained in the partition record store.
*
* @param keys the keys to be filtered
*/
private void removeExistingKeys(Collection keys) {
if (keys == null || keys.isEmpty()) {
return;
}
Storage storage = recordStore.getStorage();
keys.removeIf(storage::containsKey);
}
@Override
protected void writeInternal(ObjectDataOutput out) throws IOException {
super.writeInternal(out);
final int size = keys.size();
out.writeInt(size);
for (Data key : keys) {
IOUtil.writeData(out, key);
}
}
@Override
protected void readInternal(ObjectDataInput in) throws IOException {
super.readInternal(in);
final int size = in.readInt();
if (size > 0) {
keys = new ArrayList<>(size);
}
for (int i = 0; i < size; i++) {
Data data = IOUtil.readData(in);
keys.add(data);
}
}
@Override
public int getClassId() {
return MapDataSerializerHook.REMOVE_FROM_LOAD_ALL;
}
}