source/mir/producer/ExternalDbProducerNode.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.sql.Connection;
  33 import java.sql.DriverManager;
  34 import java.sql.ResultSet;
  35 import java.sql.ResultSetMetaData;
  36 import java.sql.Statement;
  37 import java.util.ArrayList;
  38 import java.util.HashMap;
  39 import java.util.Iterator;
  40 import java.util.Map;
  41
  42 import mir.log.LoggerWrapper;
  43 import mir.util.ExceptionFunctions;
  44 import mir.util.ParameterExpander;
  45
  46 public class ExternalDbProducerNode extends ProducerNodeDecorator {
  47   private String key;
  48   private String driver;
  49   private String host;
  50   private String port;
  51   private String database;
  52   private String username;
  53   private String password;
  54   private String query;
  55
  56
  57   public ExternalDbProducerNode(String aKey, String aDriver, String aHost, String aPort, String aDatabase, String aUsername, String aPassword,String aQuery,ProducerNode aSubNode) {
  58     super(aSubNode);
  59     key = aKey;
  60     driver = aDriver;
  61     host = aHost;
  62     port = aPort;
  63     database = aDatabase;
  64     username = aUsername;
  65     password = aPassword;
  66     query =aQuery;
  67
  68   }
  69
  70   public void produce(Map aValueMap, String aVerb, LoggerWrapper aLogger) throws ProducerFailure {
  71     try {
  72       String expandedKey = ParameterExpander.expandExpression( aValueMap, key );
  73       String expandedDriver = ParameterExpander.expandExpression( aValueMap, driver);
  74       String expandedHost = ParameterExpander.expandExpression( aValueMap, host);
  75       String expandedPort = ParameterExpander.expandExpression( aValueMap, port);
  76       String expandedDatabase = ParameterExpander.expandExpression( aValueMap, database);
  77       String expandedUsername = ParameterExpander.expandExpression( aValueMap, username);
  78       String expandedPassword = ParameterExpander.expandExpression( aValueMap, password);
  79       String expandedQuery = ParameterExpander.expandExpression( aValueMap, query);
  80
  81       if (expandedDriver.equals("postgresql")){
  82         Class.forName("org.postgresql.Driver");
  83       }
  84       if (expandedDriver.equals("mysql")){
  85         Class.forName("com.mysql.jdbc.Driver");
  86       }
  87       else {
  88           throw new Exception("Unsupported DB Driver:"+expandedDriver);
  89       }
  90
  91       Connection db = DriverManager.getConnection("jdbc:"+expandedDriver+"://"+expandedHost
  92                                                   +":"+expandedPort+"/"+expandedDatabase
  93                                                   , expandedUsername, expandedPassword);
  94
  95       Statement st = db.createStatement();
  96       ResultSet rs = st.executeQuery(expandedQuery);
  97       ResultSetMetaData rsmd = rs.getMetaData();
  98       int numberOfColumns= rsmd.getColumnCount();
  99       ArrayList fieldNames = new ArrayList(numberOfColumns);
 100       for (int i=0;i<numberOfColumns;i++){
 101           fieldNames.add(rsmd.getColumnName(i+1));
 102       }
 103
 104       while(rs.next()  && !isAborted(aValueMap)) {
 105         HashMap result=new HashMap();
 106         Iterator fields = fieldNames.iterator();
 107         while (fields.hasNext()) {
 108           String field=(String) fields.next();
 109           result.put(field,rs.getString(field));
 110         }
 111         ParameterExpander.setValueForKey(aValueMap,expandedKey,result);
 112         super.produce(aValueMap, aVerb, aLogger);
 113       }
 114       rs.close();
 115       st.close();
 116       db.close();
 117     }
 118     catch (Throwable t) {
 119       Throwable s = ExceptionFunctions.traceCauseException(t);
 120       aLogger.error("Error while accessing external database: " + s.getClass().getName()+","+ s.getMessage());
 121     }
 122   }
 123 }