/*
* 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.queue;
import com.hazelcast.core.
ItemEventType;
import com.hazelcast.nio.
IOUtil;
import com.hazelcast.nio.
ObjectDataInput;
import com.hazelcast.nio.
ObjectDataOutput;
import com.hazelcast.nio.serialization.
Data;
import com.hazelcast.spi.
Notifier;
import com.hazelcast.spi.
Operation;
import com.hazelcast.spi.
WaitNotifyKey;
import java.io.
IOException;
import java.util.
ArrayList;
import java.util.
Collection;
import java.util.
Map;
/**
* @author ali 12/20/12
*/
public class
CompareAndRemoveOperation extends
QueueBackupAwareOperation implements
Notifier {
private
Collection<
Data>
dataList;
Map<
Long,
Data>
dataMap;
boolean
retain;
public
CompareAndRemoveOperation() {
}
public
CompareAndRemoveOperation(
String name,
Collection<
Data>
dataList, boolean
retain) {
super(
name);
this.
dataList =
dataList;
this.
retain =
retain;
}
public void
run() {
dataMap =
getOrCreateContainer().
compareAndRemove(
dataList,
retain);
response =
dataMap.
size() > 0;
}
public void
afterRun() throws
Exception {
getQueueService().
getLocalQueueStatsImpl(
name).
incrementOtherOperations();
if (
hasListener()) {
for (
Data data :
dataMap.
values()) {
publishEvent(
ItemEventType.
REMOVED,
data);
}
}
}
public boolean
shouldBackup() {
return
Boolean.
TRUE.
equals(
response);
}
public
Operation getBackupOperation() {
return new
CompareAndRemoveBackupOperation(
name,
dataMap.
keySet());
}
protected void
writeInternal(
ObjectDataOutput out) throws
IOException {
super.writeInternal(
out);
out.
writeBoolean(
retain);
out.
writeInt(
dataList.
size());
for (
Data data :
dataList) {
data.
writeData(
out);
}
}
protected void
readInternal(
ObjectDataInput in) throws
IOException {
super.readInternal(
in);
retain =
in.
readBoolean();
int
size =
in.
readInt();
dataList = new
ArrayList<
Data>(
size);
for (int
i = 0;
i <
size;
i++) {
dataList.
add(
IOUtil.
readData(
in));
}
}
public boolean
shouldNotify() {
return
Boolean.
TRUE.
equals(
response);
}
public
WaitNotifyKey getNotifiedKey() {
return
getOrCreateContainer().
getOfferWaitNotifyKey();
}
public int
getId() {
return
QueueDataSerializerHook.
COMPARE_AND_REMOVE;
}
}