source/mir/producer/EntityModifyingProducerNode.java

   1 /*
   2  * Copyright (C) 2001-2006 The Mir-coders group
   3  *
   4  * This file is part of Mir.
   5  *
   6  * Mir is free software; you can redistribute it and/or modify
   7  * it under the terms of the GNU General Public License as published by
   8  * the Free Software Foundation; either version 2 of the License, or
   9  * (at your option) any later version.
  10  *
  11  * Mir is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14  * GNU General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU General Public License
  17  * along with Mir; if not, write to the Free Software
  18  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  19  *
  20  * In addition, as a special exception, The Mir-coders gives permission to link
  21  * the code of this program with  any library licensed under the Apache Software License,
  22  * and distribute linked combinations including the two.  You must obey the
  23  * GNU General Public License in all respects for all of the code used other than
  24  * the above mentioned libraries.  If you modify this file, you may extend this
  25  * exception to your version of the file, but you are not obligated to do so.
  26  * If you do not wish to do so, delete this exception statement from your version.
  27  */
  28 package mir.producer;
  29
  30 import mir.entity.Entity;
  31 import mir.entity.adapter.EntityAdapter;
  32 import mir.entity.adapter.EntityAdapterModel;
  33 import mir.log.LoggerWrapper;
  34 import mir.util.JDBCStringRoutines;
  35 import mir.util.ParameterExpander;
  36
  37 import java.util.Date;
  38 import java.util.HashMap;
  39 import java.util.Iterator;
  40 import java.util.Map;
  41
  42
  43 public class EntityModifyingProducerNode extends AbstractProducerNode {
  44   private String entityExpression;
  45   private String definition;
  46   private Map fields;
  47   private boolean create;
  48   private EntityAdapterModel model;
  49
  50   public EntityModifyingProducerNode(EntityAdapterModel aModel, boolean aCreate, String aDefinition, String anEntityExpression, Map aFieldValues) {
  51     entityExpression = anEntityExpression;
  52     definition = aDefinition;
  53     create = aCreate;
  54     model = aModel;
  55     fields = new HashMap();
  56     fields.putAll(aFieldValues);
  57   }
  58
  59   public void addField(String aField, String aValueExpression) {
  60     fields.put(aField, aValueExpression);
  61   }
  62
  63   public void produce(Map aValueMap, String aVerb, LoggerWrapper aLogger) throws ProducerExc, ProducerFailure {
  64     try {
  65       Object entityAdapter;
  66
  67       if (create) {
  68         entityAdapter = model.createNewEntity(definition);
  69         ParameterExpander.setValueForKey(aValueMap, entityExpression, entityAdapter);
  70       }
  71       else {
  72         entityAdapter = ParameterExpander.findValueForKey(aValueMap, entityExpression);
  73       }
  74
  75       if (entityAdapter instanceof EntityAdapter) {
  76         Entity entity = ((EntityAdapter) entityAdapter).getEntity();
  77         Iterator i = fields.entrySet().iterator();
  78         while (i.hasNext()) {
  79           Map.Entry entry = (Map.Entry) i.next();
  80           String entityField = (String) entry.getKey();
  81           String valueExpression = (String) entry.getValue();
  82
  83           Object value = ParameterExpander.evaluateExpression(aValueMap, valueExpression);
  84
  85           if (value instanceof String)
  86             entity.setFieldValue(entityField, (String) value);
  87           else if (value instanceof EntityAdapter)
  88             entity.setFieldValue(entityField, ((EntityAdapter) value).getEntity().getId());
  89           else if (value instanceof Date) {
  90             entity.setFieldValue(entityField, JDBCStringRoutines.formatDate((Date) value));
  91           }
  92           else
  93             aLogger.warn("Can't set value " + value + " for field " + entityField);
  94         }
  95
  96         if (create)
  97           entity.insert();
  98         else
  99           entity.update();
 100       }
 101       else
 102         throw new ProducerExc( entityExpression + " does not evaluate to an entity");
 103     }
 104     catch (Throwable t) {
 105       aLogger.error("Error while performing entity modification operation: " + t.getMessage(), t);
 106
 107       throw new ProducerFailure(t.getMessage(), t);
 108     }
 109   }
 110 }