/*
 *
 * 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();
    }
}