org.argeo.init/src/org/argeo/init/RuntimeManagerMain.java

   1 package org.argeo.init;
   2
   3 import java.lang.System.Logger;
   4 import java.lang.System.Logger.Level;
   5 import java.nio.file.Path;
   6 import java.nio.file.Paths;
   7 import java.util.HashMap;
   8 import java.util.Map;
   9
  10 import org.argeo.api.init.InitConstants;
  11 import org.argeo.api.init.RuntimeManager;
  12 import org.argeo.init.logging.ThinLoggerFinder;
  13 import org.argeo.init.osgi.OsgiRuntimeContext;
  14 import org.argeo.internal.init.InternalState;
  15
  16 /**
  17  * Dynamically configures and launches multiple runtimes, coordinated by a main
  18  * one.
  19  */
  20 public class RuntimeManagerMain {
  21         private final static Logger logger = System.getLogger(RuntimeManagerMain.class.getName());
  22
  23         private final static String ENV_STATE_DIRECTORY = "STATE_DIRECTORY";
  24         private final static String ENV_CONFIGURATION_DIRECTORY = "CONFIGURATION_DIRECTORY";
  25         private final static String ENV_CACHE_DIRECTORY = "CACHE_DIRECTORY";
  26
  27         private final static long RUNTIME_SHUTDOWN_TIMEOUT = 60 * 1000;
  28
  29         private Map<String, String> configuration = new HashMap<>();
  30
  31         RuntimeManagerMain(Path configArea, Path stateArea, Path cacheArea) {
  32                 RuntimeManager.loadDefaults(configuration);
  33
  34                 configuration.put(InitConstants.PROP_OSGI_USE_SYSTEM_PROPERTIES, "false");
  35
  36                 configuration.put(InitConstants.PROP_ARGEO_CONFIG_AREA, configArea.toString());
  37                 configuration.put(InitConstants.PROP_ARGEO_STATE_AREA, stateArea.toString());
  38                 configuration.put(InitConstants.PROP_ARGEO_CACHE_AREA, cacheArea.toString());
  39
  40                 configuration.put(InitConstants.PROP_OSGI_SHARED_CONFIGURATION_AREA, configArea.toUri().toString());
  41                 configuration.put(InitConstants.PROP_OSGI_SHARED_CONFIGURATION_AREA_RO, "true");
  42
  43                 configuration.put(InitConstants.PROP_OSGI_CONFIGURATION_AREA,
  44                                 cacheArea.resolve(RuntimeManager.OSGI_STORAGE_DIRNAME).toUri().toString());
  45                 configuration.put(InitConstants.PROP_OSGI_INSTANCE_AREA,
  46                                 stateArea.resolve(RuntimeManager.DATA).toUri().toString());
  47
  48                 // TODO find a cleaner way to configure Jackrabbit indexes
  49                 configuration.put("argeo.node.repo.indexesBase", cacheArea.resolve("indexes").toString());
  50
  51                 logger.log(Level.TRACE, () -> "Runtime manager configuration: " + configuration);
  52         }
  53
  54         public void run() {
  55                 OsgiRuntimeContext managerRuntimeContext = new OsgiRuntimeContext(OsgiRuntimeContext.loadFrameworkFactory(),
  56                                 configuration);
  57                 try {
  58                         managerRuntimeContext.run();
  59                         InternalState.setMainRuntimeContext(managerRuntimeContext);
  60
  61                         // shutdown on exit
  62                         Runtime.getRuntime().addShutdownHook(new Thread(() -> shutdown(), "Runtime shutdown"));
  63
  64 //                      BundleContext bc = managerRuntimeContext.getFramework().getBundleContext();
  65 //                      // uninstall init as a bundle since it will be available via OSGi system
  66 //                      OsgiBoot.uninstallBundles(bc, SYMBOLIC_NAME_INIT);
  67 //                      bc.registerService(RuntimeManager.class, this, new Hashtable<>(configuration));
  68                         logger.log(Level.DEBUG, "Registered runtime manager");
  69
  70                         managerRuntimeContext.waitForStop(0);
  71                 } catch (InterruptedException e) {
  72                         e.printStackTrace();
  73                         System.exit(1);
  74                 }
  75
  76         }
  77
  78         protected void shutdown() {
  79                 // shutdown manager runtime
  80                 try {
  81                         InternalState.getMainRuntimeContext().close();
  82                         InternalState.getMainRuntimeContext().waitForStop(RUNTIME_SHUTDOWN_TIMEOUT);
  83 //                      logger.log(Logger.Level.INFO, "Argeo Init stopped with PID " + ProcessHandle.current().pid());
  84                         System.out.flush();
  85                 } catch (Exception e) {
  86                         e.printStackTrace();
  87                         Runtime.getRuntime().halt(1);
  88                 }
  89         }
  90
  91         public static void main(String[] args) {
  92                 ThinLoggerFinder.reloadConfiguration();
  93                 logger.log(Logger.Level.DEBUG, () -> "Argeo Init starting with PID " + ProcessHandle.current().pid());
  94
  95                 Path writableArea = getLocalPath(InitConstants.PROP_ARGEO_STATE_AREA, ENV_STATE_DIRECTORY, null);
  96                 Path configArea = getLocalPath(InitConstants.PROP_ARGEO_CONFIG_AREA, ENV_CONFIGURATION_DIRECTORY, null);
  97                 Path cacheArea = getLocalPath(InitConstants.PROP_ARGEO_CACHE_AREA, ENV_CACHE_DIRECTORY, writableArea);
  98                 RuntimeManagerMain runtimeManager = new RuntimeManagerMain(configArea, writableArea, cacheArea);
  99                 runtimeManager.run();
 100         }
 101
 102         private static Path getLocalPath(String systemProperty, String environmentVariable, Path defaultPath) {
 103                 String prop = System.getProperty(systemProperty);
 104                 if (prop != null)
 105                         return Paths.get(prop);
 106                 String env = System.getenv().get(environmentVariable);
 107                 if (env != null)
 108                         return Paths.get(env);
 109                 if (defaultPath != null)
 110                         return defaultPath;
 111                 throw new IllegalStateException("No local path set with system property " + systemProperty
 112                                 + " or environment variable " + environmentVariable);
 113                 // TODO allocate a temporary directory? or defaults based on working directory ?
 114         }
 115 }