Java tutorial
/* * RHQ Management Platform * Copyright 2012, Red Hat Middleware LLC, and individual contributors * as indicated by the @author tags. See the copyright.txt file in the * distribution for a full listing of individual contributors. * * This program is free software; you can redistribute it and/or modify * it under the terms of the GNU General Public License as published by * the Free Software Foundation version 2 of the License. * * This program is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU General Public License for more details. * * You should have received a copy of the GNU General Public License * along with this program; if not, write to the Free Software * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA. */ package org.rhq.enterprise.server.test; import java.io.IOException; import java.io.InputStream; import java.util.Properties; import javax.annotation.PostConstruct; import javax.annotation.Resource; import javax.ejb.EJB; import javax.ejb.Singleton; import javax.ejb.Startup; import javax.ejb.Timeout; import javax.ejb.TimerConfig; import javax.ejb.TimerService; import javax.persistence.EntityManager; import javax.persistence.PersistenceContext; import org.apache.commons.logging.Log; import org.apache.commons.logging.LogFactory; import org.rhq.core.domain.cloud.Server; import org.rhq.core.domain.cloud.StorageNode; import org.rhq.enterprise.server.RHQConstants; import org.rhq.enterprise.server.cloud.instance.ServerManagerLocal; /** * This startup singleton EJB is here to work around bug AS7-5530 and to * schedule the real StartupBean's work in a delayed fashion (this is to allow * AS7 to complete its deployment work before we do our work). * * See https://issues.jboss.org/browse/AS7-5530 * * This version is a replacement for the original code (identical) code that uses the StrippedDownStartupBean instead * of the fullblown original. */ @Singleton @Startup public class StrippedDownStartupBeanPreparation { private Log log = LogFactory.getLog(this.getClass()); @EJB private StrippedDownStartupBean startupBean; @EJB private ServerManagerLocal serverManager; @PersistenceContext(unitName = RHQConstants.PERSISTENCE_UNIT_NAME) private EntityManager entityManager; @Resource private TimerService timerService; // needed to schedule our startup bean init call @PostConstruct public void initWithTransactionBecauseAS75530() throws RuntimeException { log.info("Scheduling the initialization of the testing RHQ deployment"); timerService.createSingleActionTimer(1, new TimerConfig(null, false)); // call StartupBean in 1ms startupBean.purgeTestServerAndStorageNodes(); createTestServer(); loadCassandraConnectionProps(); createStorageNodes(); } /** * The storage client initialization performed by {@link org.rhq.enterprise.server.storage.StorageClientManager#init()} * requires having a server entity in the database which will be the case with a regular deployment. This method * persists a server before any tests execute. If the server entity does not exist or cannot be loaded, then the * storage client will not initialize properly and any tests that depend on the storage client will fail. */ private void createTestServer() { Server server = new Server(); server.setName(TestConstants.RHQ_TEST_SERVER_NAME); server.setAddress("127.0.0.1"); server.setOperationMode(Server.OperationMode.INSTALLED); server.setPort(7080); server.setSecurePort(7443); serverManager.create(server); System.setProperty(TestConstants.RHQ_SERVER_NAME_PROPERTY, TestConstants.RHQ_TEST_SERVER_NAME); } private void createStorageNodes() { String[] nodes = System.getProperty("rhq.storage.nodes").split(","); String cqlPort = System.getProperty("rhq.storage.cql-port"); for (String node : nodes) { StorageNode storageNode = new StorageNode(); storageNode.setAddress(node); storageNode.setCqlPort(Integer.parseInt(cqlPort)); storageNode.setOperationMode(StorageNode.OperationMode.NORMAL); entityManager.persist(storageNode); } } public void loadCassandraConnectionProps() { InputStream stream = null; try { stream = getClass().getResourceAsStream("/cassandra-test.properties"); Properties props = new Properties(); props.load(stream); // DO NOT use System.setProperties(Properties). I previously tried that and it // caused some arquillian deployment exception. // // jsanda System.setProperty("rhq.storage.username", props.getProperty("rhq.storage.username")); System.setProperty("rhq.storage.password", props.getProperty("rhq.storage.password")); System.setProperty("rhq.storage.nodes", props.getProperty("rhq.storage.nodes")); System.setProperty("rhq.storage.cql-port", props.getProperty("rhq.storage.cql-port")); System.setProperty("rhq.storage.gossip-port", props.getProperty("rhq.storage.gossip-port")); } catch (IOException e) { throw new RuntimeException(("Failed to load cassandra-test.properties")); } String cqlPort = System.getProperty("rhq.storage.cql-port"); entityManager .createNativeQuery("update rhq_system_config set property_value = '" + cqlPort + "', default_property_value = '" + cqlPort + "' where property_key = 'STORAGE_CQL_PORT'") .executeUpdate(); String gossipPort = System.getProperty("rhq.storage.gossip-port"); entityManager.createNativeQuery("update rhq_system_config set property_value = '" + gossipPort + "', default_property_value = '" + gossipPort + "' where property_key = 'STORAGE_GOSSIP_PORT'") .executeUpdate(); String storageUserName = System.getProperty("rhq.storage.username"); entityManager.createNativeQuery("update rhq_system_config set property_value = '" + storageUserName + "', default_property_value = '" + storageUserName + "' where property_key = 'STORAGE_USERNAME'") .executeUpdate(); String storagePassword = System.getProperty("rhq.storage.password"); entityManager.createNativeQuery("update rhq_system_config set property_value = '" + storagePassword + "', default_property_value = '" + storagePassword + "' where property_key = 'STORAGE_PASSWORD'") .executeUpdate(); } @Timeout public void initializeServer() throws RuntimeException { try { log.info("Initializing the testing RHQ deployment"); this.startupBean.init(); log.info("Initialization complete"); } catch (Throwable t) { // do NOT allow exceptions to bubble out of our method because then // the EJB container would simply re-trigger the timer and call us again // and we don't want to keep failing over and over filling the logs // in an infinite loop. log.fatal("The server failed to start up properly", t); } } }