369916bfe8a78568b7898113186eaa98c6f6814b
[mir.git] / source / mir / producer / FreeQueryProducerNode.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.Map;
34
35 import mir.log.LoggerWrapper;
36 import mir.storage.Database;
37 import mir.util.ParameterExpander;
38
39 public class FreeQueryProducerNode extends AbstractProducerNode {
40   public static final int QUERY_TYPE_SET = 1;
41   public static final int QUERY_TYPE_ROW = 2;
42   public static final int QUERY_TYPE_VALUE = 3;
43   public static final int QUERY_TYPE_UPDATE = 4;
44
45   private Database database = new Database();
46   private String key;
47   private String query;
48   private String limitExpression;
49   private int type;
50
51   public FreeQueryProducerNode(String aKey, String aQuery, String aLimit, int aType) {
52     key = aKey;
53     query = aQuery;
54     limitExpression = aLimit;
55     type = aType;
56   }
57
58   public void produce(Map aValueMap, String aVerb, LoggerWrapper aLogger) throws ProducerFailure {
59     Object result = null;
60
61     try {
62       switch (type) {
63         case QUERY_TYPE_UPDATE:
64           result = new Integer(database.executeUpdate(ParameterExpander.expandExpression(aValueMap, query)));
65           break;
66         case QUERY_TYPE_VALUE:
67           result = database.executeFreeSingleValueSql(ParameterExpander.expandExpression(aValueMap, query));
68           break;
69
70         case QUERY_TYPE_ROW:
71           result = database.executeFreeSingleRowSql(ParameterExpander.expandExpression(aValueMap, query));
72           break;
73
74         case QUERY_TYPE_SET:
75           int limit=10;
76           if (limitExpression!=null)
77             limit=ParameterExpander.evaluateIntegerExpression(aValueMap, limitExpression);
78
79           result = database.executeFreeSql(
80             ParameterExpander.expandExpression( aValueMap, query ),
81             limit);
82           break;
83       }
84     }
85     catch (Throwable t) {
86       aLogger.error("Error while executing free query: " + t.toString());
87     }
88
89     try {
90       ParameterExpander.setValueForKey(
91           aValueMap,
92           ParameterExpander.expandExpression(aValueMap, key),
93           result);
94     }
95     catch (Throwable t) {
96       aLogger.error("Error while setting key " + key + ": " + t.toString());
97     }
98   }
99
100 }