2 * Copyright (C) 2001-2006 The Mir-coders group
4 * This file is part of Mir.
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.
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.
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
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.
30 import mir.util.ParameterExpander;
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;
43 public class ExternalDbProducerNode extends ProducerNodeDecorator {
45 private String driver;
48 private String database;
49 private String username;
50 private String password;
53 public ExternalDbProducerNode(String aKey, String aDriver, String aHost, String aPort, String aDatabase, String aUsername, String aPassword, String aQuery, ProducerNode aSubNode) {
66 public void produce(ProductionContext aProductionContext) throws ProducerFailure {
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);
77 if ("postgresql".equals(expandedDriver)) {
78 Class.forName("org.postgresql.Driver");
80 if ("mysql".equals(expandedDriver)) {
81 Class.forName("com.mysql.jdbc.Driver");
84 throw new Exception("Unsupported DB Driver:" + expandedDriver);
87 Connection db = DriverManager.getConnection("jdbc:" + expandedDriver + "://" + expandedHost
88 + ":" + expandedPort + "/" + expandedDatabase
89 , expandedUsername, expandedPassword);
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));
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));
107 ParameterExpander.setValueForKey(aProductionContext.getValueSet(), expandedKey, result);
108 super.produce(aProductionContext);
114 catch (Throwable t) {
115 aProductionContext.getLogger().warn("Error while accessing external database", t);