01: /*
02: * JBoss, Home of Professional Open Source.
03: * Copyright 2006, Red Hat Middleware LLC, and individual contributors
04: * as indicated by the @author tags. See the copyright.txt file in the
05: * distribution for a full listing of individual contributors.
06: *
07: * This is free software; you can redistribute it and/or modify it
08: * under the terms of the GNU Lesser General Public License as
09: * published by the Free Software Foundation; either version 2.1 of
10: * the License, or (at your option) any later version.
11: *
12: * This software is distributed in the hope that it will be useful,
13: * but WITHOUT ANY WARRANTY; without even the implied warranty of
14: * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15: * Lesser General Public License for more details.
16: *
17: * You should have received a copy of the GNU Lesser General Public
18: * License along with this software; if not, write to the Free
19: * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
20: * 02110-1301 USA, or see the FSF site: http://www.fsf.org.
21: */
22: package org.jboss.test.cmp2.dbschema.util;
23:
24: import java.sql.ResultSet;
25: import java.sql.SQLException;
26: import java.sql.DatabaseMetaData;
27: import java.sql.Connection;
28: import java.sql.DriverManager;
29: import java.util.List;
30: import java.util.ArrayList;
31: import java.util.Map;
32: import java.util.HashMap;
33:
34: /**
35: *
36: * @author <a href="mailto:alex@jboss.org">Alex Loubyansky</a>
37: */
38: public class DBSchemaHelper {
39: private static final String TABLE_NAME = "TABLE_NAME";
40:
41: public static Table getTable(DatabaseMetaData dbMD, String tableName)
42: throws SQLException {
43: ResultSet rs = dbMD.getColumns(null, null, tableName, null);
44: Map columns = new HashMap();
45: while (rs.next()) {
46: final Column column = new Column(rs);
47: columns.put(column.getName(), column);
48: }
49: safeClose(rs);
50: return new Table(tableName, columns);
51: }
52:
53: public static List getTableNames(DatabaseMetaData dbMD)
54: throws SQLException {
55: ResultSet rs = dbMD.getTables(null, null, null, null);
56: List results = new ArrayList();
57: while (rs.next()) {
58: results.add(rs.getString(TABLE_NAME));
59: }
60: safeClose(rs);
61: return results;
62: }
63:
64: public static Connection getConnection(String url, String user,
65: String pwd) throws SQLException {
66: return DriverManager.getConnection(url, user, pwd);
67: }
68:
69: public static void safeClose(Connection con) {
70: if (con != null)
71: try {
72: con.close();
73: } catch (Exception e) {
74: }
75: }
76:
77: public static void safeClose(ResultSet rs) {
78: if (rs != null)
79: try {
80: rs.close();
81: } catch (Exception e) {
82: }
83: }
84: }
|