Examples with BrokerService org.apache.activemq.broker.BrokerService used on opensource projects

Example 36 with BrokerService

use of org.apache.activemq.broker.BrokerService in project ignite by apache.

the class IgniteMqttStreamerTest method beforeTest.

/**
     * @throws Exception If failed.
     */
@Before
@SuppressWarnings("unchecked")
public void beforeTest() throws Exception {
    grid().<Integer, String>getOrCreateCache(defaultCacheConfiguration());
    // find an available local port
    try (ServerSocket ss = new ServerSocket(0)) {
        port = ss.getLocalPort();
    }
    // create the broker
    broker = new BrokerService();
    broker.setDeleteAllMessagesOnStartup(true);
    broker.setPersistent(false);
    broker.setPersistenceAdapter(null);
    broker.setPersistenceFactory(null);
    PolicyMap plcMap = new PolicyMap();
    PolicyEntry plc = new PolicyEntry();
    plc.setQueuePrefetch(1);
    broker.setDestinationPolicy(plcMap);
    broker.getDestinationPolicy().setDefaultEntry(plc);
    broker.setSchedulerSupport(false);
    // add the MQTT transport connector to the broker
    broker.addConnector("mqtt://localhost:" + port);
    broker.setStartAsync(false);
    broker.start(true);
    // create the broker URL
    brokerUrl = "tcp://localhost:" + port;
    // create the client and connect
    client = new MqttClient(brokerUrl, UUID.randomUUID().toString(), new MemoryPersistence());
    client.connect();
    // create mqtt streamer
    dataStreamer = grid().dataStreamer(DEFAULT_CACHE_NAME);
    streamer = createMqttStreamer(dataStreamer);
}

Example 37 with BrokerService

use of org.apache.activemq.broker.BrokerService in project tomee by apache.

the class ActiveMQ5Factory method createBroker.

