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