Java Broker类代码示例

import org.apache.activemq.broker.Broker; //导入依赖的package包/类
@Override
public Broker installPlugin(Broker parent) throws Exception {

    logger.info("Initialize JWTAuthenticationPlugin");

    Set<Principal> groups = new HashSet();
    StringTokenizer iter = new StringTokenizer(this.defaultUserGroups, ",");
    while (iter.hasMoreTokens()) {
        String name = iter.nextToken().trim();
        groups.add(new GroupPrincipal(name));
    }
    JaasAuthenticationPlugin jaasAuthenticationPlugin = new JaasAuthenticationPlugin();
    jaasAuthenticationPlugin.setConfiguration(this.jaasConfiguration);
    jaasAuthenticationPlugin.setDiscoverLoginConfig(this.discoverLoginConfig);
    Broker jaasAuthenticationFallbackBroker = jaasAuthenticationPlugin.installPlugin(parent);
    return new JWTAuthenticationBroker(
            parent, jaasAuthenticationFallbackBroker, this.defaultUser,
            groups, this.masterSecretKey, this.tokenHeader);
}
 

import org.apache.activemq.broker.Broker; //导入依赖的package包/类
@Test(priority=3, expectedExceptions = CmsAuthException.class)
public void addConnectionNullCloudTest() throws Exception{
	Broker broker = mock(Broker.class);

	CMSClient oddClient = mock(CMSClient.class);
	CmsCISimple cmsCISimple = new CmsCISimple();
	Map<String, String> ciAttributes = new HashMap<String, String>(1);
	ciAttributes.put(AUTH_ATTR_KEY, AUTH_KEY_VALUE);
	cmsCISimple.setCiAttributes(ciAttributes);
	when(oddClient.getCloudCi(CLOUD_NAME, CLOUD_NAME)).thenReturn(
			cmsCISimple);
	when(oddClient.getCloudCi(anyString(), anyString())).thenReturn(null); // causes the exception
	OneopsAuthBroker oneopsAuthBroker = new OneopsAuthBroker(broker,oddClient);
	oneopsAuthBroker.addConnection(connectionContextMock, connectionBadInfo); //bad cloud, is error

}
 

import org.apache.activemq.broker.Broker; //导入依赖的package包/类
public Broker installPlugin(Broker broker) throws Exception {

        configStore.addVar(PluginVariables.REDIS_HOST, "localhost");
        configStore.addVar(PluginVariables.REDIS_PORT, "6379");
        configStore.addVar(PluginVariables.REDIS_DB, "0");
        configStore.addVar(PluginVariables.REDIS_CONNECTION_STRING, null);
        configStore.addVar(PluginVariables.WEBAUTH_SERVER_URL, "https://yourserver/webauth/user");
        configStore.addVar(PluginVariables.CA_CERTS_STORE_PATH, null);

        System.out.println(PluginVariables.WEBAUTH_SERVER_URL + ": " + configStore.getVar(PluginVariables.WEBAUTH_SERVER_URL));
        System.out.println(PluginVariables.CA_CERTS_STORE_PATH + ": " + configStore.getVar(PluginVariables.CA_CERTS_STORE_PATH));


        if (null != configStore.getVar(PluginVariables.CA_CERTS_STORE_PATH)) {
            System.setProperty("javax.net.ssl.trustStore", configStore.getVar(PluginVariables.CA_CERTS_STORE_PATH));
            System.setProperty("javax.net.ssl.trustAnchors", configStore.getVar(PluginVariables.CA_CERTS_STORE_PATH));
        }


        webAuthClient = new WebAuthClient(configStore.getVar(PluginVariables.WEBAUTH_SERVER_URL), logger);

        return new WebAuthBroker(broker, configStore, webAuthClient, logger);
    }
 