@Override
public synchronized BrokerService createBroker(final URI brokerURI) throws Exception {
    Logger.getInstance(LogCategory.OPENEJB_STARTUP, ActiveMQ5Factory.class).getChildLogger("service").info("ActiveMQ5Factory creating broker");
    BrokerService broker = brokers.get(brokerURI);
    if (null == broker || !broker.isStarted()) {
        final Properties properties = getLowerCaseProperties();
        final URISupport.CompositeData compositeData = URISupport.parseComposite(new URI(brokerURI.getRawSchemeSpecificPart()));
        final Map<String, String> params = new HashMap<String, String>(compositeData.getParameters());
        final PersistenceAdapter persistenceAdapter;
        if ("true".equals(params.remove("usekahadb"))) {
            persistenceAdapter = createPersistenceAdapter("org.apache.activemq.store.kahadb.KahaDBPersistenceAdapter", "kahadb", params);
        } else if ("true".equals(params.remove("useleveldb"))) {
            persistenceAdapter = createPersistenceAdapter("org.apache.activemq.store.leveldb.LevelDBPersistenceAdapter", "leveldb", params);
        } else if (params.get("persistenceadapter") != null) {
            final String adapter = params.remove("persistenceadapter");
            persistenceAdapter = createPersistenceAdapter(adapter, "persistence", params);
        } else {
            persistenceAdapter = null;
        }
        final BrokerPlugin[] plugins = createPlugins(params);
        final URI uri = new URI(cleanUpUri(brokerURI.getRawSchemeSpecificPart(), compositeData.getParameters(), params));
        broker = "broker".equals(uri.getScheme()) ? newDefaultBroker(uri) : BrokerFactory.createBroker(uri);
        if (plugins != null) {
            broker.setPlugins(plugins);
        }
        brokers.put(brokerURI, broker);
        if (persistenceAdapter != null) {
            broker.setPersistenceAdapter(persistenceAdapter);
            // if user didn't set persistent to true then setPersistenceAdapter() alone is ignored so forcing it with the factory
            broker.setPersistenceFactory(new ProvidedPersistenceAdapterPersistenceAdapterFactory(persistenceAdapter));
            broker.setPersistent(true);
            tomeeConfig(broker);
        } else {
            final boolean notXbean = !uri.getScheme().toLowerCase(Locale.ENGLISH).startsWith("xbean");
            if (notXbean) {
                Object value = properties.get("datasource");
                if (String.class.isInstance(value) && value.toString().length() == 0) {
                    value = null;
                }
                final DataSource dataSource;
                if (value != null) {
                    if (DataSource.class.isInstance(value)) {
                        dataSource = DataSource.class.cast(value);
                    } else if (String.class.isInstance(value)) {
                        final String resouceId = String.class.cast(value);
                        try {
                            final ContainerSystem containerSystem = SystemInstance.get().getComponent(ContainerSystem.class);
                            final Context context = containerSystem.getJNDIContext();
                            final Object obj = context.lookup("openejb/Resource/" + resouceId);
                            if (!(obj instanceof DataSource)) {
                                throw new IllegalArgumentException("Resource with id " + resouceId + " is not a DataSource, but is " + obj.getClass().getName());
                            }
                            dataSource = (DataSource) obj;
                        } catch (final NamingException e) {
                            throw new IllegalArgumentException("Unknown datasource " + resouceId);
                        }
                    } else {
                        throw new IllegalArgumentException("Unexpected datasource definition: " + value);
                    }
                } else {
                    dataSource = null;
                }
                if (null != dataSource) {
                    final JDBCPersistenceAdapter adapter = new JDBCPersistenceAdapter();
                    if (properties.containsKey("usedatabaselock")) {
                        //This must be false for hsqldb
                        adapter.setUseLock(Boolean.parseBoolean(properties.getProperty("usedatabaselock", "true")));
                    }
                    adapter.setDataSource(dataSource);
                    broker.setPersistent(true);
                    broker.setPersistenceAdapter(adapter);
                } else {
                    broker.setPersistenceAdapter(new MemoryPersistenceAdapter());
                }
                tomeeConfig(broker);
            }
        }
        //We must close the broker
        broker.setUseShutdownHook(false);
        broker.setSystemExitOnShutdown(false);
        broker.setStartAsync(false);
        final BrokerService bs = broker;
        final Thread start = new Thread("ActiveMQFactory start and checkpoint") {

            @Override
            public void run() {
                Thread.currentThread().setContextClassLoader(ActiveMQResourceAdapter.class.getClassLoader());
                try {
                    //Start before returning - this is known to be safe.
                    if (!bs.isStarted()) {
                        Logger.getInstance(LogCategory.OPENEJB_STARTUP, ActiveMQ5Factory.class).getChildLogger("service").info("Starting ActiveMQ BrokerService");
                        bs.start();
                    }
                    bs.waitUntilStarted();
                    //Force a checkpoint to initialize pools
                    Logger.getInstance(LogCategory.OPENEJB_STARTUP, ActiveMQ5Factory.class).getChildLogger("service").info("Starting ActiveMQ checkpoint");
                    bs.getPersistenceAdapter().checkpoint(true);
                    started.set(true);
                } catch (final Throwable t) {
                    throwable = t;
                }
            }
        };
        /*
             * An application may require immediate access to JMS. So we need to block here until the service
             * has started. How long ActiveMQ requires to actually create a broker is unpredictable.
             *
             * A broker in OpenEJB is usually a wrapper for an embedded ActiveMQ server service. The broker configuration
             * allows the definition of a remote ActiveMQ server, in which case startup is not an issue as the broker is
             * basically a client.
             *
             * If the broker is local and the message store contains millions of messages then the startup time is obviously going to
             * be longer as these need to be indexed by ActiveMQ.
             *
             * A balanced timeout will always be use case dependent.
            */
        int timeout = 30000;
        try {
            timeout = Integer.parseInt(properties.getProperty("startuptimeout", "30000"));
            Logger.getInstance(LogCategory.OPENEJB_STARTUP, ActiveMQ5Factory.class).getChildLogger("service").info("Using ActiveMQ startup timeout of " + timeout + "ms");
        } catch (final Throwable e) {
        //Ignore
        }
        start.setDaemon(true);
        start.start();
        try {
            start.join(timeout);
        } catch (final InterruptedException e) {
        //Ignore
        }
        if (null != throwable) {
            Logger.getInstance(LogCategory.OPENEJB_STARTUP, ActiveMQ5Factory.class).getChildLogger("service").error("ActiveMQ failed to start broker", throwable);
        } else if (started.get()) {
            Logger.getInstance(LogCategory.OPENEJB_STARTUP, ActiveMQ5Factory.class).getChildLogger("service").info("ActiveMQ broker started");
        } else {
            Logger.getInstance(LogCategory.OPENEJB_STARTUP, ActiveMQ5Factory.class).getChildLogger("service").warning("ActiveMQ failed to start broker within " + timeout + " seconds - It may be unusable");
        }
    }
    return broker;
}

