source/mir/producer/EntityModifyingProducerNode.java

   1 /*
   2  * Copyright (C) 2001, 2002 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  * The Sun (tm) Java Advanced Imaging library (JAI), The Sun JIMI library
  23  * (or with modified versions of the above that use the same license as the above),
  24  * and distribute linked combinations including the two.  You must obey the
  25  * GNU General Public License in all respects for all of the code used other than
  26  * the above mentioned libraries.  If you modify this file, you may extend this
  27  * exception to your version of the file, but you are not obligated to do so.
  28  * If you do not wish to do so, delete this exception statement from your version.
  29  */
  30
  31 package mir.producer;
  32
  33 import java.util.Date;
  34 import java.util.HashMap;
  35 import java.util.Iterator;
  36 import java.util.Map;
  37
  38 import mir.entity.Entity;
  39 import mir.entity.adapter.EntityAdapter;
  40 import mir.entity.adapter.EntityAdapterModel;
  41 import mir.log.LoggerWrapper;
  42 import mir.util.JDBCStringRoutines;
  43 import mir.util.ParameterExpander;
  44
  45
  46 public class EntityModifyingProducerNode extends AbstractProducerNode {
  47   private String entityExpression;
  48   private String definition;
  49   private Map fields;
  50   private boolean create;
  51   private EntityAdapterModel model;
  52
  53   public EntityModifyingProducerNode(EntityAdapterModel aModel, boolean aCreate, String aDefinition, String anEntityExpression, Map aFieldValues) {
  54     entityExpression = anEntityExpression;
  55     definition = aDefinition;
  56     create = aCreate;
  57     model = aModel;
  58     fields = new HashMap();
  59     fields.putAll(aFieldValues);
  60   }
  61
  62   public void addField(String aField, String aValueExpression) {
  63     fields.put(aField, aValueExpression);
  64   }
  65
  66   public void produce(Map aValueMap, String aVerb, LoggerWrapper aLogger) throws ProducerExc, ProducerFailure {
  67     try {
  68       Object entityAdapter;
  69
  70       if (create) {
  71         entityAdapter = model.createNewEntity(definition);
  72         ParameterExpander.setValueForKey(aValueMap, entityExpression, entityAdapter);
  73       }
  74       else {
  75         entityAdapter = ParameterExpander.findValueForKey(aValueMap, entityExpression);
  76       }
  77
  78       if (entityAdapter instanceof EntityAdapter) {
  79         Entity entity = ((EntityAdapter) entityAdapter).getEntity();
  80         Iterator i = fields.entrySet().iterator();
  81         while (i.hasNext()) {
  82           Map.Entry entry = (Map.Entry) i.next();
  83           String entityField = (String) entry.getKey();
  84           String valueExpression = (String) entry.getValue();
  85
  86           Object value = ParameterExpander.evaluateExpression(aValueMap, valueExpression);
  87
  88           if (value instanceof String)
  89             entity.setFieldValue(entityField, (String) value);
  90           else if (value instanceof EntityAdapter)
  91             entity.setFieldValue(entityField, ((EntityAdapter) value).getEntity().getId());
  92           else if (value instanceof Date) {
  93             entity.setFieldValue(entityField, JDBCStringRoutines.formatDate((Date) value));
  94           }
  95           else
  96             aLogger.warn("Can't set value " + value + " for field " + entityField);
  97         }
  98
  99         if (create)
 100           entity.insert();
 101         else
 102           entity.update();
 103       }
 104       else
 105         throw new ProducerExc( entityExpression + " does not evaluate to an entity");
 106     }
 107     catch (Throwable t) {
 108       aLogger.error("Error while performing entity modification operation: " + t.getMessage(), t);
 109
 110       throw new ProducerFailure(t.getMessage(), t);
 111     }
 112   }
 113 }