Changeset 1436 for trunk/universe/kauri-template/src/main/java/org/kauriproject/template/VariableBlock.java

Timestamp:

2010-02-16 09:06:52 (2 years ago)

Author:

mpo

Message:

fix for #191

File:

• trunk/universe/kauri-template/src/main/java/org/kauriproject/template/VariableBlock.java (modified) (4 diffs)

trunk/universe/kauri-template/src/main/java/org/kauriproject/template/VariableBlock.java

@@ -59 +59 @@
     }
 
+    private static final Object NO_VALUE = new Object();
+
     class StartStep extends Step {
 
@@ -101 +103 @@
         }
 
+        
         @Override
         public Step executeAndProceed(ExecutionContext context, TemplateResult result) throws SAXException {
             TemplateContext tmplContext = context.getTemplateContext();
             if (overwrite || !tmplContext.containsKey(name)) {
+                Object value = NO_VALUE;
                 if (valueExpression != null) {
-                    final Object value = valueExpression.evaluate(context.getTemplateContext());
-                    tmplContext.put(name, value);
-                } else if (srcExpression != null) {
-                    Object data = loadFromSrc(context);
-                    tmplContext.put(name, data);
+                    value = valueExpression.evaluate(context.getTemplateContext());
                 } else {
                     // use element body as value
@@ -123 +123 @@
                     buffer.endDocument();
                     buffer.flush();
-                    String value;
                     try {
                         bos.flush();
-                        value = bos.toString("UTF-8");
+                        final String content = bos.toString("UTF-8");
+                        if (content != null && content.length() > 0)
+                            value = content; 
                     } catch (IOException ex) {
                         throw new TemplateException("Error building variable value.", ex);
                     }
-
-                    tmplContext.put(name, value);
-                }
-            }
+                }
+
+                // value retrieved from @value or body can be overwritten by loaded data from @src
+                if (srcExpression != null) {
+                    value = loadFromSrc(context, value);
+                }
+
+                tmplContext.put(name, value);
+            }
+
             return endStep.getCompiledNext();
         }
 
-        private Object loadFromSrc(ExecutionContext context) {
+        private Object loadFromSrc(ExecutionContext context, Object fallbackValue) {
             String src = (String)srcExpression.evaluate(context.getTemplateContext());
             String accept = acceptExpression != null ? (String)acceptExpression.evaluate(context.getTemplateContext()) : null;
@@ -152 +159 @@
                 }
             } catch (Throwable e) {
+                if (fallbackValue != NO_VALUE)
+                    return fallbackValue;
+                // else
                 throw new TemplateException("Some error occured while loading data for variable " + name + " from URI " + src, e);
             } finally {