import org.apache.activemq.broker.Broker; //导入依赖的package包/类
public void configure(Broker broker,Queue queue) {
    baseConfiguration(broker,queue);
    if (dispatchPolicy != null) {
        queue.setDispatchPolicy(dispatchPolicy);
    }
    queue.setDeadLetterStrategy(getDeadLetterStrategy());
    queue.setMessageGroupMapFactory(getMessageGroupMapFactory());
    if (memoryLimit > 0) {
        queue.getMemoryUsage().setLimit(memoryLimit);
    }
    if (pendingQueuePolicy != null) {
        PendingMessageCursor messages = pendingQueuePolicy.getQueuePendingMessageCursor(broker,queue);
        queue.setMessages(messages);
    }

    queue.setUseConsumerPriority(isUseConsumerPriority());
    queue.setStrictOrderDispatch(isStrictOrderDispatch());
    queue.setOptimizedDispatch(isOptimizedDispatch());
    queue.setLazyDispatch(isLazyDispatch());
    queue.setTimeBeforeDispatchStarts(getTimeBeforeDispatchStarts());
    queue.setConsumersBeforeDispatchStarts(getConsumersBeforeDispatchStarts());
    queue.setAllConsumersExclusiveByDefault(isAllConsumersExclusiveByDefault());
}
 

import org.apache.activemq.broker.Broker; //导入依赖的package包/类
/**
 * @param broker Broker for this cursor
 * @param clientId clientId for this cursor
 * @param subscriberName subscriber name for this cursor
 * @param maxBatchSize currently ignored
 * @param subscription  subscription for this cursor
 */
public StoreDurableSubscriberCursor(Broker broker,String clientId, String subscriberName,int maxBatchSize, DurableTopicSubscription subscription) {
    super(AbstractPendingMessageCursor.isPrioritizedMessageSubscriber(broker,subscription));
    this.subscription=subscription;
    this.clientId = clientId;
    this.subscriberName = subscriberName;
    if (broker.getBrokerService().isPersistent()) {
        this.nonPersistent = new FilePendingMessageCursor(broker,clientId + subscriberName,this.prioritizedMessages);
    } else {
        this.nonPersistent = new VMPendingMessageCursor(this.prioritizedMessages);
    }

    this.nonPersistent.setMaxBatchSize(maxBatchSize);
    this.nonPersistent.setSystemUsage(systemUsage);
    this.storePrefetches.add(this.nonPersistent);

    if (prioritizedMessages) {
        setMaxAuditDepth(10*getMaxAuditDepth());
    }
}
 

import org.apache.activemq.broker.Broker; //导入依赖的package包/类
private boolean matchesSomeConsumer(final Broker broker, Message message, Destination dest) throws IOException {
    boolean matches = false;
    MessageEvaluationContext msgContext = new NonCachedMessageEvaluationContext();
    msgContext.setDestination(dest.getActiveMQDestination());
    msgContext.setMessageReference(message);
    List<Subscription> subs = dest.getConsumers();
    for (Subscription sub : subs) {
        if (sub.matches(message, msgContext)) {
            matches = true;
            break;

        }
    }
    if (matches == false && subs.size() == 0) {
        matches = tryMatchingCachedSubs(broker, dest, msgContext);
    }
    return matches;
}
 

import org.apache.activemq.broker.Broker; //导入依赖的package包/类
private boolean tryMatchingCachedSubs(final Broker broker, Destination dest, MessageEvaluationContext msgContext) {
    boolean matches = false;
    LOG.debug("No active consumer match found. Will try cache if configured...");

    //retrieve the specific plugin class and lookup the selector for the destination.
    final SubQueueSelectorCacheBroker cache = getSubQueueSelectorCacheBrokerPlugin(broker);

    if (cache != null) {
        final String selector = cache.getSelector(dest.getActiveMQDestination().getQualifiedName());
        if (selector != null) {
            try {
                final BooleanExpression expression = getExpression(selector);
                matches = expression.matches(msgContext);
            } catch (Exception e) {
                LOG.error(e.getMessage(), e);
            }
        }
    }
    return matches;
}
 

import org.apache.activemq.broker.Broker; //导入依赖的package包/类
public JWTAuthenticationBroker(
        Broker next, Broker jaasAuthenticationFallbackBroker, String defaultUser,
        Set<Principal> defaultUserGroups, String masterSecretKey, String tokenHeader) {

    super(next);
    this.jaasAuthenticationFallbackBroker = jaasAuthenticationFallbackBroker;
    this.defaultUser = defaultUser;
    this.defaultUserGroups = defaultUserGroups;
    this.masterSecretKey = masterSecretKey;
    this.tokenHeader = tokenHeader;
    logger.info("Initialize JWTAuthenticationBroker");
}
 

