1 package org
.argeo
.cms
.internal
.kernel
;
3 import static bitronix
.tm
.TransactionManagerServices
.getTransactionManager
;
4 import static bitronix
.tm
.TransactionManagerServices
.getTransactionSynchronizationRegistry
;
5 import static java
.util
.Locale
.ENGLISH
;
8 import java
.net
.InetAddress
;
9 import java
.net
.UnknownHostException
;
10 import java
.util
.ArrayList
;
11 import java
.util
.List
;
12 import java
.util
.Locale
;
13 import java
.util
.UUID
;
15 import javax
.jcr
.RepositoryFactory
;
16 import javax
.transaction
.TransactionManager
;
17 import javax
.transaction
.TransactionSynchronizationRegistry
;
18 import javax
.transaction
.UserTransaction
;
20 import org
.apache
.commons
.logging
.Log
;
21 import org
.apache
.commons
.logging
.LogFactory
;
22 import org
.argeo
.cms
.i18n
.LocaleUtils
;
23 import org
.argeo
.node
.NodeConstants
;
24 import org
.argeo
.node
.NodeState
;
25 import org
.argeo
.util
.LangUtils
;
26 import org
.osgi
.framework
.Bundle
;
27 import org
.osgi
.framework
.BundleContext
;
28 import org
.osgi
.framework
.Constants
;
29 import org
.osgi
.framework
.FrameworkUtil
;
30 import org
.osgi
.framework
.ServiceReference
;
31 import org
.osgi
.service
.cm
.ManagedServiceFactory
;
33 import bitronix
.tm
.BitronixTransactionManager
;
34 import bitronix
.tm
.BitronixTransactionSynchronizationRegistry
;
35 import bitronix
.tm
.TransactionManagerServices
;
37 public class CmsState
implements NodeState
{
38 private final Log log
= LogFactory
.getLog(CmsState
.class);
39 private final BundleContext bc
= FrameworkUtil
.getBundle(CmsState
.class).getBundleContext();
42 private Long availableSince
;
45 private Locale defaultLocale
;
46 private List
<Locale
> locales
= null;
48 private ThreadGroup threadGroup
= new ThreadGroup("CMS");
49 private KernelThread kernelThread
;
50 private List
<Runnable
> shutdownHooks
= new ArrayList
<>();
52 private final String stateUuid
;
53 private final boolean cleanState
;
54 private String hostname
;
56 public CmsState(String stateUuid
) {
57 this.stateUuid
= stateUuid
;
58 String frameworkUuid
= KernelUtils
.getFrameworkProp(Constants
.FRAMEWORK_UUID
);
59 this.cleanState
= stateUuid
.equals(frameworkUuid
);
61 this.hostname
= InetAddress
.getLocalHost().getHostName();
62 } catch (UnknownHostException e
) {
63 log
.error("Cannot set hostname", e
);
66 availableSince
= System
.currentTimeMillis();
67 if (log
.isDebugEnabled())
68 log
.debug("## CMS STARTED " + this.stateUuid
+ (cleanState ?
" (clean state) " : " "));
74 kernelThread
= new KernelThread(threadGroup
, "Kernel Thread");
75 kernelThread
.setContextClassLoader(getClass().getClassLoader());
79 private void initI18n() {
80 Object defaultLocaleValue
= KernelUtils
.getFrameworkProp(NodeConstants
.I18N_DEFAULT_LOCALE
);
81 defaultLocale
= defaultLocaleValue
!= null ?
new Locale(defaultLocaleValue
.toString())
82 : new Locale(ENGLISH
.getLanguage());
83 locales
= LocaleUtils
.asLocaleList(KernelUtils
.getFrameworkProp(NodeConstants
.I18N_LOCALES
));
86 private void initServices() {
88 initTransactionManager();
91 RepositoryServiceFactory repositoryServiceFactory
= new RepositoryServiceFactory();
92 shutdownHooks
.add(() -> repositoryServiceFactory
.shutdown());
93 bc
.registerService(ManagedServiceFactory
.class, repositoryServiceFactory
,
94 LangUtils
.dico(Constants
.SERVICE_PID
, NodeConstants
.NODE_REPOS_FACTORY_PID
));
96 NodeRepositoryFactory repositoryFactory
= new NodeRepositoryFactory();
97 bc
.registerService(RepositoryFactory
.class, repositoryFactory
, null);
100 NodeUserAdmin userAdmin
= new NodeUserAdmin(NodeConstants
.ROLES_BASEDN
);
101 shutdownHooks
.add(() -> userAdmin
.destroy());
102 bc
.registerService(ManagedServiceFactory
.class, userAdmin
,
103 LangUtils
.dico(Constants
.SERVICE_PID
, NodeConstants
.NODE_USER_ADMIN_PID
));
106 private void initTransactionManager() {
107 // TODO manage it in a managed service, as startup could be long
108 ServiceReference
<TransactionManager
> existingTm
= bc
.getServiceReference(TransactionManager
.class);
109 if (existingTm
!= null) {
110 if (log
.isDebugEnabled())
111 log
.debug("Using provided transaction manager " + existingTm
);
113 bitronix
.tm
.Configuration tmConf
= TransactionManagerServices
.getConfiguration();
114 tmConf
.setServerId(UUID
.randomUUID().toString());
116 Bundle bitronixBundle
= FrameworkUtil
.getBundle(bitronix
.tm
.Configuration
.class);
117 File tmBaseDir
= bitronixBundle
.getDataFile(KernelConstants
.DIR_TRANSACTIONS
);
118 File tmDir1
= new File(tmBaseDir
, "btm1");
120 tmConf
.setLogPart1Filename(new File(tmDir1
, tmDir1
.getName() + ".tlog").getAbsolutePath());
121 File tmDir2
= new File(tmBaseDir
, "btm2");
123 tmConf
.setLogPart2Filename(new File(tmDir2
, tmDir2
.getName() + ".tlog").getAbsolutePath());
125 BitronixTransactionManager transactionManager
= getTransactionManager();
126 shutdownHooks
.add(() -> transactionManager
.shutdown());
127 BitronixTransactionSynchronizationRegistry transactionSynchronizationRegistry
= getTransactionSynchronizationRegistry();
129 bc
.registerService(TransactionManager
.class, transactionManager
, null);
130 bc
.registerService(UserTransaction
.class, transactionManager
, null);
131 bc
.registerService(TransactionSynchronizationRegistry
.class, transactionSynchronizationRegistry
, null);
132 if (log
.isDebugEnabled())
133 log
.debug("Initialised default Bitronix transaction manager");
137 applyShutdownHooks();
139 if (kernelThread
!= null)
140 kernelThread
.destroyAndJoin();
142 if (log
.isDebugEnabled())
143 log
.debug("## CMS STOPPED");
146 /** Apply shutdown hoos in reverse order. */
147 private void applyShutdownHooks() {
148 for (int i
= shutdownHooks
.size() - 1; i
>= 0; i
--) {
150 shutdownHooks
.get(i
).run();
151 } catch (Exception e
) {
152 log
.error("Could not run shutdown hook #" + i
);
155 // Clean hanging Gogo shell thread
156 new GogoShellKiller().start();
160 public boolean isClean() {
165 public Long
getAvailableSince() {
166 return availableSince
;
172 public Locale
getDefaultLocale() {
173 return defaultLocale
;
176 public List
<Locale
> getLocales() {
180 public String
getHostname() {
184 /** Workaround for blocking Gogo shell by system shutdown. */
185 private class GogoShellKiller
extends Thread
{
187 public GogoShellKiller() {
188 super("Gogo Shell Killer");
194 ThreadGroup rootTg
= getRootThreadGroup(null);
195 Thread gogoShellThread
= findGogoShellThread(rootTg
);
196 if (gogoShellThread
== null)
198 while (getNonDaemonCount(rootTg
) > 2) {
201 } catch (InterruptedException e
) {
205 gogoShellThread
= findGogoShellThread(rootTg
);
206 if (gogoShellThread
== null)
212 private static ThreadGroup
getRootThreadGroup(ThreadGroup tg
) {
214 tg
= Thread
.currentThread().getThreadGroup();
215 if (tg
.getParent() == null)
218 return getRootThreadGroup(tg
.getParent());
221 private static int getNonDaemonCount(ThreadGroup rootThreadGroup
) {
222 Thread
[] threads
= new Thread
[rootThreadGroup
.activeCount()];
223 rootThreadGroup
.enumerate(threads
);
224 int nonDameonCount
= 0;
225 for (Thread t
: threads
)
226 if (t
!= null && !t
.isDaemon())
228 return nonDameonCount
;
231 private static Thread
findGogoShellThread(ThreadGroup rootThreadGroup
) {
232 Thread
[] threads
= new Thread
[rootThreadGroup
.activeCount()];
233 rootThreadGroup
.enumerate(threads
, true);
234 for (Thread thread
: threads
) {
235 if (thread
.getName().equals("Gogo shell"))