mirror of
https://github.com/yacy/yacy_search_server.git
synced 2024-09-21 00:00:13 +02:00
f348190566
git-svn-id: https://svn.berlios.de/svnroot/repos/yacy/trunk@6065 6c8d7289-2bf4-0310-a012-ef5d649a1542
100 lines
3.4 KiB
Java
100 lines
3.4 KiB
Java
// DatabaseConnection.java
|
|
// (C) 2009 by Michael Peter Christen; mc@yacy.net, Frankfurt a. M., Germany
|
|
// first published 11.06.2009 on http://yacy.net
|
|
//
|
|
// $LastChangedDate: 2006-04-02 22:40:07 +0200 (So, 02 Apr 2006) $
|
|
// $LastChangedRevision: 1986 $
|
|
// $LastChangedBy: orbiter $
|
|
//
|
|
// LICENSE
|
|
//
|
|
// This program is free software; you can redistribute it and/or modify
|
|
// it under the terms of the GNU General Public License as published by
|
|
// the Free Software Foundation; either version 2 of the License, or
|
|
// (at your option) any later version.
|
|
//
|
|
// This program is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU General Public License for more details.
|
|
//
|
|
// You should have received a copy of the GNU General Public License
|
|
// along with this program; if not, write to the Free Software
|
|
// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
|
|
|
package de.anomic.content.dao;
|
|
|
|
import java.sql.Connection;
|
|
import java.sql.DriverManager;
|
|
import java.sql.ResultSet;
|
|
import java.sql.SQLException;
|
|
import java.sql.Statement;
|
|
|
|
public class DatabaseConnection {
|
|
|
|
private Connection connection;
|
|
|
|
public DatabaseConnection(final String dbType, String host, int port, String dbname, String user, String pw) throws SQLException {
|
|
String dbDriverStr = null, dbConnStr = null;
|
|
if (dbType.equalsIgnoreCase("mysql")) {
|
|
dbDriverStr = "com.mysql.jdbc.Driver";
|
|
dbConnStr = "jdbc:mysql://" + host + ":" + port + "/" + dbname;
|
|
} else if (dbType.equalsIgnoreCase("pgsql")) {
|
|
dbDriverStr = "org.postgresql.Driver";
|
|
dbConnStr = "jdbc:postgresql://" + host + ":" + port + "/" + dbname;
|
|
} else throw new IllegalArgumentException();
|
|
|
|
try {
|
|
Class.forName(dbDriverStr).newInstance();
|
|
} catch (final Exception e) {
|
|
throw new SQLException("Unable to load the jdbc driver: " + e.getMessage());
|
|
}
|
|
|
|
try {
|
|
this.connection = DriverManager.getConnection(dbConnStr, user, pw);
|
|
} catch (final Exception e) {
|
|
throw new SQLException("Unable to establish a database connection: " + e.getMessage());
|
|
}
|
|
}
|
|
|
|
public void setAutoCommit(boolean b) {
|
|
try {
|
|
this.connection.setAutoCommit(b);
|
|
} catch (SQLException e) {
|
|
e.printStackTrace();
|
|
}
|
|
}
|
|
|
|
public int count(String tablename) throws SQLException {
|
|
Statement stmt = null;
|
|
ResultSet rs = null;
|
|
try {
|
|
stmt = this.connection.createStatement();
|
|
rs = stmt.executeQuery("select count(*) from " + tablename);
|
|
if (rs.next()) {
|
|
return rs.getInt(1);
|
|
}
|
|
return 0;
|
|
} catch (SQLException e) {
|
|
throw e;
|
|
} finally {
|
|
if (rs != null) try {rs.close();} catch (SQLException e) {}
|
|
if (stmt != null) try {stmt.close();} catch (SQLException e) {}
|
|
}
|
|
}
|
|
|
|
public void close() {
|
|
if (connection != null) {
|
|
try {
|
|
connection.close();
|
|
connection = null;
|
|
} catch (SQLException e) {
|
|
}
|
|
}
|
|
}
|
|
|
|
public Statement statement() throws SQLException {
|
|
return this.connection.createStatement();
|
|
}
|
|
}
|