rss input update
[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.Map;
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
43
44 public class EntityModifyingProducerNode implements ProducerNode {
45   private String entityExpression;
46   private String definition;
47   private Map fields;
48   private boolean create;
49   private EntityAdapterModel model;
50
51   public EntityModifyingProducerNode(EntityAdapterModel aModel, boolean aCreate, String aDefinition, String anEntityExpression, Map aFieldValues) {
52     entityExpression = anEntityExpression;
53     definition = aDefinition;
54     create = aCreate;
55     model = aModel;
56     fields = new HashMap();
57     fields.putAll(aFieldValues);
58   }
59
60   public void addField(String aField, String aValueExpression) {
61     fields.put(aField, aValueExpression);
62   }
63
64   public void produce(Map aValueMap, String aVerb, LoggerWrapper aLogger) throws ProducerExc, ProducerFailure {
65     try {
66       Object entityAdapter;
67
68       if (create) {
69         entityAdapter = model.createNewEntity(definition);
70         ParameterExpander.setValueForKey(aValueMap, entityExpression, entityAdapter);
71       }
72       else {
73         entityAdapter = ParameterExpander.findValueForKey(aValueMap, entityExpression);
74       }
75
76       if (entityAdapter instanceof EntityAdapter) {
77         Entity entity = ((EntityAdapter) entityAdapter).getEntity();
78         Iterator i = fields.entrySet().iterator();
79         while (i.hasNext()) {
80           Map.Entry entry = (Map.Entry) i.next();
81           String entityField = (String) entry.getKey();
82           String valueExpression = (String) entry.getValue();
83
84           Object value = ParameterExpander.evaluateExpression(aValueMap, valueExpression);
85
86           if (value instanceof String)
87             entity.setValueForProperty(entityField, (String) value);
88           else if (value instanceof EntityAdapter)
89             entity.setValueForProperty(entityField, ((EntityAdapter) value).getEntity().getId());
90           else
91             aLogger.warn("Can't set value " + value.toString() + " for field " + entityField);
92         }
93         entity.update();
94       }
95       else
96         throw new ProducerExc( entityExpression + " does not evaluate to an entity");
97     }
98     catch (Throwable t) {
99       aLogger.error("Error while performing entity modification operation: " + t.getMessage());
100
101       throw new ProducerFailure(t.getMessage(), t);
102     }
103   }
104 }