Examples with PersistenceAdapter org.apache.activemq.store.PersistenceAdapter used on opensource projects

Example 1 with PersistenceAdapter

use of org.apache.activemq.store.PersistenceAdapter in project tomee by apache.

the class OpenEjbBrokerFactoryTest method testLookupDataSource.

public void testLookupDataSource() 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();
    MockInitialContextFactory.install(Collections.singletonMap("openejb/Resource/TestDs", dataSource));
    assertSame(dataSource, new InitialContext().lookup("openejb/Resource/TestDs"));
    final CoreContainerSystem containerSystem = new CoreContainerSystem(new IvmJndiFactory());
    containerSystem.getJNDIContext().bind("openejb/Resource/TestDs", dataSource);
    SystemInstance.get().setComponent(ContainerSystem.class, containerSystem);
    properties.put("DataSource", "TestDs");
    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);
    }
}

Example 2 with PersistenceAdapter

use of org.apache.activemq.store.PersistenceAdapter in project tomee by apache.

the class OpenEjbBrokerFactoryTest method testNoDataSource.

public void testNoDataSource() throws Exception {
    final BrokerService 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 MemoryPersistenceAdapter", persistenceAdapter instanceof MemoryPersistenceAdapter);
    stopBroker(broker);
}

Example 3 with PersistenceAdapter

use of org.apache.activemq.store.PersistenceAdapter in project tomee by apache.

the class ActiveMQ5Factory method createPersistenceAdapter.

private static PersistenceAdapter createPersistenceAdapter(final String clazz, final String prefix, final Map<String, String> params) throws IllegalAccessException, InvocationTargetException, ClassNotFoundException, InstantiationException {
    Class<?> aClass = Thread.currentThread().getContextClassLoader().loadClass(clazz);
    final PersistenceAdapter persistenceAdapter = PersistenceAdapter.class.cast(aClass.newInstance());
    while (aClass != null) {
        for (final Method m : aClass.getDeclaredMethods()) {
            if (m.getName().startsWith("set") && m.getParameterTypes().length == 1 && Modifier.isPublic(m.getModifiers())) {
                final String key = prefix + "." + m.getName().substring(3).toLowerCase(Locale.ENGLISH);
                final Object field = params.remove(key);
                if (field != null) {
                    try {
                        final Object toSet = PropertyEditors.getValue(m.getParameterTypes()[0], field.toString());
                        m.invoke(persistenceAdapter, toSet);
                    } catch (final PropertyEditorException cantConvertException) {
                        throw new IllegalArgumentException("can't convert " + field + " for " + m.getName(), cantConvertException);
                    }
                }
            }
        }
        aClass = aClass.getSuperclass();
    }
    return persistenceAdapter;
}

Example 4 with PersistenceAdapter

use of org.apache.activemq.store.PersistenceAdapter 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 5 with PersistenceAdapter

use of org.apache.activemq.store.PersistenceAdapter 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);
    }
}