import org.apache.activemq.broker.Broker; //导入依赖的package包/类
/**
 * Instantiates a new auth broker filter.
 *
 * @param next the next
 * @param cms  the cms
 */
public OneopsAuthBroker(Broker next, CMSClient cms) {
    super(next);
    this.superpass = System.getenv("KLOOPZ_AMQ_PASS");
    if (this.superpass == null) {
        logger.error("The env var KLOOPZ_AMQ_PASS is not set");
        throw new CmsAuthException("The env var KLOOPZ_AMQ_PASS is not set");
    }
    this.cmsClient = cms;
    userMap.put("system", "*");
}
 

import org.apache.activemq.broker.Broker; //导入依赖的package包/类
@Test(priority=2)
/**
 * construct object ok. Note - it becomes the instance variable used from now on
 * succeeds because of the Env setup that's done off the bat
 */
public void constructorTest(){
	setEnvironmentVariable();

	Broker broker = mock(Broker.class);
	this.oneopsAuthBroker = new OneopsAuthBroker(broker, cmsClient); 
}
 

import org.apache.activemq.broker.Broker; //导入依赖的package包/类
public WebAuthBroker(Broker next, ConfigStore configStore, WebAuthClient webAuthClient, Logger logger) {
    super(next);
    this.configStore = configStore;
    this.webAuthClient = webAuthClient;
    this.logger = logger;

    prepareRedisConnectionString();
}
 

import org.apache.activemq.broker.Broker; //导入依赖的package包/类
@Override
public Broker installPlugin(final Broker next) throws Exception {

	LOG.info("Installing plugin...");

	return new BrokerFilter(next) {

		@Override
		public void send(final ProducerBrokerExchange producerExchange, final Message messageSend) throws Exception {

			try {
				final String destination = messageSend.getDestination().getPhysicalName();
				final IMessagePublisher<Message, K, V> publisher = messagePublisherProvider.getPublisher(destination);
				if (publisher == null) {
					LOG.debug("No kakfa-publisher found for destination={}. Will forward message.", destination);
					super.send(producerExchange, messageSend);
				} else {
					publisher.publish(messageSend, kafkaTemplate);
				}
			} catch (final Throwable t) {
				LOG.error("Exception occured", t);
				throw t;
			}
		}
	};

}
 

import org.apache.activemq.broker.Broker; //导入依赖的package包/类
/**
 * {@inheritDoc}
 */
@Override
public Broker installPlugin(Broker broker) {
    initialiseJaas();
    return new CommunoteJaasAuthenticationBroker(settingsDao, embeddedConfiguration, broker,
            getConfiguration(),
            getSslConfiguration());
}
 

import org.apache.activemq.broker.Broker; //导入依赖的package包/类
@Override
public Broker installPlugin(Broker broker) {

   HashMap<String, String> u = new HashMap<>();
   u.put("system", "manager");
   u.put("user", "password");
   u.put("guest", "password");

   Map<String, Set<Principal>> groups = new HashMap<>();
   groups.put("system", new HashSet<>(Arrays.asList(new Principal[]{ADMINS, USERS})));
   groups.put("user", new HashSet<>(Arrays.asList(new Principal[]{USERS})));
   groups.put("guest", new HashSet<>(Arrays.asList(new Principal[]{GUESTS})));

   return new SimpleAuthenticationBroker(broker, u, groups);
}
 

import org.apache.activemq.broker.Broker; //导入依赖的package包/类
private void sendToDestination(Broker broker, ProducerBrokerExchange context, Message message, ActiveMQDestination destination) throws Exception {
    if (broker.getDestinations(destination).isEmpty()) {
        broker.getBrokerService().getRegionBroker().addDestination(broker.getAdminConnectionContext(), destination,true);   
    }

    for (Destination dest : broker.getDestinations(destination)) {
        logger.debug("Following search sending request to Destination " + dest.getName());
        dest.send(context, message.copy());
    }    
}