Example 38 with BrokerService

use of org.apache.activemq.broker.BrokerService in project tomee by apache.

the class ActiveMQ5FactoryTest method setPlugins.

@Test
public void setPlugins() throws Exception {
    final URI brokerURI = new URI("amq5factory:broker:(tcp://localhost:" + getNextAvailablePort() + ")?" + "amq.plugins=jaas&" + "jaas.class=" + JaasAuthenticationPlugin.class.getName() + "&" + "jaas.discoverLoginConfig=false");
    final BrokerService bs = new ActiveMQ5Factory().createBroker(brokerURI);
    bs.stop();
    ActiveMQ5Factory.brokers.remove(brokerURI);
    assertNotNull(bs.getPlugins());
    assertEquals(1, bs.getPlugins().length);
    assertTrue(JaasAuthenticationPlugin.class.isInstance(bs.getPlugins()[0]));
    // default is true
    assertFalse(JaasAuthenticationPlugin.class.cast(bs.getPlugins()[0]).isDiscoverLoginConfig());
}

Example 39 with BrokerService

use of org.apache.activemq.broker.BrokerService in project tomee by apache.

the class OpenEjbBrokerFactoryTest method testBrokerDoubleCreate.

public void testBrokerDoubleCreate() throws Exception {
    BrokerService broker = BrokerFactory.createBroker(new URI(getBrokerUri("broker:(tcp://localhost:" + brokerPort + ")?useJmx=false")));
    stopBroker(broker);
    broker = BrokerFactory.createBroker(new URI(getBrokerUri("broker:(tcp://localhost:" + brokerPort + ")?useJmx=false")));
    stopBroker(broker);
}

Example 40 with BrokerService

use of org.apache.activemq.broker.BrokerService in project tomee by apache.

the class OpenEjbBrokerFactoryTest method testDirectDataSource.

public void testDirectDataSource() throws Exception {
    final Properties properties = new Properties();
    final JDBCDataSource dataSource = new JDBCDataSource();
    dataSource.setDatabase("jdbc:hsqldb:mem:testdb" + System.currentTimeMillis());
    dataSource.setUser("sa");
    dataSource.setPassword("");
    dataSource.getConnection().close();
    properties.put("DataSource", dataSource);
    properties.put("UseDatabaseLock", "false");
    properties.put("StartupTimeout", "10000");
    ActiveMQFactory.setThreadProperties(properties);
    BrokerService broker = null;
    try {
        broker = BrokerFactory.createBroker(new URI(getBrokerUri("broker:(tcp://localhost:" + brokerPort + ")?useJmx=false")));
        assertNotNull("broker is null", broker);
        final PersistenceAdapter persistenceAdapter = broker.getPersistenceAdapter();
        assertNotNull("persistenceAdapter is null", persistenceAdapter);
        assertTrue("persistenceAdapter should be an instance of JDBCPersistenceAdapter", persistenceAdapter instanceof JDBCPersistenceAdapter);
        final JDBCPersistenceAdapter jdbcPersistenceAdapter = (JDBCPersistenceAdapter) persistenceAdapter;
        assertSame(dataSource, jdbcPersistenceAdapter.getDataSource());
    } finally {
        stopBroker(broker);
        ActiveMQFactory.setThreadProperties(null);
    }
}