1 | |
|
2 | |
|
3 | |
|
4 | |
|
5 | |
|
6 | |
|
7 | |
|
8 | |
|
9 | |
|
10 | |
|
11 | |
package org.mule.impl; |
12 | |
|
13 | |
import org.mule.config.MuleProperties; |
14 | |
import org.mule.umo.UMOEvent; |
15 | |
import org.mule.umo.UMOEventContext; |
16 | |
import org.mule.umo.UMOExceptionPayload; |
17 | |
import org.mule.umo.UMOMessage; |
18 | |
|
19 | |
import java.util.Iterator; |
20 | |
|
21 | |
import org.apache.commons.logging.Log; |
22 | |
import org.apache.commons.logging.LogFactory; |
23 | |
|
24 | |
|
25 | |
|
26 | |
|
27 | |
|
28 | |
|
29 | |
|
30 | |
|
31 | |
|
32 | |
|
33 | |
|
34 | |
public final class RequestContext |
35 | |
{ |
36 | |
|
37 | 0 | private static boolean SAFE = true; |
38 | 0 | private static final Log logger = LogFactory.getLog(RequestContext.class); |
39 | 0 | private static final ThreadLocal currentEvent = new ThreadLocal(); |
40 | |
|
41 | |
|
42 | |
protected RequestContext() |
43 | 0 | { |
44 | |
|
45 | 0 | } |
46 | |
|
47 | |
public static UMOEventContext getEventContext() |
48 | |
{ |
49 | 0 | UMOEvent event = getEvent(); |
50 | 0 | if (event != null) |
51 | |
{ |
52 | 0 | return new MuleEventContext(event); |
53 | |
} |
54 | |
else |
55 | |
{ |
56 | 0 | return null; |
57 | |
} |
58 | |
} |
59 | |
|
60 | |
public static UMOEvent getEvent() |
61 | |
{ |
62 | 0 | return (UMOEvent) currentEvent.get(); |
63 | |
} |
64 | |
|
65 | |
|
66 | |
|
67 | |
|
68 | |
|
69 | |
|
70 | |
|
71 | |
public static UMOEvent setEvent(UMOEvent event) |
72 | |
{ |
73 | 0 | return internalSetEvent(newEvent(event, SAFE, false)); |
74 | |
} |
75 | |
|
76 | |
protected static UMOEvent internalSetEvent(UMOEvent event) |
77 | |
{ |
78 | 0 | currentEvent.set(event); |
79 | 0 | return event; |
80 | |
} |
81 | |
|
82 | |
|
83 | |
|
84 | |
|
85 | |
|
86 | |
|
87 | |
|
88 | |
|
89 | |
|
90 | |
public static UMOMessage rewriteEvent(UMOMessage message) |
91 | |
{ |
92 | 0 | return internalRewriteEvent(message, SAFE, false); |
93 | |
} |
94 | |
|
95 | |
protected static UMOMessage internalRewriteEvent(UMOMessage message, boolean safe, boolean required) |
96 | |
{ |
97 | 0 | if (message != null) |
98 | |
{ |
99 | 0 | UMOEvent event = getEvent(); |
100 | 0 | if (event != null) |
101 | |
{ |
102 | 0 | UMOMessage copy = newMessage(message, safe, required); |
103 | 0 | UMOEvent newEvent = new MuleEvent(copy, event); |
104 | 0 | if (safe) |
105 | |
{ |
106 | 0 | resetAccessControl(copy); |
107 | |
} |
108 | 0 | internalSetEvent(newEvent); |
109 | 0 | return copy; |
110 | |
} |
111 | |
} |
112 | 0 | return message; |
113 | |
} |
114 | |
|
115 | |
public static UMOMessage writeResponse(UMOMessage message) |
116 | |
{ |
117 | 0 | return internalWriteResponse(message, SAFE, false); |
118 | |
} |
119 | |
|
120 | |
protected static UMOMessage internalWriteResponse(UMOMessage message, boolean safe, boolean required) |
121 | |
{ |
122 | 0 | if (message != null) |
123 | |
{ |
124 | 0 | UMOEvent event = getEvent(); |
125 | 0 | if (event != null) |
126 | |
{ |
127 | 0 | UMOMessage copy = newMessage(message, safe, required); |
128 | 0 | combineProperties(event, copy); |
129 | 0 | MuleEvent newEvent = new MuleEvent(copy, event.getEndpoint(), event.getSession(), event.isSynchronous()); |
130 | 0 | if (safe) |
131 | |
{ |
132 | 0 | resetAccessControl(copy); |
133 | |
} |
134 | 0 | internalSetEvent(newEvent); |
135 | 0 | return copy; |
136 | |
} |
137 | |
} |
138 | 0 | return message; |
139 | |
} |
140 | |
|
141 | |
protected static void combineProperties(UMOEvent event, UMOMessage message) |
142 | |
{ |
143 | 0 | for (Iterator iterator = event.getMessage().getPropertyNames().iterator(); iterator.hasNext();) |
144 | |
{ |
145 | 0 | String key = (String) iterator.next(); |
146 | 0 | if (key == null) |
147 | |
{ |
148 | 0 | logger.warn("Message property key is null: please report the following stack trace to dev@mule.codehaus.org.", |
149 | |
new IllegalArgumentException()); |
150 | |
} |
151 | |
else |
152 | |
{ |
153 | 0 | if (key.startsWith(MuleProperties.PROPERTY_PREFIX)) |
154 | |
{ |
155 | 0 | Object newValue = message.getProperty(key); |
156 | 0 | Object oldValue = event.getMessage().getProperty(key); |
157 | 0 | if (newValue == null) |
158 | |
{ |
159 | 0 | message.setProperty(key, oldValue); |
160 | |
} |
161 | 0 | else if (logger.isInfoEnabled() && !newValue.equals(oldValue)) |
162 | |
{ |
163 | 0 | logger.info("Message already contains property " + key + "=" + newValue |
164 | |
+ " not replacing old value: " + oldValue); |
165 | |
} |
166 | |
} |
167 | |
} |
168 | |
} |
169 | 0 | } |
170 | |
|
171 | |
|
172 | |
|
173 | |
|
174 | |
public static void clear() |
175 | |
{ |
176 | 0 | setEvent(null); |
177 | 0 | } |
178 | |
|
179 | |
public static void setExceptionPayload(UMOExceptionPayload exceptionPayload) |
180 | |
{ |
181 | 0 | getEvent().getMessage().setExceptionPayload(exceptionPayload); |
182 | 0 | } |
183 | |
|
184 | |
public static UMOExceptionPayload getExceptionPayload() |
185 | |
{ |
186 | 0 | return getEvent().getMessage().getExceptionPayload(); |
187 | |
} |
188 | |
|
189 | |
|
190 | |
|
191 | |
|
192 | |
protected static void noteUse(String type) |
193 | |
{ |
194 | 0 | if (logger.isDebugEnabled()) |
195 | |
{ |
196 | 0 | logger.debug("copying " + type, new Exception()); |
197 | |
} |
198 | 0 | } |
199 | |
|
200 | |
protected static UMOEvent newEvent(UMOEvent event, boolean safe, boolean required) |
201 | |
{ |
202 | 0 | if (safe && event instanceof ThreadSafeAccess) |
203 | |
{ |
204 | 0 | if (! required) |
205 | |
{ |
206 | 0 | noteUse("event"); |
207 | |
} |
208 | 0 | return (UMOEvent) ((ThreadSafeAccess)event).newThreadCopy(); |
209 | |
} |
210 | |
else |
211 | |
{ |
212 | 0 | return event; |
213 | |
} |
214 | |
} |
215 | |
|
216 | |
protected static UMOMessage newMessage(UMOMessage message, boolean safe, boolean required) |
217 | |
{ |
218 | 0 | if (safe && message instanceof ThreadSafeAccess) |
219 | |
{ |
220 | 0 | if (! required) |
221 | |
{ |
222 | 0 | noteUse("message"); |
223 | |
} |
224 | 0 | return (UMOMessage) ((ThreadSafeAccess)message).newThreadCopy(); |
225 | |
} |
226 | |
else |
227 | |
{ |
228 | 0 | return message; |
229 | |
} |
230 | |
} |
231 | |
|
232 | |
protected static void resetAccessControl(Object object) |
233 | |
{ |
234 | 0 | if (object instanceof ThreadSafeAccess) |
235 | |
{ |
236 | 0 | ((ThreadSafeAccess) object).resetAccessControl(); |
237 | |
} |
238 | 0 | } |
239 | |
|
240 | |
} |