766565a07631f365f5cfaab243e7bc56c125fd92
[mir.git] / 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.HashMap;
34 import java.util.Iterator;
35 import java.util.*;
36
37 import mir.entity.Entity;
38 import mir.entity.adapter.EntityAdapter;
39 import mir.entity.adapter.EntityAdapterModel;
40 import mir.log.LoggerWrapper;
41 import mir.util.ParameterExpander;
42 import mir.util.*;
43
44
45 public class EntityModifyingProducerNode implements ProducerNode {
46   private String entityExpression;
47   private String definition;
48   private Map fields;
49   private boolean create;
50   private EntityAdapterModel model;
51
52   public EntityModifyingProducerNode(EntityAdapterModel aModel, boolean aCreate, String aDefinition, String anEntityExpression, Map aFieldValues) {
53     entityExpression = anEntityExpression;
54     definition = aDefinition;
55     create = aCreate;
56     model = aModel;
57     fields = new HashMap();
58     fields.putAll(aFieldValues);
59   }
60
61   public void addField(String aField, String aValueExpression) {
62     fields.put(aField, aValueExpression);
63   }
64
65   public void produce(Map aValueMap, String aVerb, LoggerWrapper aLogger) throws ProducerExc, ProducerFailure {
66     try {
67       Object entityAdapter;
68
69       if (create) {
70         entityAdapter = model.createNewEntity(definition);
71         ParameterExpander.setValueForKey(aValueMap, entityExpression, entityAdapter);
72       }
73       else {
74         entityAdapter = ParameterExpander.findValueForKey(aValueMap, entityExpression);
75       }
76
77       if (entityAdapter instanceof EntityAdapter) {
78         Entity entity = ((EntityAdapter) entityAdapter).getEntity();
79         Iterator i = fields.entrySet().iterator();
80         while (i.hasNext()) {
81           Map.Entry entry = (Map.Entry) i.next();
82           String entityField = (String) entry.getKey();
83           String valueExpression = (String) entry.getValue();
84
85           Object value = ParameterExpander.evaluateExpression(aValueMap, valueExpression);
86
87           aLogger.debug("  setting " + entityField + " to " + valueExpression+ " ("+ value + ")");
88
89 //          if (value==null)
90 //            entity.setValueForProperty(entityField, "");
91           if (value instanceof String)
92             entity.setValueForProperty(entityField, (String) value);
93           else if (value instanceof EntityAdapter)
94             entity.setValueForProperty(entityField, ((EntityAdapter) value).getEntity().getId());
95           else if (value instanceof Date) {
96             entity.setValueForProperty(entityField, JDBCStringRoutines.formatDate((Date) value));
97           }
98           else
99             aLogger.warn("Can't set value " + value + " for field " + entityField);
100         }
101
102         if (create)
103           entity.insert();
104         else
105           entity.update();
106       }
107       else
108         throw new ProducerExc( entityExpression + " does not evaluate to an entity");
109     }
110     catch (Throwable t) {
111       aLogger.error("Error while performing entity modification operation: " + t.getMessage());
112       t.printStackTrace(aLogger.asPrintWriter(aLogger.DEBUG_MESSAGE));
113
114       throw new ProducerFailure(t.getMessage(), t);
115     }
116   }
117 }