Java ConnectionContext类代码示例

import org.apache.activemq.broker.ConnectionContext; //导入依赖的package包/类
/**
 * Add new message consumer.
 *
 * @param context
 * @param info
 * @return
 * @throws Exception
 * @see org.apache.activemq.broker.BrokerFilter#addConsumer(org.apache.activemq.broker.ConnectionContext, org.apache.activemq.command.ConsumerInfo)
 */
public Subscription addConsumer(ConnectionContext context, ConsumerInfo info) throws Exception {

    ActiveMQDestination dest = info.getDestination();
    Connection conn = context.getConnection();
    if (dest != null) {
        String destName = info.getDestination().getPhysicalName();
        String clientId = context.getClientId();
        String allowedDest = userMap.get(clientId);

        logger.info(">>> Got Consumer Add request { Destination: " + destName
                + ", Remote Address: " + conn.getRemoteAddress()
                + ", ClientID: " + clientId
                + " }");
        if (allowedDest != null && (allowedDest.equals("*") || allowedDest.equals(destName) || destName.startsWith("ActiveMQ"))) {
            logger.info(">>> Subscription allowed");
        } else {
            logger.error(">>> Destination not allowed. Subscription denied!");
            throw new CmsAuthException(">>> Subscription denied!");
        }
    } else {
        logger.error("<<< Got Consumer Add request from Remote Address:" + conn.getRemoteAddress() + ". But destination is NULL.");
    }
    return super.addConsumer(context, info);
}
 

import org.apache.activemq.broker.ConnectionContext; //导入依赖的package包/类
/**
 * Add message producer.
 *
 * @param context
 * @param info
 * @throws Exception
 * @see org.apache.activemq.broker.BrokerFilter#addProducer(org.apache.activemq.broker.ConnectionContext, org.apache.activemq.command.ProducerInfo)
 */
public void addProducer(ConnectionContext context, ProducerInfo info) throws Exception {

    Connection conn = context.getConnection();
    ActiveMQDestination dest = info.getDestination();
    if (dest != null) {
        String destName = dest.getPhysicalName();
        String clientId = context.getClientId();

        logger.info(">>> Got Producer Add request { Destination: " + destName
                + ", Remote Address: " + conn.getRemoteAddress()
                + ", ClientID: " + clientId
                + " }");

        String allowedDest = userMap.get(context.getClientId());
        if (allowedDest != null && (allowedDest.equals("*") || "controller.response".equals(destName))) {
            logger.info("<<< Producer allowed");
        } else {
            logger.error("<<< Destination not allowed. Producer denied!");
            throw new CmsAuthException("<<< Producer denied!");
        }
    } else {
        logger.error("<<< Got Producer Add request from Remote Address:" + conn.getRemoteAddress() + ". But destination is NULL.");
    }
    super.addProducer(context, info);
}
 

import org.apache.activemq.broker.ConnectionContext; //导入依赖的package包/类
@Test(priority = 6, expectedExceptions = RuntimeException.class)
public void addProducerTestProducerDenied() {

	//create a mock ConnectionContext which was not used when setting up user map
	ConnectionContext ccForbidden = mock(ConnectionContext.class);
	when(ccForbidden.getClientId()).thenReturn("this-is-not-in-user-map");
	 final Connection connectionMock = mock(Connection.class);
		when(connectionMock.getRemoteAddress()).thenReturn(MOCK_REMOTE_ADDR);
			when(ccForbidden.getConnection()).thenAnswer(new Answer<Connection>() {
		@Override
		public Connection answer(InvocationOnMock invocation)
				throws Throwable {
			return connectionMock;
		}
	});
	
	try {
		this.oneopsAuthBroker.addProducer(ccForbidden, producerInfo);
	} catch (Exception e) {
		logger.warn("caught exception, make sure Broker is mocked",e);
		throw new RuntimeException(e);
	}
}
 

import org.apache.activemq.broker.ConnectionContext; //导入依赖的package包/类
/**
 * Check if the connection is internal
 * 
 * @param context
 *            the context
 * @param info
 *            the connection info
 * @return true if the connection is embedded
 */
private boolean isEmbeddedConnection(ConnectionContext context, ConnectionInfo info) {
    if (embeddedConfiguration != null) {
        // check the username and password of the connection
        if (embeddedConfiguration.getUsername().equals(info.getUserName())
                && embeddedConfiguration.getPassword()
                        .equals(info.getPassword())) {

            // check if its local
            if (CommunoteJaasLoginModule.isAddressLocal(context
                    .getConnection().getRemoteAddress())) {

                // it is as an embedded connection
                return true;
            }
        }
    }
    return false;
}
 

