1 | |
|
2 | |
|
3 | |
|
4 | |
|
5 | |
|
6 | |
|
7 | |
package org.mule.config.spring; |
8 | |
|
9 | |
import org.mule.api.MuleContext; |
10 | |
import org.mule.api.MuleRuntimeException; |
11 | |
import org.mule.api.lifecycle.InitialisationException; |
12 | |
import org.mule.api.registry.RegistrationException; |
13 | |
import org.mule.config.i18n.MessageFactory; |
14 | |
import org.mule.lifecycle.RegistryLifecycleManager; |
15 | |
import org.mule.registry.AbstractRegistry; |
16 | |
import org.mule.util.StringUtils; |
17 | |
|
18 | |
import java.util.Collection; |
19 | |
import java.util.Collections; |
20 | |
import java.util.Map; |
21 | |
import java.util.concurrent.atomic.AtomicBoolean; |
22 | |
|
23 | |
import org.springframework.beans.FatalBeanException; |
24 | |
import org.springframework.beans.factory.NoSuchBeanDefinitionException; |
25 | |
import org.springframework.context.ApplicationContext; |
26 | |
import org.springframework.context.ConfigurableApplicationContext; |
27 | |
|
28 | |
public class SpringRegistry extends AbstractRegistry |
29 | |
{ |
30 | |
public static final String REGISTRY_ID = "org.mule.Registry.Spring"; |
31 | |
|
32 | |
|
33 | |
|
34 | |
|
35 | |
|
36 | |
public static final String SPRING_APPLICATION_CONTEXT = "springApplicationContext"; |
37 | |
|
38 | |
protected ApplicationContext applicationContext; |
39 | |
|
40 | |
|
41 | |
|
42 | 0 | protected AtomicBoolean springContextInitialised = new AtomicBoolean(false); |
43 | |
|
44 | |
public SpringRegistry(MuleContext muleContext) |
45 | |
{ |
46 | 0 | super(REGISTRY_ID, muleContext); |
47 | 0 | } |
48 | |
|
49 | |
public SpringRegistry(String id, MuleContext muleContext) |
50 | |
{ |
51 | 0 | super(id, muleContext); |
52 | 0 | } |
53 | |
|
54 | |
public SpringRegistry(ApplicationContext applicationContext, MuleContext muleContext) |
55 | |
{ |
56 | 0 | super(REGISTRY_ID, muleContext); |
57 | 0 | this.applicationContext = applicationContext; |
58 | 0 | } |
59 | |
|
60 | |
public SpringRegistry(String id, ApplicationContext applicationContext, MuleContext muleContext) |
61 | |
{ |
62 | 0 | super(id, muleContext); |
63 | 0 | this.applicationContext = applicationContext; |
64 | 0 | } |
65 | |
|
66 | |
public SpringRegistry(ConfigurableApplicationContext applicationContext, ApplicationContext parentContext, MuleContext muleContext) |
67 | |
{ |
68 | 0 | super(REGISTRY_ID, muleContext); |
69 | 0 | applicationContext.setParent(parentContext); |
70 | 0 | this.applicationContext = applicationContext; |
71 | 0 | } |
72 | |
|
73 | |
public SpringRegistry(String id, ConfigurableApplicationContext applicationContext, ApplicationContext parentContext, MuleContext muleContext) |
74 | |
{ |
75 | 0 | super(id, muleContext); |
76 | 0 | applicationContext.setParent(parentContext); |
77 | 0 | this.applicationContext = applicationContext; |
78 | 0 | } |
79 | |
|
80 | |
@Override |
81 | |
protected void doInitialise() throws InitialisationException |
82 | |
{ |
83 | 0 | if (applicationContext instanceof ConfigurableApplicationContext) |
84 | |
{ |
85 | 0 | ((ConfigurableApplicationContext) applicationContext).refresh(); |
86 | |
} |
87 | |
|
88 | 0 | springContextInitialised.set(true); |
89 | 0 | } |
90 | |
|
91 | |
@Override |
92 | |
public void doDispose() |
93 | |
{ |
94 | |
|
95 | |
|
96 | 0 | if (!this.springContextInitialised.get()) |
97 | |
{ |
98 | 0 | return; |
99 | |
} |
100 | |
|
101 | 0 | if (applicationContext instanceof ConfigurableApplicationContext |
102 | |
&& ((ConfigurableApplicationContext) applicationContext).isActive()) |
103 | |
{ |
104 | 0 | ((ConfigurableApplicationContext) applicationContext).close(); |
105 | |
} |
106 | |
|
107 | |
|
108 | 0 | applicationContext = null; |
109 | |
|
110 | 0 | this.springContextInitialised.set(false); |
111 | 0 | } |
112 | |
|
113 | |
@Override |
114 | |
protected RegistryLifecycleManager createLifecycleManager() |
115 | |
{ |
116 | 0 | return new SpringRegistryLifecycleManager(getRegistryId(), this, muleContext); |
117 | |
} |
118 | |
|
119 | |
public Object lookupObject(String key) |
120 | |
{ |
121 | 0 | if (StringUtils.isBlank(key)) |
122 | |
{ |
123 | 0 | logger.warn( |
124 | |
MessageFactory.createStaticMessage("Detected a lookup attempt with an empty or null key"), |
125 | |
new Throwable().fillInStackTrace()); |
126 | 0 | return null; |
127 | |
} |
128 | |
|
129 | 0 | if (key.equals(SPRING_APPLICATION_CONTEXT) && applicationContext != null) |
130 | |
{ |
131 | 0 | return applicationContext; |
132 | |
} |
133 | |
else |
134 | |
{ |
135 | |
try |
136 | |
{ |
137 | 0 | return applicationContext.getBean(key); |
138 | |
} |
139 | 0 | catch (NoSuchBeanDefinitionException e) |
140 | |
{ |
141 | 0 | logger.debug(e); |
142 | 0 | return null; |
143 | |
} |
144 | |
} |
145 | |
} |
146 | |
|
147 | |
public <T> Collection<T> lookupObjects(Class<T> type) |
148 | |
{ |
149 | 0 | return lookupByType(type).values(); |
150 | |
} |
151 | |
|
152 | |
|
153 | |
|
154 | |
|
155 | |
@Override |
156 | |
public <T> Collection<T> lookupObjectsForLifecycle(Class<T> type) |
157 | |
{ |
158 | 0 | return internalLookupByType(type, false, false).values(); |
159 | |
} |
160 | |
|
161 | |
@SuppressWarnings("unchecked") |
162 | |
public <T> Map<String, T> lookupByType(Class<T> type) |
163 | |
{ |
164 | 0 | return internalLookupByType(type, true, true); |
165 | |
} |
166 | |
|
167 | |
@SuppressWarnings("unchecked") |
168 | |
protected <T> Map<String, T> internalLookupByType(Class<T> type, boolean nonSingletons, boolean eagerInit) |
169 | |
{ |
170 | |
try |
171 | |
{ |
172 | 0 | return applicationContext.getBeansOfType(type, nonSingletons, eagerInit); |
173 | |
} |
174 | 0 | catch (FatalBeanException fbex) |
175 | |
{ |
176 | |
|
177 | 0 | String message = String.format("Failed to lookup beans of type %s from the Spring registry", type); |
178 | 0 | throw new MuleRuntimeException(MessageFactory.createStaticMessage(message), fbex); |
179 | |
} |
180 | 0 | catch (Exception e) |
181 | |
{ |
182 | 0 | logger.debug(e); |
183 | 0 | return Collections.emptyMap(); |
184 | |
} |
185 | |
} |
186 | |
|
187 | |
|
188 | |
|
189 | |
|
190 | |
|
191 | |
|
192 | |
public void registerObject(String key, Object value) throws RegistrationException |
193 | |
{ |
194 | 0 | throw new UnsupportedOperationException("Registry is read-only so objects cannot be registered or unregistered."); |
195 | |
} |
196 | |
|
197 | |
public void registerObject(String key, Object value, Object metadata) throws RegistrationException |
198 | |
{ |
199 | 0 | throw new UnsupportedOperationException("Registry is read-only so objects cannot be registered or unregistered."); |
200 | |
} |
201 | |
|
202 | |
public void registerObjects(Map objects) throws RegistrationException |
203 | |
{ |
204 | 0 | throw new UnsupportedOperationException("Registry is read-only so objects cannot be registered or unregistered."); |
205 | |
} |
206 | |
|
207 | |
public void unregisterObject(String key) |
208 | |
{ |
209 | 0 | throw new UnsupportedOperationException("Registry is read-only so objects cannot be registered or unregistered."); |
210 | |
} |
211 | |
|
212 | |
public void unregisterObject(String key, Object metadata) throws RegistrationException |
213 | |
{ |
214 | 0 | throw new UnsupportedOperationException("Registry is read-only so objects cannot be registered or unregistered."); |
215 | |
} |
216 | |
|
217 | |
|
218 | |
|
219 | |
|
220 | |
|
221 | |
public boolean isReadOnly() |
222 | |
{ |
223 | 0 | return true; |
224 | |
} |
225 | |
|
226 | |
public boolean isRemote() |
227 | |
{ |
228 | 0 | return false; |
229 | |
} |
230 | |
|
231 | |
} |