package org.drools.grid.io.impl;
import java.util.concurrent.atomic.AtomicInteger;
import org.drools.grid.io.Connector;
import org.drools.grid.io.Conversation;
import org.drools.grid.io.ConversationManager;
import org.drools.grid.io.IoWriter;
import org.drools.grid.io.Message;
import org.drools.grid.io.MessageReceiverHandler;
public class ConversationImpl
implements
Conversation {
private Connector conn;
private IoWriter writer;
private String conversationId;
private String senderId;
private String recipientId;
private AtomicInteger requestId;
private RequestResponseDispatchListener dispathListener;
private ConversationManager conversationManager;
private Message receivedMessage;
public ConversationImpl(Connector conn,
String conversationId,
String senderId,
String recipientId,
RequestResponseDispatchListener dispathListener,
IoWriter writer,
ConversationManager conversationManager) {
this( conn,
conversationId,
senderId,
recipientId,
dispathListener,
null,
writer,
conversationManager );
}
public ConversationImpl(Connector conn,
String conversationId,
String senderId,
String recipientId,
RequestResponseDispatchListener dispathListener,
Message receivedMessage,
IoWriter writer,
ConversationManager conversationManager) {
this.conn = conn;
this.conversationId = conversationId;
this.senderId = senderId;
this.recipientId = recipientId;
this.dispathListener = dispathListener;
this.writer = writer;
this.requestId = new AtomicInteger();
this.receivedMessage = receivedMessage;
this.conversationManager = conversationManager;
}
public void respond(Object body) {
Message msg = new MessageImpl( this.conversationId,
this.senderId,
recipientId,
-1,
receivedMessage.getRequestId(),
body );
writer.write( msg );
}
public void sendMessage(Object body,
MessageReceiverHandler handler) {
int requestId = -1;
if ( handler != null ) {
requestId = this.requestId.getAndIncrement();
}
Message msg = new MessageImpl( this.conversationId,
this.senderId,
recipientId,
requestId,
-1,
body );
this.dispathListener.addMessageReceiverHandler( requestId,
handler );
writer.write( msg );
}
public void endConversation() {
this.conn.close();
}
}