9b151d1a904a5d15493b5f1d25efd8783ee0314e
[mir.git] / source / mir / producer / EntityListProducerNode.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 package mir.producer;
31
32 import java.util.List;
33 import java.util.Map;
34 import java.util.Vector;
35
36 import mir.entity.adapter.EntityAdapterModel;
37 import mir.entity.adapter.EntityIteratorAdapter;
38 import mir.log.LoggerWrapper;
39 import mir.util.CachingRewindableIterator;
40 import mir.util.ParameterExpander;
41
42 public class EntityListProducerNode extends ProducerNodeDecorator {
43   private String key;
44   private String whereClause;
45   private String orderByClause;
46   private String mainTablePrefix;
47   private List extraTables;
48   private int batchSize;
49   private EntityAdapterModel model;
50   private String definition;
51   private String limitExpression;
52   private String skipExpression;
53
54   public EntityListProducerNode(String aKey,
55       EntityAdapterModel aModel, String aMainTablePrefix, List someExtraTables,
56       String aDefinition, String aWhereClause, String anOrderByClause,
57       String aLimitExpression, String aSkipExpression, ProducerNode aSubNode) {
58     super(aSubNode);
59
60     model = aModel;
61     mainTablePrefix = aMainTablePrefix;
62     extraTables = someExtraTables;
63     definition = aDefinition;
64     key = aKey;
65     whereClause = aWhereClause;
66     orderByClause = anOrderByClause;
67     limitExpression = aLimitExpression;
68     skipExpression = aSkipExpression;
69   }
70
71   public EntityListProducerNode(String aKey,
72       EntityAdapterModel aModel, String aMainTablePrefix, List someExtraTables,
73       String aDefinition, String aWhereClause, String anOrderByClause,
74       int aLimit, int aSkip, ProducerNode aSubNode) {
75     this(aKey,  aModel, aMainTablePrefix, someExtraTables,
76           aDefinition, aWhereClause, anOrderByClause,
77           Integer.toString(aLimit), Integer.toString(aSkip), aSubNode);
78   }
79
80   public void produce(Map aValueMap, String aVerb, LoggerWrapper aLogger) throws ProducerFailure, ProducerExc {
81     try {
82       int limit = ParameterExpander.evaluateIntegerExpressionWithDefault(aValueMap, limitExpression, -1);
83       int skip = ParameterExpander.evaluateIntegerExpressionWithDefault(aValueMap, skipExpression, 0);
84
85       if (skipExpression != null && !skipExpression.trim().equals(""))
86         skip = ParameterExpander.evaluateIntegerExpression(aValueMap, skipExpression);
87
88       ParameterExpander.setValueForKey(
89         aValueMap,
90         key,
91         new CachingRewindableIterator(
92           new EntityIteratorAdapter( mainTablePrefix, extraTables,
93             ParameterExpander.expandExpression( aValueMap, whereClause ),
94             ParameterExpander.expandExpression( aValueMap, orderByClause ),
95             Math.min(50, limit),
96             model,
97             definition,
98             limit,
99             skip )
100         )
101       );
102     }
103     catch (Throwable t) {
104       aLogger.error("cannot retrieve list into key " + key + ": " + t.getMessage());
105       try {
106         ParameterExpander.setValueForKey(
107           aValueMap,
108           key,
109           new CachingRewindableIterator(new Vector().iterator())
110         );
111       }
112       catch (Throwable s) {
113       }
114     }
115
116     super.produce(aValueMap, aVerb, aLogger);
117   };
118
119 }