fix of human readable size and media info

[mir.git] / source / mir / producer / EntityModifyingProducerNode.java

diff --git a/source/mir/producer/EntityModifyingProducerNode.java b/source/mir/producer/EntityModifyingProducerNode.java
index 3145235..87a7d66 100755 (executable)
--- a/source/mir/producer/EntityModifyingProducerNode.java
+++ b/source/mir/producer/EntityModifyingProducerNode.java
@@ -18,50 +18,94 @@
  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  *
  * In addition, as a special exception, The Mir-coders gives permission to link
- * the code of this program with  any library licensed under the Apache Software License, 
- * The Sun (tm) Java Advanced Imaging library (JAI), The Sun JIMI library 
- * (or with modified versions of the above that use the same license as the above), 
- * and distribute linked combinations including the two.  You must obey the 
- * GNU General Public License in all respects for all of the code used other than 
- * the above mentioned libraries.  If you modify this file, you may extend this 
- * exception to your version of the file, but you are not obligated to do so.  
+ * the code of this program with  any library licensed under the Apache Software License,
+ * The Sun (tm) Java Advanced Imaging library (JAI), The Sun JIMI library
+ * (or with modified versions of the above that use the same license as the above),
+ * and distribute linked combinations including the two.  You must obey the
+ * GNU General Public License in all respects for all of the code used other than
+ * the above mentioned libraries.  If you modify this file, you may extend this
+ * exception to your version of the file, but you are not obligated to do so.
  * If you do not wish to do so, delete this exception statement from your version.
  */
+
 package mir.producer;
 
-import java.io.PrintWriter;
+import java.util.Date;
+import java.util.HashMap;
+import java.util.Iterator;
 import java.util.Map;
 
+import mir.entity.Entity;
 import mir.entity.adapter.EntityAdapter;
+import mir.entity.adapter.EntityAdapterModel;
+import mir.log.LoggerWrapper;
+import mir.util.JDBCStringRoutines;
 import mir.util.ParameterExpander;
 
 
-public abstract class EntityModifyingProducerNode implements ProducerNode {
-  String entityExpression;
-  String entityField;
-  String valueExpression;
+public class EntityModifyingProducerNode extends AbstractProducerNode {
+  private String entityExpression;
+  private String definition;
+  private Map fields;
+  private boolean create;
+  private EntityAdapterModel model;
 
-  public EntityModifyingProducerNode(String anEntityExpression, String anEntityField, String aValueExpression) {
+  public EntityModifyingProducerNode(EntityAdapterModel aModel, boolean aCreate, String aDefinition, String anEntityExpression, Map aFieldValues) {
     entityExpression = anEntityExpression;
-    entityField = anEntityField;
-    valueExpression = aValueExpression;
+    definition = aDefinition;
+    create = aCreate;
+    model = aModel;
+    fields = new HashMap();
+    fields.putAll(aFieldValues);
   }
 
-  public void produce(Map aValueMap, String aVerb, PrintWriter aLogger) throws ProducerFailure {
-    Object entity;
+  public void addField(String aField, String aValueExpression) {
+    fields.put(aField, aValueExpression);
+  }
 
+  public void produce(Map aValueMap, String aVerb, LoggerWrapper aLogger) throws ProducerExc, ProducerFailure {
     try {
-      entity = ParameterExpander.findValueForKey( aValueMap, entityExpression );
+      Object entityAdapter;
+
+      if (create) {
+        entityAdapter = model.createNewEntity(definition);
+        ParameterExpander.setValueForKey(aValueMap, entityExpression, entityAdapter);
+      }
+      else {
+        entityAdapter = ParameterExpander.findValueForKey(aValueMap, entityExpression);
+      }
+
+      if (entityAdapter instanceof EntityAdapter) {
+        Entity entity = ((EntityAdapter) entityAdapter).getEntity();
+        Iterator i = fields.entrySet().iterator();
+        while (i.hasNext()) {
+          Map.Entry entry = (Map.Entry) i.next();
+          String entityField = (String) entry.getKey();
+          String valueExpression = (String) entry.getValue();
+
+          Object value = ParameterExpander.evaluateExpression(aValueMap, valueExpression);
+
+          if (value instanceof String)
+            entity.setFieldValue(entityField, (String) value);
+          else if (value instanceof EntityAdapter)
+            entity.setFieldValue(entityField, ((EntityAdapter) value).getEntity().getId());
+          else if (value instanceof Date) {
+            entity.setFieldValue(entityField, JDBCStringRoutines.formatDate((Date) value));
+          }
+          else
+            aLogger.warn("Can't set value " + value + " for field " + entityField);
+        }
 
-      if (entity instanceof EntityAdapter) {
-        ((EntityAdapter) entity).getEntity().setValueForProperty(entityField, valueExpression);
-        ((EntityAdapter) entity).getEntity().update();
+        if (create)
+          entity.insert();
+        else
+          entity.update();
       }
       else
-        throw new Exception( entityExpression + " does not evaluate to an entity");
+        throw new ProducerExc( entityExpression + " does not evaluate to an entity");
     }
     catch (Throwable t) {
-      aLogger.println("Error while performing entity modifying operation: " + t.getMessage());
+      aLogger.error("Error while performing entity modification operation: " + t.getMessage(), t);
 
       throw new ProducerFailure(t.getMessage(), t);
     }