[44] | 1 | package omq.server; |
---|
| 2 | |
---|
| 3 | import java.lang.reflect.InvocationTargetException; |
---|
| 4 | import java.util.concurrent.BlockingQueue; |
---|
| 5 | |
---|
| 6 | import omq.common.message.Request; |
---|
| 7 | import omq.common.message.Response; |
---|
| 8 | import omq.common.util.Serializer; |
---|
| 9 | import omq.exception.OmqException; |
---|
| 10 | |
---|
[49] | 11 | import org.apache.log4j.Logger; |
---|
| 12 | |
---|
[44] | 13 | import com.rabbitmq.client.AMQP.BasicProperties; |
---|
| 14 | import com.rabbitmq.client.Channel; |
---|
| 15 | import com.rabbitmq.client.QueueingConsumer.Delivery; |
---|
| 16 | |
---|
| 17 | /** |
---|
| 18 | * |
---|
| 19 | * @author Sergi Toda <sergi.toda@estudiants.urv.cat> |
---|
| 20 | * |
---|
| 21 | */ |
---|
| 22 | public class InvocationThread extends Thread { |
---|
[49] | 23 | private static final Logger logger = Logger.getLogger(InvocationThread.class.getName()); |
---|
[44] | 24 | private RemoteObject obj; |
---|
[53] | 25 | private transient Serializer serializer; |
---|
[44] | 26 | private BlockingQueue<Delivery> deliveryQueue; |
---|
| 27 | private boolean killed = false; |
---|
| 28 | |
---|
[53] | 29 | public InvocationThread(RemoteObject obj, BlockingQueue<Delivery> deliveryQueue, Serializer serializer) { |
---|
[44] | 30 | this.obj = obj; |
---|
| 31 | this.deliveryQueue = deliveryQueue; |
---|
[53] | 32 | this.serializer = serializer; |
---|
[44] | 33 | } |
---|
| 34 | |
---|
| 35 | @Override |
---|
| 36 | public void run() { |
---|
| 37 | while (!killed) { |
---|
| 38 | try { |
---|
| 39 | // Get the delivery |
---|
| 40 | Delivery delivery = deliveryQueue.take(); |
---|
| 41 | |
---|
[47] | 42 | String serializerType = delivery.getProperties().getType(); |
---|
| 43 | |
---|
[44] | 44 | // Deserialize the json |
---|
[53] | 45 | Request request = serializer.deserializeRequest(serializerType, delivery.getBody(), obj); |
---|
[44] | 46 | String methodName = request.getMethod(); |
---|
| 47 | String requestID = request.getId(); |
---|
| 48 | |
---|
[49] | 49 | logger.debug("Object: " + obj.getRef() + ", method: " + methodName + " corrID: " + requestID); |
---|
[44] | 50 | |
---|
| 51 | // Invoke the method |
---|
| 52 | Object result = null; |
---|
| 53 | OmqException error = null; |
---|
| 54 | try { |
---|
| 55 | result = obj.invokeMethod(request.getMethod(), request.getParams()); |
---|
| 56 | } catch (InvocationTargetException e) { |
---|
| 57 | Throwable throwable = e.getTargetException(); |
---|
[49] | 58 | logger.error("Object: " + obj.getRef() + " at method: " + methodName + ", corrID" + requestID, throwable); |
---|
[44] | 59 | error = new OmqException(throwable.getClass().getCanonicalName(), throwable.getMessage()); |
---|
| 60 | } catch (NoSuchMethodException e) { |
---|
[49] | 61 | logger.error("Object: " + obj.getRef() + " cannot find method: " + methodName); |
---|
[44] | 62 | error = new OmqException(e.getClass().getCanonicalName(), e.getMessage()); |
---|
| 63 | } |
---|
| 64 | |
---|
| 65 | // Reply if it's necessary |
---|
| 66 | if (!request.isAsync()) { |
---|
| 67 | Response resp = new Response(request.getId(), obj.getRef(), result, error); |
---|
| 68 | |
---|
| 69 | Channel channel = obj.getChannel(); |
---|
| 70 | |
---|
| 71 | BasicProperties props = delivery.getProperties(); |
---|
| 72 | |
---|
| 73 | BasicProperties replyProps = new BasicProperties.Builder().appId(obj.getRef()).correlationId(props.getCorrelationId()).build(); |
---|
| 74 | |
---|
[53] | 75 | byte[] bytesResponse = serializer.serialize(serializerType, resp); |
---|
[44] | 76 | channel.basicPublish("", props.getReplyTo(), replyProps, bytesResponse); |
---|
[54] | 77 | logger.debug("Publish sync response -> Object: " + obj.getRef() + ", method: " + methodName + " corrID: " + requestID + " replyTo: " |
---|
| 78 | + props.getReplyTo()); |
---|
[44] | 79 | } |
---|
| 80 | |
---|
| 81 | } catch (InterruptedException i) { |
---|
[49] | 82 | logger.error(i); |
---|
[44] | 83 | killed = true; |
---|
| 84 | } catch (Exception e) { |
---|
[49] | 85 | logger.error("Object: " + obj.getRef(), e); |
---|
[44] | 86 | } |
---|
| 87 | |
---|
| 88 | } |
---|
| 89 | } |
---|
| 90 | |
---|
| 91 | public RemoteObject getObj() { |
---|
| 92 | return obj; |
---|
| 93 | } |
---|
| 94 | |
---|
| 95 | public void setObj(RemoteObject obj) { |
---|
| 96 | this.obj = obj; |
---|
| 97 | } |
---|
| 98 | |
---|
| 99 | public BlockingQueue<Delivery> getDeliveryQueue() { |
---|
| 100 | return deliveryQueue; |
---|
| 101 | } |
---|
| 102 | |
---|
| 103 | public void setDeliveryQueue(BlockingQueue<Delivery> deliveryQueue) { |
---|
| 104 | this.deliveryQueue = deliveryQueue; |
---|
| 105 | } |
---|
| 106 | } |
---|