X-Git-Url: https://git.argeo.org/?a=blobdiff_plain;f=org.argeo.cms%2Fsrc%2Forg%2Fargeo%2Fcms%2Finternal%2Fkernel%2FKernel.java;h=08dad56b803c28bedf147e9e055ffa214e247e21;hb=a4a78ecbc3cfd119477264534c7d0cab541ae6ad;hp=4b893f13d0b6e195be13c8354b8ab2dc756d8cb0;hpb=559786a622e24c7d213960a7873e105db82a03ab;p=lgpl%2Fargeo-commons.git

diff --git a/org.argeo.cms/src/org/argeo/cms/internal/kernel/Kernel.java b/org.argeo.cms/src/org/argeo/cms/internal/kernel/Kernel.java
index 4b893f13d..08dad56b8 100644
--- a/org.argeo.cms/src/org/argeo/cms/internal/kernel/Kernel.java
+++ b/org.argeo.cms/src/org/argeo/cms/internal/kernel/Kernel.java
@@ -1,19 +1,32 @@
 package org.argeo.cms.internal.kernel;
 
-import java.util.Hashtable;
+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.argeo.security.core.InternalAuthentication;
-import org.eclipse.rap.rwt.application.ApplicationConfiguration;
-import org.eclipse.rap.rwt.osgi.ApplicationLauncher;
-import org.eclipse.rap.ui.internal.servlet.WorkbenchApplicationConfiguration;
+import org.eclipse.equinox.http.servlet.ExtendedHttpService;
 import org.osgi.framework.BundleContext;
-import org.osgi.framework.ServiceRegistration;
+import org.osgi.framework.ServiceEvent;
+import org.osgi.framework.ServiceListener;
+import org.osgi.framework.ServiceReference;
+import org.osgi.util.tracker.ServiceTracker;
 import org.springframework.security.core.context.SecurityContextHolder;
 
 /**
@@ -27,87 +40,174 @@ import org.springframework.security.core.context.SecurityContextHolder;
  * <li>OS access</li>
  * </ul>
  */
-@SuppressWarnings("restriction")
-final class Kernel {
+final class Kernel implements ServiceListener {
 	private final static Log log = LogFactory.getLog(Kernel.class);
-	private static final String PROP_WORKBENCH_AUTOSTART = "org.eclipse.rap.workbenchAutostart";
 
-	private final BundleContext bundleContext;
+	private final BundleContext bundleContext = Activator.getBundleContext();
+	private final NodeSecurity nodeSecurity;
 
-	private JackrabbitNode node;
-	private RepositoryFactory repositoryFactory;
-	private 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);
+	}
 
-	private ServiceRegistration<ApplicationConfiguration> workbenchReg;
+	final void init() {
+		Subject.doAs(nodeSecurity.getKernelSubject(),
+				new PrivilegedAction<Void>() {
 
-	Kernel(BundleContext bundleContext) {
-		this.bundleContext = bundleContext;
+					@Override
+					public Void run() {
+						doInit();
+						return null;
+					}
+
+				});
 	}
 
-	void init() {
+	private void doInit() {
+		ClassLoader currentContextCl = Thread.currentThread()
+				.getContextClassLoader();
+		Thread.currentThread().setContextClassLoader(
+				Kernel.class.getClassLoader());
 		long begin = System.currentTimeMillis();
 		InternalAuthentication initAuth = new InternalAuthentication(
 				KernelConstants.DEFAULT_SECURITY_KEY);
 		SecurityContextHolder.getContext().setAuthentication(initAuth);
 
 		try {
+			// Transaction
+			transactionManager = new SimpleTransactionManager();
+
+			// Jackrabbit node
 			node = new JackrabbitNode(bundleContext);
+
+			// JCR repository factory
 			repositoryFactory = new OsgiJackrabbitRepositoryFactory();
-			nodeSecurity = new NodeSecurity(bundleContext, node);
-			nodeHttp = new NodeHttp(bundleContext, node, nodeSecurity);
+
+			// 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();
+			node.publish(repositoryFactory);
 			bundleContext.registerService(RepositoryFactory.class,
 					repositoryFactory, null);
-			nodeHttp.publish();
 
-			if ("false".equals(bundleContext
-					.getProperty(PROP_WORKBENCH_AUTOSTART))) {
-				WorkbenchApplicationConfiguration wac = new WorkbenchApplicationConfiguration();
-				registerWorkbench(wac);
-			}
+			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 duration = System.currentTimeMillis() - begin;
-		log.info("## ARGEO CMS UP in " + (duration / 1000) + "."
-				+ (duration % 1000) + "s ##");
-		directorsCut();
+		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();
 
-		// OSGi
-		workbenchReg.unregister();
+		kernelThread.destroyAndJoin();
 
-		nodeHttp = null;
-		nodeSecurity.destroy();
-		node.destroy();
+		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 ##");
 	}
 
-	private void registerWorkbench(final WorkbenchApplicationConfiguration wac) {
-		new Thread("Worbench Launcher") {
-			public void run() {
-				Hashtable<String, String> props = new Hashtable<String, String>();
-				props.put(ApplicationLauncher.PROPERTY_CONTEXT_NAME, "ui");
-				workbenchReg = bundleContext.registerService(
-						ApplicationConfiguration.class, wac, props);
+	@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<String, Object> props = new HashMap<String, Object>();
+			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);
 			}
-		}.start();
+		}
+
 	}
 
-	private void directorsCut() {
-		final long ms = 128l + (long) (Math.random() * 128d);
+	private ExtendedHttpService waitForHttpService() {
+		final ServiceTracker<ExtendedHttpService, ExtendedHttpService> st = new ServiceTracker<ExtendedHttpService, ExtendedHttpService>(
+				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...");
@@ -122,7 +222,65 @@ final class Kernel {
 		double sleepAccuracy = ((double) durationNano) / (ms * M);
 		if (log.isDebugEnabled())
 			log.debug("Sleep accuracy: "
-					+ String.format("%.2f", sleepAccuracy * 100) + " %");
+					+ 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;
 	}
 
-}
+}
\ No newline at end of file