/*
* 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.nio.serialization.
IdentifiedDataSerializable;
import com.hazelcast.spi.
Notifier;
import com.hazelcast.spi.
Operation;
import com.hazelcast.spi.
WaitNotifyKey;
import com.hazelcast.spi.
WaitSupport;
import java.io.
IOException;
/**
* User: ali
* Date: 11/14/12
* Time: 12:14 AM
*/
public final class
OfferOperation extends
QueueBackupAwareOperation
implements
WaitSupport,
Notifier,
IdentifiedDataSerializable {
private
Data data;
private long
itemId;
public
OfferOperation() {
}
public
OfferOperation(final
String name, final long
timeout, final
Data data) {
super(
name,
timeout);
this.
data =
data;
}
public void
run() {
QueueContainer container =
getOrCreateContainer();
if (
container.
hasEnoughCapacity()) {
itemId =
container.
offer(
data);
response = true;
} else {
response = false;
}
}
public void
afterRun() throws
Exception {
if (
Boolean.
TRUE.
equals(
response)) {
getQueueService().
getLocalQueueStatsImpl(
name).
incrementOffers();
publishEvent(
ItemEventType.
ADDED,
data);
} else {
getQueueService().
getLocalQueueStatsImpl(
name).
incrementRejectedOffers();
}
}
public
Operation getBackupOperation() {
return new
OfferBackupOperation(
name,
data,
itemId);
}
public boolean
shouldBackup() {
return
Boolean.
TRUE.
equals(
response);
}
public boolean
shouldNotify() {
return
Boolean.
TRUE.
equals(
response);
}
public
WaitNotifyKey getNotifiedKey() {
return
getOrCreateContainer().
getPollWaitNotifyKey();
}
public
WaitNotifyKey getWaitKey() {
return
getOrCreateContainer().
getOfferWaitNotifyKey();
}
public boolean
shouldWait() {
QueueContainer container =
getOrCreateContainer();
return
getWaitTimeout() != 0 && !
container.
hasEnoughCapacity();
}
public void
onWaitExpire() {
getResponseHandler().
sendResponse(
Boolean.
FALSE);
}
protected void
writeInternal(
ObjectDataOutput out) throws
IOException {
super.writeInternal(
out);
data.
writeData(
out);
}
protected void
readInternal(
ObjectDataInput in) throws
IOException {
super.readInternal(
in);
data =
IOUtil.
readData(
in);
}
public int
getFactoryId() {
return
QueueDataSerializerHook.
F_ID;
}
public int
getId() {
return
QueueDataSerializerHook.
OFFER;
}
}