78 lines
1.8 KiB
Java
78 lines
1.8 KiB
Java
package org.jivesoftware.smack;
|
|
|
|
import java.util.concurrent.ArrayBlockingQueue;
|
|
import java.util.concurrent.BlockingQueue;
|
|
import java.util.concurrent.LinkedBlockingQueue;
|
|
|
|
import org.jivesoftware.smack.packet.IQ;
|
|
import org.jivesoftware.smack.packet.Message;
|
|
import org.jivesoftware.smack.packet.Packet;
|
|
import org.jivesoftware.smack.packet.IQ.Type;
|
|
|
|
public class ThreadedDummyConnection extends DummyConnection
|
|
{
|
|
private BlockingQueue<IQ> replyQ = new ArrayBlockingQueue<IQ>(1);
|
|
private BlockingQueue<Packet> messageQ = new LinkedBlockingQueue<Packet>(5);
|
|
|
|
@Override
|
|
public void sendPacket(Packet packet)
|
|
{
|
|
super.sendPacket(packet);
|
|
|
|
if ((packet instanceof IQ) && !replyQ.isEmpty())
|
|
{
|
|
// Set reply packet to match one being sent. We haven't started the
|
|
// other thread yet so this is still safe.
|
|
IQ replyPacket = replyQ.peek();
|
|
replyPacket.setPacketID(packet.getPacketID());
|
|
replyPacket.setFrom(packet.getTo());
|
|
replyPacket.setTo(packet.getFrom());
|
|
replyPacket.setType(Type.RESULT);
|
|
|
|
new ProcessQueue(replyQ).start();
|
|
}
|
|
}
|
|
|
|
public void addMessage(Message msgToProcess)
|
|
{
|
|
messageQ.add(msgToProcess);
|
|
}
|
|
|
|
public void addIQReply(IQ reply)
|
|
{
|
|
replyQ.add(reply);
|
|
}
|
|
|
|
public void processMessages()
|
|
{
|
|
if (!messageQ.isEmpty())
|
|
new ProcessQueue(messageQ).start();
|
|
else
|
|
System.out.println("No messages to process");
|
|
}
|
|
|
|
class ProcessQueue extends Thread
|
|
{
|
|
private BlockingQueue<? extends Packet> processQ;
|
|
|
|
ProcessQueue(BlockingQueue<? extends Packet> queue)
|
|
{
|
|
processQ = queue;
|
|
}
|
|
|
|
@Override
|
|
public void run()
|
|
{
|
|
try
|
|
{
|
|
processPacket(processQ.take());
|
|
}
|
|
catch (InterruptedException e)
|
|
{
|
|
e.printStackTrace();
|
|
}
|
|
}
|
|
};
|
|
|
|
}
|