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