/*
*
* 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.amqp_1_0.client;
import org.apache.qpid.amqp_1_0.messaging.SectionDecoder;
import org.apache.qpid.amqp_1_0.messaging.SectionDecoderImpl;
import org.apache.qpid.amqp_1_0.messaging.SectionEncoder;
import org.apache.qpid.amqp_1_0.messaging.SectionEncoderImpl;
import org.apache.qpid.amqp_1_0.transport.DeliveryStateHandler;
import org.apache.qpid.amqp_1_0.transport.SendingLinkEndpoint;
import org.apache.qpid.amqp_1_0.transport.SessionEndpoint;
import org.apache.qpid.amqp_1_0.transport.SessionState;
import org.apache.qpid.amqp_1_0.type.*;
import org.apache.qpid.amqp_1_0.type.messaging.Filter;
import org.apache.qpid.amqp_1_0.type.messaging.Source;
import org.apache.qpid.amqp_1_0.type.messaging.StdDistMode;
import org.apache.qpid.amqp_1_0.type.messaging.Target;
import org.apache.qpid.amqp_1_0.type.transaction.TxnCapability;
import org.apache.qpid.amqp_1_0.type.transport.ReceiverSettleMode;
import org.apache.qpid.amqp_1_0.type.transport.SenderSettleMode;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.UUID;
public class Session
{
private SessionEndpoint _endpoint;
private List<Receiver> _receivers = new ArrayList<Receiver>();
private List<Sender> _senders = new ArrayList<Sender>();
private SectionEncoder _sectionEncoder;
private SectionDecoder _sectionDecoder;
private TransactionController _sessionLocalTC;
private Connection _connection;
public Session(final Connection connection, String name) throws ChannelsExhaustedException
{
_connection = connection;
_endpoint = connection.getEndpoint().createSession(name);
if(_endpoint == null)
{
throw new ChannelsExhaustedException("Cannot create session as all channels are in use");
}
_sectionEncoder = new SectionEncoderImpl(connection.getEndpoint().getDescribedTypeRegistry());
_sectionDecoder = new SectionDecoderImpl(connection.getEndpoint().getDescribedTypeRegistry());
}
public Sender createSender(final String targetName)
throws Sender.SenderCreationException, ConnectionClosedException
{
final String sourceName = UUID.randomUUID().toString();
return new Sender(this, targetName +"<-"+sourceName, targetName, sourceName, false);
}
public Sender createSender(final String targetName, final SourceConfigurator configurator)
throws Sender.SenderCreationException, ConnectionClosedException
{
final String sourceName = UUID.randomUUID().toString();
return new Sender(this, targetName +"<-"+sourceName, targetName, sourceName, false)
{
@Override
protected void configureSource(final Source source)
{
configurator.configureSource(source);
}
};
}
public Sender createSender(final String targetName, int window)
throws Sender.SenderCreationException, ConnectionClosedException
{
final String sourceName = UUID.randomUUID().toString();
return new Sender(this, targetName+"<-"+sourceName, targetName, sourceName, window);
}
public Sender createSender(String targetName, int window, AcknowledgeMode mode, String linkName)
throws Sender.SenderCreationException, ConnectionClosedException
{
return createSender(targetName, window, mode, linkName, false, null);
}
public Sender createSender(String targetName, int window, AcknowledgeMode mode, String linkName,
boolean isDurable, Map<Binary, Outcome> unsettled)
throws Sender.SenderCreationException, ConnectionClosedException
{
return new Sender(this, linkName == null ? "->" + targetName + '(' + UUID.randomUUID().toString()+')': linkName,
targetName, null, window, mode, isDurable, unsettled);
}
public SendingLinkEndpoint createSendingLinkEndpoint(final String linkName,
final Target target,
final Source source,
AcknowledgeMode mode,
Map<Binary, Outcome> unsettled,
final DeliveryStateHandler deliveryStateHandler)
{
SessionEndpoint endpoint = this.getEndpoint();
synchronized(endpoint.getLock())
{
SendingLinkEndpoint link = endpoint.createSendingLinkEndpoint(linkName, source, target,
unsettled, deliveryStateHandler);
switch(mode)
{
case ALO:
link.setSendingSettlementMode(SenderSettleMode.UNSETTLED);
link.setReceivingSettlementMode(ReceiverSettleMode.FIRST);
break;
case AMO:
link.setSendingSettlementMode(SenderSettleMode.SETTLED);
break;
case EO:
link.setSendingSettlementMode(SenderSettleMode.UNSETTLED);
link.setReceivingSettlementMode(ReceiverSettleMode.SECOND);
break;
}
link.attach();
return link;
}
}
public Receiver createReceiver(final String sourceAddr) throws ConnectionErrorException
{
return createReceiver(sourceAddr, null, AcknowledgeMode.ALO);
}
public Receiver createReceiver(final String queue, final AcknowledgeMode mode)
throws ConnectionErrorException
{
return createReceiver(queue, null, mode);
}
public Receiver createReceiver(final String queue, final AcknowledgeMode mode, String linkName)
throws ConnectionErrorException
{
return createReceiver(queue, null, mode, linkName);
}
public Receiver createReceiver(final String queue, final AcknowledgeMode mode, String linkName, boolean isDurable)
throws ConnectionErrorException
{
return createReceiver(queue, null, mode, linkName, isDurable);
}
public Receiver createReceiver(final String queue, final AcknowledgeMode mode, String linkName, boolean isDurable,
Map<Symbol, Filter> filters, Map<Binary, Outcome> unsettled)
throws ConnectionErrorException
{
return createReceiver(queue, null, mode, linkName, isDurable, filters, unsettled);
}
public Receiver createReceiver(final String queue, final AcknowledgeMode mode, String linkName,
boolean isDurable, Map<Binary, Outcome> unsettled)
throws ConnectionErrorException
{
return createReceiver(queue, null, mode, linkName, isDurable, unsettled);
}
private synchronized Receiver createReceiver(final String sourceAddr, DistributionMode mode)
throws ConnectionErrorException
{
return createReceiver(sourceAddr, mode, AcknowledgeMode.ALO);
}
private synchronized Receiver createReceiver(final String sourceAddr, DistributionMode mode, String linkName)
throws ConnectionErrorException
{
return createReceiver(sourceAddr, mode, AcknowledgeMode.ALO, linkName);
}
private synchronized Receiver createReceiver(final String sourceAddr, DistributionMode mode,
final AcknowledgeMode ackMode)
throws ConnectionErrorException
{
return createReceiver(sourceAddr, mode, ackMode, null);
}
private synchronized Receiver createReceiver(final String sourceAddr, DistributionMode mode,
final AcknowledgeMode ackMode, String linkName)
throws ConnectionErrorException
{
return createReceiver(sourceAddr,mode, ackMode, linkName, false);
}
private synchronized Receiver createReceiver(final String sourceAddr, DistributionMode mode,
final AcknowledgeMode ackMode, String linkName, boolean isDurable)
throws ConnectionErrorException
{
return createReceiver(sourceAddr, mode, ackMode, linkName, isDurable, null);
}
private synchronized Receiver createReceiver(final String sourceAddr, DistributionMode mode,
final AcknowledgeMode ackMode, String linkName, boolean isDurable,
Map<Binary, Outcome> unsettled)
throws ConnectionErrorException
{
return createReceiver(sourceAddr,mode,ackMode, linkName, isDurable, null, unsettled);
}
public synchronized Receiver createReceiver(final String sourceAddr, DistributionMode mode,
final AcknowledgeMode ackMode, String linkName, boolean isDurable,
Map<Symbol, Filter> filters, Map<Binary, Outcome> unsettled)
throws ConnectionErrorException
{
final Target target = new Target();
final Source source = new Source();
source.setAddress(sourceAddr);
source.setDistributionMode(mode);
source.setFilter(filters);
if(linkName == null)
{
linkName = sourceAddr + "-> (" + UUID.randomUUID().toString() + ")";
}
final Receiver receiver =
new Receiver(this, linkName,
target, source, ackMode, isDurable, unsettled);
_receivers.add(receiver);
return receiver;
}
public synchronized Receiver createCopyingReceiver(final String sourceAddr) throws ConnectionErrorException
{
return createReceiver(sourceAddr, StdDistMode.COPY);
}
public synchronized Receiver createMovingReceiver(final String sourceAddr) throws ConnectionErrorException
{
return createReceiver(sourceAddr, StdDistMode.MOVE);
}
public Receiver createTemporaryQueueReceiver() throws AmqpErrorException, ConnectionErrorException
{
Source source = new Source();
source.setDynamic(true);
final Receiver receiver = new Receiver(this, "tempSender"+UUID.randomUUID().toString(), new Target(),
source, AcknowledgeMode.ALO);
_receivers.add(receiver);
return receiver;
}
public Sender createTemporaryQueueSender() throws Sender.SenderCreationException, ConnectionClosedException
{
Target target = new Target();
target.setDynamic(true);
final Sender sender;
sender = new Sender(this, "tempSender"+ UUID.randomUUID().toString(), target,
new Source(), 0, AcknowledgeMode.ALO);
_senders.add(sender);
return sender;
}
public SessionEndpoint getEndpoint()
{
return _endpoint;
}
public synchronized void close()
{
try
{
for(Sender sender : new ArrayList<Sender>(_senders))
{
sender.close();
}
for(Receiver receiver : new ArrayList<Receiver>(_receivers))
{
receiver.detach();
}
if(_sessionLocalTC != null)
{
_sessionLocalTC.close();
}
_endpoint.end();
}
catch (Sender.SenderClosingException e)
{
// TODO
e.printStackTrace();
}
//TODO
}
void removeSender(Sender sender)
{
_senders.remove(sender);
}
void removeReceiver(Receiver receiver)
{
_receivers.remove(receiver);
}
public SectionEncoder getSectionEncoder()
{
return _sectionEncoder;
}
public SectionDecoder getSectionDecoder()
{
return _sectionDecoder;
}
public Transaction createSessionLocalTransaction()
{
TransactionController localController = getSessionLocalTransactionController();
return localController.beginTransaction();
}
private TransactionController getSessionLocalTransactionController()
{
if(_sessionLocalTC == null)
{
_sessionLocalTC = createSessionLocalTransactionController();
}
return _sessionLocalTC;
}
private TransactionController createSessionLocalTransactionController()
{
String name = "txnControllerLink";
SendingLinkEndpoint tcLinkEndpoint = _endpoint.createTransactionController(name, TxnCapability.LOCAL_TXN,
TxnCapability.MULTI_TXNS_PER_SSN);
tcLinkEndpoint.setReceivingSettlementMode(ReceiverSettleMode.FIRST);
tcLinkEndpoint.setSendingSettlementMode(SenderSettleMode.UNSETTLED);
tcLinkEndpoint.attach();
return new TransactionController(this, tcLinkEndpoint);
}
public Message receive()
{
while(getEndpoint().getState() == SessionState.ACTIVE)
{
synchronized (getEndpoint().getLock())
{
try
{
for(Receiver r : _receivers)
{
Message m = r.receive(false);
if(m != null)
return m;
}
wait();
}
catch (InterruptedException e)
{
}
}
}
return null;
}
public Connection getConnection()
{
return _connection;
}
public void awaitActive()
{
synchronized(getEndpoint().getLock())
{
while(!getEndpoint().isEnded() && !getEndpoint().isActive())
{
try
{
getEndpoint().getLock().wait();
}
catch (InterruptedException e)
{
e.printStackTrace(); //To change body of catch statement use File | Settings | File Templates.
}
}
}
}
public static interface SourceConfigurator
{
public void configureSource(final Source source);
}
}