| /* |
| * |
| * Licensed to the Apache Software Foundation (ASF) under one |
| * or more contributor license agreements. See the NOTICE file |
| * distributed with this work for additional information |
| * regarding copyright ownership. The ASF licenses this file |
| * to you 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 org.apache.qpid.transacted; |
| |
| import org.apache.qpid.client.AMQSession; |
| |
| import javax.jms.MessageProducer; |
| import javax.jms.MessageConsumer; |
| import javax.jms.Session; |
| import javax.jms.Destination; |
| import javax.jms.Connection; |
| import javax.jms.JMSException; |
| import javax.jms.Message; |
| import javax.jms.TextMessage; |
| import javax.jms.DeliveryMode; |
| |
| class Relay implements Runnable |
| { |
| private final Connection _con; |
| private final Session _session; |
| private final MessageConsumer _src; |
| private final MessageProducer _dest; |
| private final int _batch; |
| private final boolean _echo; |
| private int _counter; |
| private long start; |
| private boolean _running; |
| |
| Relay(Destination src, Destination dest, Connection con) throws JMSException |
| { |
| this(src, dest, con, false, 100, true); |
| } |
| |
| Relay(Destination src, Destination dest, Connection con, boolean echo, int batch, boolean persistent) throws JMSException |
| { |
| _echo = echo; |
| _batch = batch; |
| _con = con; |
| _session = con.createSession(true, AMQSession.NO_ACKNOWLEDGE); |
| _src = _session.createConsumer(src); |
| _dest = _session.createProducer(dest); |
| _dest.setDeliveryMode(persistent ? DeliveryMode.PERSISTENT : DeliveryMode.NON_PERSISTENT); |
| |
| } |
| |
| public void run() |
| { |
| start = System.currentTimeMillis(); |
| try{ |
| while(true) relay(); |
| } |
| catch(JMSException e) |
| { |
| e.printStackTrace(); |
| } |
| try |
| { |
| _session.close(); |
| } |
| catch (JMSException e) |
| { |
| e.printStackTrace(); |
| } |
| } |
| |
| void relay() throws JMSException |
| { |
| _dest.send(relay(_src.receive())); |
| _session.commit(); |
| } |
| |
| Message relay(Message in) throws JMSException |
| { |
| if(!_running) |
| { |
| System.out.println(_con.getClientID() + " started."); |
| _running = true; |
| } |
| if(++_counter % _batch == 0) |
| { |
| long time = System.currentTimeMillis() - start; |
| System.out.println(_batch + " iterations performed in " + time + " ms"); |
| try |
| { |
| Thread.sleep(100); |
| } |
| catch (InterruptedException e) |
| { |
| e.printStackTrace(); |
| } |
| start = System.currentTimeMillis(); |
| } |
| if(_echo) |
| { |
| System.out.println("Received: " + ((TextMessage) in).getText()); |
| } |
| return _session.createTextMessage(_con.getClientID() + _counter); |
| } |
| |
| void start() throws InterruptedException, JMSException |
| { |
| Thread runner = new Thread(this); |
| runner.start(); |
| _con.start(); |
| System.out.println(_con.getClientID() + " waiting..."); |
| runner.join(); |
| _con.close(); |
| } |
| } |