import org.apache.activemq.broker.ConnectionContext; //导入依赖的package包/类
protected Subscription createSubscription(ConnectionContext context, ConsumerInfo info) throws JMSException {
    if (info.isDurable()) {
        throw new JMSException("A durable subscription cannot be created for a temporary topic.");
    }
    try {
        TopicSubscription answer = new TopicSubscription(broker, context, info, usageManager);
        // lets configure the subscription depending on the destination
        ActiveMQDestination destination = info.getDestination();
        if (destination != null && broker.getDestinationPolicy() != null) {
            PolicyEntry entry = broker.getDestinationPolicy().getEntryFor(destination);
            if (entry != null) {
                entry.configure(broker, usageManager, answer);
            }
        }
        answer.init();
        return answer;
    } catch (Exception e) {
        LOG.error("Failed to create TopicSubscription ", e);
        JMSException jmsEx = new JMSException("Couldn't create TopicSubscription");
        jmsEx.setLinkedException(e);
        throw jmsEx;
    }
}
 

import org.apache.activemq.broker.ConnectionContext; //导入依赖的package包/类
@Override
public void addProducer(ConnectionContext context, ProducerInfo info) throws Exception {
    final SecurityContext securityContext = checkSecurityContext(context);

    if (!securityContext.isBrokerContext() && info.getDestination() != null) {

        Set<?> allowedACLs = null;
        if (!info.getDestination().isTemporary()) {
            allowedACLs = authorizationMap.getWriteACLs(info.getDestination());
        } else {
            allowedACLs = authorizationMap.getTempDestinationWriteACLs();
        }
        if (allowedACLs != null && !securityContext.isInOneOf(allowedACLs)) {
            throw new SecurityException("User " + securityContext.getUserName() + " is not authorized to write to: " + info.getDestination());
        }
        securityContext.getAuthorizedWriteDests().put(info.getDestination(), info.getDestination());
    }

    super.addProducer(context, info);
}
 

import org.apache.activemq.broker.ConnectionContext; //导入依赖的package包/类
/**
 * Overriding removeConnection to make sure the security context is cleaned.
 */
@Override
public void removeConnection(ConnectionContext context, ConnectionInfo info, Throwable error) throws Exception {
    boolean isSSL;
    Connector connector = context.getConnector();
    if (connector instanceof TransportConnector) {
        TransportConnector transportConnector = (TransportConnector) connector;
        isSSL = (transportConnector.getServer() instanceof SslTransportServer);
    } else {
        isSSL = false;
    }
    super.removeConnection(context, info, error);
    if (isSSL) {
        this.sslBroker.removeConnection(context, info, error);
    } else {
        this.nonSslBroker.removeConnection(context, info, error);
    }
}
 

import org.apache.activemq.broker.ConnectionContext; //导入依赖的package包/类
@Override
public void messageExpired(ConnectionContext context, Subscription subs, MessageReference reference) {
    LOG.debug("message expired: {}", reference);
    broker.messageExpired(context, reference, subs);
    destinationStatistics.getExpired().increment();
    try {
        removeMessage(context, subs, (QueueMessageReference) reference);
        messagesLock.writeLock().lock();
        try {
            messages.rollback(reference.getMessageId());
        } finally {
            messagesLock.writeLock().unlock();
        }
    } catch (IOException e) {
        LOG.error("Failed to remove expired Message from the store ", e);
    }
}
 

import org.apache.activemq.broker.ConnectionContext; //导入依赖的package包/类
/**
 * @param message
 * @throws IOException
 */
void addMessage(ConnectionContext context, final MessageStore destination, final Message message)
        throws IOException {

    if (message.getTransactionId() != null) {
        if (message.getTransactionId().isXATransaction() || theStore.isConcurrentStoreAndDispatchTransactions() == false) {
            destination.addMessage(context, message);
        } else {
            Tx tx = getTx(message.getTransactionId());
            tx.add(new AddMessageCommand(context) {
                @Override
                public Message getMessage() {
                    return message;
                }
                @Override
                public Future<Object> run(ConnectionContext ctx) throws IOException {
                    destination.addMessage(ctx, message);
                    return AbstractMessageStore.FUTURE;
                }

            });
        }
    } else {
        destination.addMessage(context, message);
    }
}
 

