package org.argeo.cms.internal.kernel; import java.lang.management.ManagementFactory; import java.security.PrivilegedAction; import java.util.HashMap; import java.util.Map; import javax.jcr.Repository; import javax.jcr.RepositoryFactory; import javax.security.auth.Subject; import javax.transaction.TransactionManager; import javax.transaction.TransactionSynchronizationRegistry; import javax.transaction.UserTransaction; import org.apache.commons.logging.Log; import org.apache.commons.logging.LogFactory; import org.apache.jackrabbit.util.TransientFileFactory; import org.argeo.ArgeoException; import org.argeo.cms.CmsException; import org.argeo.cms.internal.transaction.SimpleTransactionManager; import org.argeo.jackrabbit.OsgiJackrabbitRepositoryFactory; import org.argeo.jcr.ArgeoJcrConstants; import org.eclipse.equinox.http.servlet.ExtendedHttpService; import org.osgi.framework.BundleContext; import org.osgi.framework.ServiceEvent; import org.osgi.framework.ServiceListener; import org.osgi.framework.ServiceReference; import org.osgi.util.tracker.ServiceTracker; /** * Argeo CMS Kernel. Responsible for : * */ final class Kernel implements ServiceListener { private final static Log log = LogFactory.getLog(Kernel.class); private final BundleContext bundleContext = Activator.getBundleContext(); private final NodeSecurity nodeSecurity; ThreadGroup threadGroup = new ThreadGroup(Kernel.class.getSimpleName()); JackrabbitNode node; private SimpleTransactionManager transactionManager; private OsgiJackrabbitRepositoryFactory repositoryFactory; private NodeHttp nodeHttp; private KernelThread kernelThread; public Kernel() { nodeSecurity = new NodeSecurity(bundleContext); } final void init() { Subject.doAs(nodeSecurity.getKernelSubject(), new PrivilegedAction() { @Override public Void run() { doInit(); return null; } }); } private void doInit() { ClassLoader currentContextCl = Thread.currentThread() .getContextClassLoader(); Thread.currentThread().setContextClassLoader( Kernel.class.getClassLoader()); long begin = System.currentTimeMillis(); try { // Transaction transactionManager = new SimpleTransactionManager(); // Jackrabbit node node = new JackrabbitNode(bundleContext); // JCR repository factory repositoryFactory = new OsgiJackrabbitRepositoryFactory(); // Authentication nodeSecurity.getUserAdmin().setTransactionManager( transactionManager); // Equinox dependency ExtendedHttpService httpService = waitForHttpService(); nodeHttp = new NodeHttp(httpService, node); // Kernel thread kernelThread = new KernelThread(this); kernelThread.setContextClassLoader(Kernel.class.getClassLoader()); kernelThread.start(); // Publish services to OSGi bundleContext.registerService(TransactionManager.class, transactionManager, null); bundleContext.registerService(UserTransaction.class, transactionManager, null); bundleContext.registerService( TransactionSynchronizationRegistry.class, transactionManager.getTransactionSynchronizationRegistry(), null); nodeSecurity.publish(); node.publish(repositoryFactory); bundleContext.registerService(RepositoryFactory.class, repositoryFactory, null); bundleContext.addServiceListener(Kernel.this); } catch (Exception e) { log.error("Cannot initialize Argeo CMS", e); throw new ArgeoException("Cannot initialize", e); } finally { Thread.currentThread().setContextClassLoader(currentContextCl); } long jvmUptime = ManagementFactory.getRuntimeMXBean().getUptime(); log.info("## ARGEO CMS UP in " + (jvmUptime / 1000) + "." + (jvmUptime % 1000) + "s ##"); long initDuration = System.currentTimeMillis() - begin; if (log.isTraceEnabled()) log.trace("Kernel initialization took " + initDuration + "ms"); directorsCut(initDuration); } void destroy() { long begin = System.currentTimeMillis(); kernelThread.destroyAndJoin(); if (nodeHttp != null) nodeHttp.destroy(); // if (nodeSecurity != null) // nodeSecurity.destroy(); if (node != null) node.destroy(); bundleContext.removeServiceListener(this); // Clean hanging threads from Jackrabbit TransientFileFactory.shutdown(); // Clean hanging Gogo shell thread new GogoShellKiller().start(); nodeSecurity.destroy(); long duration = System.currentTimeMillis() - begin; log.info("## ARGEO CMS DOWN in " + (duration / 1000) + "." + (duration % 1000) + "s ##"); } @Override public void serviceChanged(ServiceEvent event) { ServiceReference sr = event.getServiceReference(); Object jcrRepoAlias = sr .getProperty(ArgeoJcrConstants.JCR_REPOSITORY_ALIAS); if (jcrRepoAlias != null) {// JCR repository String alias = jcrRepoAlias.toString(); Repository repository = (Repository) bundleContext.getService(sr); Map props = new HashMap(); for (String key : sr.getPropertyKeys()) props.put(key, sr.getProperty(key)); if (ServiceEvent.REGISTERED == event.getType()) { try { repositoryFactory.register(repository, props); nodeHttp.registerRepositoryServlets(alias, repository); } catch (Exception e) { throw new CmsException("Could not publish JCR repository " + alias, e); } } else if (ServiceEvent.UNREGISTERING == event.getType()) { repositoryFactory.unregister(repository, props); nodeHttp.unregisterRepositoryServlets(alias); } } } private ExtendedHttpService waitForHttpService() { final ServiceTracker st = new ServiceTracker( bundleContext, ExtendedHttpService.class, null); st.open(); ExtendedHttpService httpService; try { httpService = st.waitForService(1000); } catch (InterruptedException e) { httpService = null; } if (httpService == null) throw new CmsException("Could not find " + ExtendedHttpService.class + " service."); return httpService; } final private static void directorsCut(long initDuration) { // final long ms = 128l + (long) (Math.random() * 128d); long ms = initDuration / 100; log.info("Spend " + ms + "ms" + " reflecting on the progress brought to mankind" + " by Free Software..."); long beginNano = System.nanoTime(); try { Thread.sleep(ms, 0); } catch (InterruptedException e) { // silent } long durationNano = System.nanoTime() - beginNano; final double M = 1000d * 1000d; double sleepAccuracy = ((double) durationNano) / (ms * M); if (log.isDebugEnabled()) log.debug("Sleep accuracy: " + String.format("%.2f", 100 - (sleepAccuracy * 100 - 100)) + " %"); } /** Workaround for blocking Gogo shell by system shutdown. */ private class GogoShellKiller extends Thread { public GogoShellKiller() { super("Gogo shell killer"); setDaemon(true); } @Override public void run() { ThreadGroup rootTg = getRootThreadGroup(null); Thread gogoShellThread = findGogoShellThread(rootTg); if (gogoShellThread == null) return; while (getNonDaemonCount(rootTg) > 2) { try { Thread.sleep(100); } catch (InterruptedException e) { // silent } } gogoShellThread = findGogoShellThread(rootTg); if (gogoShellThread == null) return; System.exit(0); } } private static ThreadGroup getRootThreadGroup(ThreadGroup tg) { if (tg == null) tg = Thread.currentThread().getThreadGroup(); if (tg.getParent() == null) return tg; else return getRootThreadGroup(tg.getParent()); } private static int getNonDaemonCount(ThreadGroup rootThreadGroup) { Thread[] threads = new Thread[rootThreadGroup.activeCount()]; rootThreadGroup.enumerate(threads); int nonDameonCount = 0; for (Thread t : threads) if (!t.isDaemon()) nonDameonCount++; return nonDameonCount; } private static Thread findGogoShellThread(ThreadGroup rootThreadGroup) { Thread[] threads = new Thread[rootThreadGroup.activeCount()]; rootThreadGroup.enumerate(threads, true); for (Thread thread : threads) { if (thread.getName().equals("Gogo shell")) return thread; } return null; } }