1 |
|
|
2 |
|
|
3 |
|
|
4 |
|
|
5 |
|
|
6 |
|
|
7 |
|
|
8 |
|
|
9 |
|
|
10 |
|
|
11 |
|
|
12 |
|
|
13 |
|
|
14 |
|
|
15 |
|
|
16 |
|
|
17 |
|
package org.apache.camel.builder.script; |
18 |
|
|
19 |
|
import java.io.File; |
20 |
|
import java.io.IOException; |
21 |
|
import java.io.InputStreamReader; |
22 |
|
import java.net.URL; |
23 |
|
|
24 |
|
import javax.script.Compilable; |
25 |
|
import javax.script.CompiledScript; |
26 |
|
import javax.script.ScriptContext; |
27 |
|
import javax.script.ScriptEngine; |
28 |
|
import javax.script.ScriptEngineManager; |
29 |
|
import javax.script.ScriptException; |
30 |
|
|
31 |
|
import org.apache.camel.Exchange; |
32 |
|
import org.apache.camel.Expression; |
33 |
|
import org.apache.camel.Predicate; |
34 |
|
import org.apache.camel.Processor; |
35 |
|
import org.apache.camel.converter.ObjectConverter; |
36 |
|
import org.apache.commons.logging.Log; |
37 |
|
import org.apache.commons.logging.LogFactory; |
38 |
|
|
39 |
|
import org.springframework.core.io.FileSystemResource; |
40 |
|
import org.springframework.core.io.Resource; |
41 |
|
import org.springframework.core.io.UrlResource; |
42 |
|
|
43 |
|
|
44 |
|
|
45 |
|
|
46 |
|
|
47 |
|
|
48 |
|
|
49 |
9 |
public class ScriptBuilder<E extends Exchange> implements Expression<E>, Predicate<E>, Processor { |
50 |
1 |
private static final transient Log LOG = LogFactory.getLog(ScriptBuilder.class); |
51 |
|
|
52 |
|
private String scriptEngineName; |
53 |
|
private Resource scriptResource; |
54 |
|
private String scriptText; |
55 |
|
private ScriptEngine engine; |
56 |
|
private CompiledScript compiledScript; |
57 |
|
|
58 |
13 |
public ScriptBuilder(String scriptEngineName) { |
59 |
13 |
this.scriptEngineName = scriptEngineName; |
60 |
13 |
} |
61 |
|
|
62 |
|
public ScriptBuilder(String scriptEngineName, String scriptText) { |
63 |
13 |
this(scriptEngineName); |
64 |
13 |
this.scriptText = scriptText; |
65 |
13 |
} |
66 |
|
|
67 |
|
public ScriptBuilder(String scriptEngineName, Resource scriptResource) { |
68 |
0 |
this(scriptEngineName); |
69 |
0 |
this.scriptResource = scriptResource; |
70 |
0 |
} |
71 |
|
|
72 |
|
@Override |
73 |
|
public String toString() { |
74 |
20 |
return getScriptDescription(); |
75 |
|
} |
76 |
|
|
77 |
|
public Object evaluate(E exchange) { |
78 |
6 |
return evaluateScript(exchange); |
79 |
|
} |
80 |
|
|
81 |
|
public boolean matches(E exchange) { |
82 |
6 |
Object scriptValue = evaluateScript(exchange); |
83 |
6 |
return matches(exchange, scriptValue); |
84 |
|
} |
85 |
|
|
86 |
|
public void assertMatches(String text, E exchange) throws AssertionError { |
87 |
3 |
Object scriptValue = evaluateScript(exchange); |
88 |
3 |
if (!matches(exchange, scriptValue)) { |
89 |
0 |
throw new AssertionError(this + " failed on " + exchange + " as script returned <" + scriptValue + ">"); |
90 |
|
} |
91 |
3 |
} |
92 |
|
|
93 |
|
public void process(Exchange exchange) { |
94 |
1 |
evaluateScript(exchange); |
95 |
1 |
} |
96 |
|
|
97 |
|
|
98 |
|
|
99 |
|
|
100 |
|
|
101 |
|
|
102 |
|
|
103 |
|
|
104 |
|
|
105 |
|
|
106 |
|
|
107 |
|
|
108 |
|
public ScriptBuilder attribute(String name, Object value) { |
109 |
1 |
getScriptContext().setAttribute(name, value, ScriptContext.ENGINE_SCOPE); |
110 |
1 |
return this; |
111 |
|
} |
112 |
|
|
113 |
|
|
114 |
|
|
115 |
|
|
116 |
|
|
117 |
|
|
118 |
|
|
119 |
|
|
120 |
|
|
121 |
|
|
122 |
|
|
123 |
|
public static ScriptBuilder script(String language, String scriptText) { |
124 |
0 |
return new ScriptBuilder(language, scriptText); |
125 |
|
} |
126 |
|
|
127 |
|
|
128 |
|
|
129 |
|
|
130 |
|
|
131 |
|
|
132 |
|
|
133 |
|
|
134 |
|
|
135 |
|
public static ScriptBuilder script(String language, Resource scriptResource) { |
136 |
0 |
return new ScriptBuilder(language, scriptResource); |
137 |
|
} |
138 |
|
|
139 |
|
|
140 |
|
|
141 |
|
|
142 |
|
|
143 |
|
|
144 |
|
|
145 |
|
|
146 |
|
|
147 |
|
public static ScriptBuilder script(String language, File scriptFile) { |
148 |
0 |
return new ScriptBuilder(language, new FileSystemResource(scriptFile)); |
149 |
|
} |
150 |
|
|
151 |
|
|
152 |
|
|
153 |
|
|
154 |
|
|
155 |
|
|
156 |
|
|
157 |
|
|
158 |
|
|
159 |
|
public static ScriptBuilder script(String language, URL scriptURL) { |
160 |
0 |
return new ScriptBuilder(language, new UrlResource(scriptURL)); |
161 |
|
} |
162 |
|
|
163 |
|
|
164 |
|
|
165 |
|
|
166 |
|
|
167 |
|
|
168 |
|
|
169 |
|
|
170 |
|
|
171 |
|
|
172 |
|
public static ScriptBuilder groovy(String scriptText) { |
173 |
9 |
return new ScriptBuilder("groovy", scriptText); |
174 |
|
} |
175 |
|
|
176 |
|
|
177 |
|
|
178 |
|
|
179 |
|
|
180 |
|
|
181 |
|
|
182 |
|
|
183 |
|
public static ScriptBuilder groovy(Resource scriptResource) { |
184 |
0 |
return new ScriptBuilder("groovy", scriptResource); |
185 |
|
} |
186 |
|
|
187 |
|
|
188 |
|
|
189 |
|
|
190 |
|
|
191 |
|
|
192 |
|
|
193 |
|
|
194 |
|
public static ScriptBuilder groovy(File scriptFile) { |
195 |
0 |
return new ScriptBuilder("groovy", new FileSystemResource(scriptFile)); |
196 |
|
} |
197 |
|
|
198 |
|
|
199 |
|
|
200 |
|
|
201 |
|
|
202 |
|
|
203 |
|
|
204 |
|
|
205 |
|
public static ScriptBuilder groovy(URL scriptURL) { |
206 |
0 |
return new ScriptBuilder("groovy", new UrlResource(scriptURL)); |
207 |
|
} |
208 |
|
|
209 |
|
|
210 |
|
|
211 |
|
|
212 |
|
|
213 |
|
|
214 |
|
|
215 |
|
|
216 |
|
|
217 |
|
|
218 |
|
public static ScriptBuilder javaScript(String scriptText) { |
219 |
0 |
return new ScriptBuilder("js", scriptText); |
220 |
|
} |
221 |
|
|
222 |
|
|
223 |
|
|
224 |
|
|
225 |
|
|
226 |
|
|
227 |
|
|
228 |
|
|
229 |
|
public static ScriptBuilder javaScript(Resource scriptResource) { |
230 |
0 |
return new ScriptBuilder("js", scriptResource); |
231 |
|
} |
232 |
|
|
233 |
|
|
234 |
|
|
235 |
|
|
236 |
|
|
237 |
|
|
238 |
|
|
239 |
|
|
240 |
|
public static ScriptBuilder javaScript(File scriptFile) { |
241 |
0 |
return new ScriptBuilder("js", new FileSystemResource(scriptFile)); |
242 |
|
} |
243 |
|
|
244 |
|
|
245 |
|
|
246 |
|
|
247 |
|
|
248 |
|
|
249 |
|
|
250 |
|
|
251 |
|
public static ScriptBuilder javaScript(URL scriptURL) { |
252 |
0 |
return new ScriptBuilder("js", new UrlResource(scriptURL)); |
253 |
|
} |
254 |
|
|
255 |
|
|
256 |
|
|
257 |
|
|
258 |
|
|
259 |
|
|
260 |
|
|
261 |
|
|
262 |
|
|
263 |
|
|
264 |
|
public static ScriptBuilder php(String scriptText) { |
265 |
0 |
return new ScriptBuilder("php", scriptText); |
266 |
|
} |
267 |
|
|
268 |
|
|
269 |
|
|
270 |
|
|
271 |
|
|
272 |
|
|
273 |
|
|
274 |
|
|
275 |
|
public static ScriptBuilder php(Resource scriptResource) { |
276 |
0 |
return new ScriptBuilder("php", scriptResource); |
277 |
|
} |
278 |
|
|
279 |
|
|
280 |
|
|
281 |
|
|
282 |
|
|
283 |
|
|
284 |
|
|
285 |
|
|
286 |
|
public static ScriptBuilder php(File scriptFile) { |
287 |
0 |
return new ScriptBuilder("php", new FileSystemResource(scriptFile)); |
288 |
|
} |
289 |
|
|
290 |
|
|
291 |
|
|
292 |
|
|
293 |
|
|
294 |
|
|
295 |
|
|
296 |
|
|
297 |
|
public static ScriptBuilder php(URL scriptURL) { |
298 |
0 |
return new ScriptBuilder("php", new UrlResource(scriptURL)); |
299 |
|
} |
300 |
|
|
301 |
|
|
302 |
|
|
303 |
|
|
304 |
|
|
305 |
|
|
306 |
|
|
307 |
|
|
308 |
|
|
309 |
|
|
310 |
|
public static ScriptBuilder python(String scriptText) { |
311 |
0 |
return new ScriptBuilder("python", scriptText); |
312 |
|
} |
313 |
|
|
314 |
|
|
315 |
|
|
316 |
|
|
317 |
|
|
318 |
|
|
319 |
|
|
320 |
|
|
321 |
|
public static ScriptBuilder python(Resource scriptResource) { |
322 |
0 |
return new ScriptBuilder("python", scriptResource); |
323 |
|
} |
324 |
|
|
325 |
|
|
326 |
|
|
327 |
|
|
328 |
|
|
329 |
|
|
330 |
|
|
331 |
|
|
332 |
|
public static ScriptBuilder python(File scriptFile) { |
333 |
0 |
return new ScriptBuilder("python", new FileSystemResource(scriptFile)); |
334 |
|
} |
335 |
|
|
336 |
|
|
337 |
|
|
338 |
|
|
339 |
|
|
340 |
|
|
341 |
|
|
342 |
|
|
343 |
|
public static ScriptBuilder python(URL scriptURL) { |
344 |
0 |
return new ScriptBuilder("python", new UrlResource(scriptURL)); |
345 |
|
} |
346 |
|
|
347 |
|
|
348 |
|
|
349 |
|
|
350 |
|
|
351 |
|
|
352 |
|
|
353 |
|
|
354 |
|
|
355 |
|
|
356 |
|
public static ScriptBuilder ruby(String scriptText) { |
357 |
0 |
return new ScriptBuilder("jruby", scriptText); |
358 |
|
} |
359 |
|
|
360 |
|
|
361 |
|
|
362 |
|
|
363 |
|
|
364 |
|
|
365 |
|
|
366 |
|
|
367 |
|
public static ScriptBuilder ruby(Resource scriptResource) { |
368 |
0 |
return new ScriptBuilder("jruby", scriptResource); |
369 |
|
} |
370 |
|
|
371 |
|
|
372 |
|
|
373 |
|
|
374 |
|
|
375 |
|
|
376 |
|
|
377 |
|
|
378 |
|
public static ScriptBuilder ruby(File scriptFile) { |
379 |
0 |
return new ScriptBuilder("jruby", new FileSystemResource(scriptFile)); |
380 |
|
} |
381 |
|
|
382 |
|
|
383 |
|
|
384 |
|
|
385 |
|
|
386 |
|
|
387 |
|
|
388 |
|
|
389 |
|
public static ScriptBuilder ruby(URL scriptURL) { |
390 |
0 |
return new ScriptBuilder("jruby", new UrlResource(scriptURL)); |
391 |
|
} |
392 |
|
|
393 |
|
|
394 |
|
|
395 |
|
public ScriptEngine getEngine() { |
396 |
33 |
checkInitialised(); |
397 |
33 |
return engine; |
398 |
|
} |
399 |
|
|
400 |
|
public CompiledScript getCompiledScript() { |
401 |
0 |
return compiledScript; |
402 |
|
} |
403 |
|
|
404 |
|
public String getScriptText() { |
405 |
0 |
return scriptText; |
406 |
|
} |
407 |
|
|
408 |
|
public void setScriptText(String scriptText) { |
409 |
0 |
this.scriptText = scriptText; |
410 |
0 |
} |
411 |
|
|
412 |
|
public String getScriptEngineName() { |
413 |
0 |
return scriptEngineName; |
414 |
|
} |
415 |
|
|
416 |
|
|
417 |
|
|
418 |
|
|
419 |
|
|
420 |
|
|
421 |
|
public String getScriptDescription() { |
422 |
21 |
if (scriptText != null) { |
423 |
21 |
return scriptEngineName + ": " + scriptText; |
424 |
0 |
} else if (scriptResource != null) { |
425 |
0 |
return scriptEngineName + ": " + scriptResource.getDescription(); |
426 |
|
} else { |
427 |
0 |
return scriptEngineName + ": null script"; |
428 |
|
} |
429 |
|
} |
430 |
|
|
431 |
|
|
432 |
|
|
433 |
|
|
434 |
|
|
435 |
|
public ScriptContext getScriptContext() { |
436 |
17 |
return getEngine().getContext(); |
437 |
|
} |
438 |
|
|
439 |
|
|
440 |
|
|
441 |
|
|
442 |
|
public void setScriptContext(ScriptContext scriptContext) { |
443 |
0 |
getEngine().setContext(scriptContext); |
444 |
0 |
} |
445 |
|
|
446 |
|
public Resource getScriptResource() { |
447 |
0 |
return scriptResource; |
448 |
|
} |
449 |
|
|
450 |
|
public void setScriptResource(Resource scriptResource) { |
451 |
0 |
this.scriptResource = scriptResource; |
452 |
0 |
} |
453 |
|
|
454 |
|
|
455 |
|
|
456 |
|
protected void checkInitialised() { |
457 |
49 |
if (scriptText == null && scriptResource == null) { |
458 |
0 |
throw new IllegalArgumentException("Neither scriptText or scriptResource are specified"); |
459 |
|
} |
460 |
49 |
if (engine == null) { |
461 |
13 |
engine = createScriptEngine(); |
462 |
|
} |
463 |
49 |
if (compiledScript == null) { |
464 |
13 |
if (engine instanceof Compilable) { |
465 |
13 |
compileScript((Compilable)engine); |
466 |
|
} |
467 |
|
} |
468 |
49 |
} |
469 |
|
|
470 |
|
protected boolean matches(E exchange, Object scriptValue) { |
471 |
9 |
return ObjectConverter.toBoolean(scriptValue); |
472 |
|
} |
473 |
|
|
474 |
|
protected ScriptEngine createScriptEngine() { |
475 |
13 |
ScriptEngineManager manager = new ScriptEngineManager(); |
476 |
13 |
return manager.getEngineByName(scriptEngineName); |
477 |
|
} |
478 |
|
|
479 |
|
protected void compileScript(Compilable compilable) { |
480 |
|
try { |
481 |
13 |
if (scriptText != null) { |
482 |
13 |
compiledScript = compilable.compile(scriptText); |
483 |
13 |
} else if (scriptResource != null) { |
484 |
0 |
compiledScript = compilable.compile(createScriptReader()); |
485 |
|
} |
486 |
0 |
} catch (ScriptException e) { |
487 |
0 |
if (LOG.isDebugEnabled()) { |
488 |
0 |
LOG.debug("Script compile failed: " + e, e); |
489 |
|
} |
490 |
0 |
throw createScriptCompileException(e); |
491 |
0 |
} catch (IOException e) { |
492 |
0 |
throw createScriptCompileException(e); |
493 |
13 |
} |
494 |
13 |
} |
495 |
|
|
496 |
|
protected synchronized Object evaluateScript(Exchange exchange) { |
497 |
|
try { |
498 |
16 |
getScriptContext(); |
499 |
16 |
populateBindings(getEngine(), exchange); |
500 |
16 |
return runScript(); |
501 |
1 |
} catch (ScriptException e) { |
502 |
1 |
if (LOG.isDebugEnabled()) { |
503 |
0 |
LOG.debug("Script evaluation failed: " + e, e); |
504 |
|
} |
505 |
1 |
throw createScriptEvaluationException(e.getCause()); |
506 |
0 |
} catch (IOException e) { |
507 |
0 |
throw createScriptEvaluationException(e); |
508 |
|
} |
509 |
|
} |
510 |
|
|
511 |
|
protected Object runScript() throws ScriptException, IOException { |
512 |
16 |
checkInitialised(); |
513 |
16 |
if (compiledScript != null) { |
514 |
16 |
return compiledScript.eval(); |
515 |
|
} else { |
516 |
0 |
if (scriptText != null) { |
517 |
0 |
return getEngine().eval(scriptText); |
518 |
|
} else { |
519 |
0 |
return getEngine().eval(createScriptReader()); |
520 |
|
} |
521 |
|
} |
522 |
|
} |
523 |
|
|
524 |
|
protected void populateBindings(ScriptEngine engine, Exchange exchange) { |
525 |
16 |
ScriptContext context = engine.getContext(); |
526 |
16 |
int scope = ScriptContext.ENGINE_SCOPE; |
527 |
16 |
context.setAttribute("context", exchange.getContext(), scope); |
528 |
16 |
context.setAttribute("exchange", exchange, scope); |
529 |
16 |
context.setAttribute("request", exchange.getIn(), scope); |
530 |
16 |
context.setAttribute("response", exchange.getOut(), scope); |
531 |
16 |
} |
532 |
|
|
533 |
|
protected InputStreamReader createScriptReader() throws IOException { |
534 |
|
|
535 |
0 |
return new InputStreamReader(scriptResource.getInputStream()); |
536 |
|
} |
537 |
|
|
538 |
|
protected ScriptEvaluationException createScriptCompileException(Exception e) { |
539 |
0 |
return new ScriptEvaluationException("Failed to compile: " + getScriptDescription() + ". Cause: " + e, e); |
540 |
|
} |
541 |
|
|
542 |
|
protected ScriptEvaluationException createScriptEvaluationException(Throwable e) { |
543 |
1 |
return new ScriptEvaluationException("Failed to evaluate: " + getScriptDescription() + ". Cause: " + e, e); |
544 |
|
} |
545 |
|
|
546 |
|
} |