import org.apache.activemq.broker.ConnectionContext; //导入依赖的package包/类
@Override
public void removeSubscription(ConnectionContext context, RemoveSubscriptionInfo info) throws Exception {
    SubscriptionKey key = new SubscriptionKey(context.getClientId(), info.getSubscriptionName());
    DurableTopicSubscription sub = ((TopicRegion)((RegionBroker)next).getTopicRegion()).getDurableSubscription(key);

    super.removeSubscription(context, info);

    if (sub == null) {
        LOG.warn("We cannot send an advisory message for a durable sub removal when we don't know about the durable sub");
        return;
    }

    ActiveMQDestination dest = sub.getConsumerInfo().getDestination();

    // Don't advise advisory topics.
    if (!AdvisorySupport.isAdvisoryTopic(dest)) {
        ActiveMQTopic topic = AdvisorySupport.getConsumerAdvisoryTopic(dest);
        fireConsumerAdvisory(context, dest, topic, info);
    }

}
 

import org.apache.activemq.broker.ConnectionContext; //导入依赖的package包/类
@Override
public void messageExpired(ConnectionContext context, MessageReference messageReference, Subscription subscription) {
    super.messageExpired(context, messageReference, subscription);
    try {
        if(!messageReference.isAdvisory()) {
            ActiveMQTopic topic = AdvisorySupport.getExpiredMessageTopic(messageReference.getMessage().getDestination());
            Message payload = messageReference.getMessage().copy();
            payload.clearBody();
            ActiveMQMessage advisoryMessage = new ActiveMQMessage();
            advisoryMessage.setStringProperty(AdvisorySupport.MSG_PROPERTY_MESSAGE_ID, payload.getMessageId().toString());
            fireAdvisory(context, topic, payload, null, advisoryMessage);
        }
    } catch (Exception e) {
        handleFireFailure("expired", e);
    }
}
 

import org.apache.activemq.broker.ConnectionContext; //导入依赖的package包/类
@Override
public void networkBridgeStopped(BrokerInfo brokerInfo) {
    try {
     if (brokerInfo != null) {
         ActiveMQMessage advisoryMessage = new ActiveMQMessage();
         advisoryMessage.setBooleanProperty("started", false);
         networkBridges.remove(brokerInfo);

         ActiveMQTopic topic = AdvisorySupport.getNetworkBridgeAdvisoryTopic();

         ConnectionContext context = new ConnectionContext();
         context.setSecurityContext(SecurityContext.BROKER_SECURITY_CONTEXT);
         context.setBroker(getBrokerService().getBroker());
         fireAdvisory(context, topic, brokerInfo, null, advisoryMessage);
     }
    } catch (Exception e) {
        handleFireFailure("network bridge stopped", e);
    }
}
 

import org.apache.activemq.broker.ConnectionContext; //导入依赖的package包/类
/**
 * @returns the ObjectName of the Connection that created this subscription
 */
@Override
public ObjectName getConnection() {
    ObjectName result = null;

    if (clientId != null && subscription != null) {
        ConnectionContext ctx = subscription.getContext();
        if (ctx != null && ctx.getBroker() != null && ctx.getBroker().getBrokerService() != null) {
            BrokerService service = ctx.getBroker().getBrokerService();
            ManagementContext managementCtx = service.getManagementContext();
            if (managementCtx != null) {

                try {
                    ObjectName query = createConnectionQuery(managementCtx, service.getBrokerName());
                    Set<ObjectName> names = managementCtx.queryNames(query, null);
                    if (names.size() == 1) {
                        result = names.iterator().next();
                    }
                } catch (Exception e) {
                }
            }
        }
    }
    return result;
}
 

import org.apache.activemq.broker.ConnectionContext; //导入依赖的package包/类
public void acknowledge(final TopicMessageStore destination, final String clientId, final String subscriptionName,
                       final MessageId messageId, final MessageAck ack) throws IOException {
    if (doingRecover) {
        return;
    }

    if (ack.isInTransaction()) {
        Tx tx = getTx(ack.getTransactionId());
        tx.add(new RemoveMessageCommand() {
            public MessageAck getMessageAck() {
                return ack;
            }

            public void run(ConnectionContext ctx) throws IOException {
                destination.acknowledge(ctx, clientId, subscriptionName, messageId, ack);
            }

            @Override
            public MessageStore getMessageStore() {
                return destination;
            }
        });
    } else {
        destination.acknowledge(null, clientId, subscriptionName, messageId, ack);
    }
}
 

import org.apache.activemq.broker.ConnectionContext; //导入依赖的package包/类
@Override
public void slowConsumer(ConnectionContext context, Subscription subs) {
    if (maxSlowCount < 0 && maxSlowDuration < 0) {
        // nothing to do
        LOG.info("no limits set, slowConsumer strategy has nothing to do");
        return;
    }

    if (taskStarted.compareAndSet(false, true)) {
        scheduler.executePeriodically(this, checkPeriod);
    }

    if (!slowConsumers.containsKey(subs)) {
        slowConsumers.put(subs, new SlowConsumerEntry(context));
    } else if (maxSlowCount > 0) {
        slowConsumers.get(subs).slow();
    }
}