/**
* 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.activemq.usecases;
import java.net.URI;
import java.util.concurrent.Executor;
import java.util.concurrent.Executors;
import org.apache.activemq.JmsMultipleBrokersTestSupport;
import org.apache.activemq.broker.BrokerService;
import org.apache.activemq.network.DiscoveryNetworkConnector;
import org.apache.activemq.network.NetworkConnector;
import org.apache.activemq.util.SocketProxy;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
public class NetworkAsyncStartTest extends JmsMultipleBrokersTestSupport {
private static final Logger LOG = LoggerFactory.getLogger(NetworkAsyncStartTest.class);
private String brokerBDomain = "localhost:61617";
private String brokerCDomain = "localhost:61618";
int bridgeCount=0;
public static final String KEYSTORE_TYPE = "jks";
public static final String PASSWORD = "password";
public static final String SERVER_KEYSTORE = "src/test/resources/server.keystore";
public static final String TRUST_KEYSTORE = "src/test/resources/client.keystore";
static {
System.setProperty("javax.net.ssl.trustStore", TRUST_KEYSTORE);
System.setProperty("javax.net.ssl.trustStorePassword", PASSWORD);
System.setProperty("javax.net.ssl.trustStoreType", KEYSTORE_TYPE);
System.setProperty("javax.net.ssl.keyStore", SERVER_KEYSTORE);
System.setProperty("javax.net.ssl.keyStoreType", KEYSTORE_TYPE);
System.setProperty("javax.net.ssl.keyStorePassword", PASSWORD);
}
public void testAsyncNetworkStartupTcp() throws Exception {
testAsyncNetworkStartup("tcp");
}
public void testAsyncNetworkStartupWithSlowConnectionCreationTcp() throws Exception {
testAsyncNetworkStartupWithSlowConnectionCreation("tcp");
}
public void testAsyncNetworkStartupNio() throws Exception {
testAsyncNetworkStartup("nio");
}
public void testAsyncNetworkStartupWithSlowConnectionCreationNio() throws Exception {
testAsyncNetworkStartupWithSlowConnectionCreation("nio");
}
public void testAsyncNetworkStartupAuto() throws Exception {
testAsyncNetworkStartup("auto");
}
public void testAsyncNetworkStartupWithSlowConnectionCreationAuto() throws Exception {
testAsyncNetworkStartupWithSlowConnectionCreation("auto");
}
public void testAsyncNetworkStartupAutoNio() throws Exception {
testAsyncNetworkStartup("auto+nio");
}
public void testAsyncNetworkStartupWithSlowConnectionCreationAutoNio() throws Exception {
testAsyncNetworkStartupWithSlowConnectionCreation("auto+nio");
}
public void testAsyncNetworkStartupSsl() throws Exception {
testAsyncNetworkStartup("ssl");
}
public void testAsyncNetworkStartupWithSlowConnectionCreationSsl() throws Exception {
testAsyncNetworkStartupWithSlowConnectionCreation("ssl");
}
public void testAsyncNetworkStartupAutoSsl() throws Exception {
testAsyncNetworkStartup("auto+ssl");
}
public void testAsyncNetworkStartupWithSlowConnectionCreationAutoSsl() throws Exception {
testAsyncNetworkStartupWithSlowConnectionCreation("auto+ssl");
}
public void testAsyncNetworkStartupNioSsl() throws Exception {
testAsyncNetworkStartup("nio+ssl");
}
public void testAsyncNetworkStartupWithSlowConnectionCreationNioSsl() throws Exception {
testAsyncNetworkStartupWithSlowConnectionCreation("nio+ssl");
}
public void testAsyncNetworkStartupAutoNioSsl() throws Exception {
testAsyncNetworkStartup("auto+nio+ssl");
}
public void testAsyncNetworkStartupWithSlowConnectionCreationAutoNioSsl() throws Exception {
testAsyncNetworkStartupWithSlowConnectionCreation("auto+nio+ssl");
}
protected void testAsyncNetworkStartup(String transport) throws Exception {
String brokerBUri = transport + "://" + brokerBDomain;
String brokerCUri = transport + "://" + brokerCDomain;
BrokerService brokerA = brokers.get("BrokerA").broker;
bridgeBroker(brokerA, brokerBUri);
bridgeBroker(brokerA, brokerCUri);
LOG.info("starting A, no blocking on failed network connectors");
brokerA.start();
LOG.info("starting C transport connector");
BrokerService brokerC = brokers.get("BrokerC").broker;
brokerC.addConnector(brokerCUri);
brokerC.start();
assertTrue("got bridge to C", waitForBridgeFormation(brokerA, 1, 1));
LOG.info("Got bridge A->C");
LOG.info("starting B transport connector");
BrokerService brokerB = brokers.get("BrokerB").broker;
brokerB.addConnector(brokerBUri);
brokerB.start();
assertTrue("got bridge to B", waitForBridgeFormation(brokerA, 1, 0));
assertTrue("got bridge to B&C", waitForBridgeFormation(brokerA, 1, 1));
}
protected void testAsyncNetworkStartupWithSlowConnectionCreation(String transport) throws Exception {
String brokerBUri = transport + "://" + brokerBDomain;
String brokerCUri = transport + "://" + brokerCDomain;
final BrokerService brokerA = brokers.get("BrokerA").broker;
SocketProxy proxyToB = new SocketProxy();
// don't accept any connections so limited to one connection with backlog
proxyToB.setPauseAtStart(true);
proxyToB.setAcceptBacklog(1);
proxyToB.setTarget(new URI(brokerBUri));
proxyToB.open();
bridgeBroker(brokerA, proxyToB.getUrl().toString());
bridgeBroker(brokerA, proxyToB.getUrl().toString());
bridgeBroker(brokerA, proxyToB.getUrl().toString());
bridgeBroker(brokerA, proxyToB.getUrl().toString());
bridgeBroker(brokerA, proxyToB.getUrl().toString());
bridgeBroker(brokerA, proxyToB.getUrl().toString());
bridgeBroker(brokerA, proxyToB.getUrl().toString());
bridgeBroker(brokerA, brokerCUri);
Executor e = Executors.newCachedThreadPool();
e.execute(new Runnable() {
@Override
public void run() {
LOG.info("starting A");
try {
brokerA.setNetworkConnectorStartAsync(true);
brokerA.start();
} catch (Exception e) {
LOG.error("start failed", e);
}
}
});
LOG.info("starting transport connector on C");
BrokerService brokerC = brokers.get("BrokerC").broker;
brokerC.addConnector(brokerCUri);
brokerC.start();
final long maxWaitMillis = 20*1000;
assertTrue("got bridge to C in 10 seconds", waitForBridgeFormation(brokerA, 1, 7, maxWaitMillis));
}
private void bridgeBroker(BrokerService localBroker, String remoteURI) throws Exception {
String uri = "static:(" + remoteURI + ")";
NetworkConnector connector = new DiscoveryNetworkConnector(new URI(uri));
connector.setName("bridge-" + bridgeCount++);
localBroker.addNetworkConnector(connector);
}
@Override
public void setUp() throws Exception {
super.setAutoFail(true);
super.setUp();
// initially with no tcp transport connector
createBroker(new URI("broker:()BrokerA?persistent=false&useJmx=false"));
createBroker(new URI("broker:()BrokerB?persistent=false&useJmx=false"));
createBroker(new URI("broker:()BrokerC?persistent=false&useJmx=false"));
}
}