source/mir/producer/FreeQueryProducerNode.java

   1 /*
   2  * Copyright (C) 2001-2006 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  * and distribute linked combinations including the two.  You must obey the
  23  * GNU General Public License in all respects for all of the code used other than
  24  * the above mentioned libraries.  If you modify this file, you may extend this
  25  * exception to your version of the file, but you are not obligated to do so.
  26  * If you do not wish to do so, delete this exception statement from your version.
  27  */
  28 package mir.producer;
  29
  30 import mir.log.LoggerWrapper;
  31 import mir.storage.Database;
  32 import mir.util.ParameterExpander;
  33
  34 import java.util.Map;
  35
  36 public class FreeQueryProducerNode extends AbstractProducerNode {
  37   public static final int QUERY_TYPE_SET = 1;
  38   public static final int QUERY_TYPE_ROW = 2;
  39   public static final int QUERY_TYPE_VALUE = 3;
  40   public static final int QUERY_TYPE_UPDATE = 4;
  41
  42   private Database database = new Database();
  43   private String key;
  44   private String query;
  45   private String limitExpression;
  46   private int type;
  47
  48   public FreeQueryProducerNode(String aKey, String aQuery, String aLimit, int aType) {
  49     key = aKey;
  50     query = aQuery;
  51     limitExpression = aLimit;
  52     type = aType;
  53   }
  54
  55   public void produce(Map aValueMap, String aVerb, LoggerWrapper aLogger) throws ProducerFailure {
  56     Object result = null;
  57
  58     try {
  59       switch (type) {
  60         case QUERY_TYPE_UPDATE:
  61           result = new Integer(database.executeUpdate(ParameterExpander.expandExpression(aValueMap, query)));
  62           break;
  63         case QUERY_TYPE_VALUE:
  64           result = database.executeFreeSingleValueSql(ParameterExpander.expandExpression(aValueMap, query));
  65           break;
  66
  67         case QUERY_TYPE_ROW:
  68           result = database.executeFreeSingleRowSql(ParameterExpander.expandExpression(aValueMap, query));
  69           break;
  70
  71         case QUERY_TYPE_SET:
  72           int limit=10;
  73           if (limitExpression!=null)
  74             limit=ParameterExpander.evaluateIntegerExpression(aValueMap, limitExpression);
  75
  76           result = database.executeFreeSql(
  77             ParameterExpander.expandExpression( aValueMap, query ),
  78             limit);
  79           break;
  80       }
  81     }
  82     catch (Throwable t) {
  83       aLogger.error("Error while executing free query: " + t.toString());
  84     }
  85
  86     try {
  87       ParameterExpander.setValueForKey(
  88           aValueMap,
  89           ParameterExpander.expandExpression(aValueMap, key),
  90           result);
  91     }
  92     catch (Throwable t) {
  93       aLogger.error("Error while setting key " + key + ": " + t.toString());
  94     }
  95   }
  96
  97 }