src/org/cacert/gigi/database/DatabaseConnection.java

   1 package org.cacert.gigi.database;
   2
   3 import java.io.FileInputStream;
   4 import java.io.IOException;
   5 import java.sql.Connection;
   6 import java.sql.DriverManager;
   7 import java.sql.PreparedStatement;
   8 import java.sql.ResultSet;
   9 import java.sql.SQLException;
  10 import java.util.HashMap;
  11 import java.util.Properties;
  12
  13 public class DatabaseConnection {
  14         Connection c;
  15         HashMap<String, PreparedStatement> statements = new HashMap<String, PreparedStatement>();
  16         static Properties credentials = new Properties();
  17         static {
  18                 try {
  19                         credentials.load(new FileInputStream("config/sql.properties"));
  20                 } catch (IOException e) {
  21                         e.printStackTrace();
  22                 }
  23         }
  24         public DatabaseConnection() {
  25                 try {
  26                         Class.forName(credentials.getProperty("driver"));
  27                 } catch (ClassNotFoundException e) {
  28                         e.printStackTrace();
  29                 }
  30                 try {
  31                         c = DriverManager.getConnection(credentials.getProperty("url"),
  32                                         credentials.getProperty("user"),
  33                                         credentials.getProperty("password"));
  34                 } catch (SQLException e) {
  35                         e.printStackTrace();
  36                 }
  37
  38         }
  39         public PreparedStatement prepare(String query) throws SQLException {
  40                 PreparedStatement statement = statements.get(query);
  41                 if (statement == null) {
  42                         statement = c.prepareStatement(query);
  43                         statements.put(query, statement);
  44                 }
  45                 return statement;
  46         }
  47
  48         public static int lastInsertId(PreparedStatement query) throws SQLException {
  49                 ResultSet rs = query.getGeneratedKeys();
  50                 rs.next();
  51                 int id = rs.getInt(1);
  52                 rs.close();
  53                 return id;
  54         }
  55         static ThreadLocal<DatabaseConnection> instances = new ThreadLocal<DatabaseConnection>() {
  56                 @Override
  57                 protected DatabaseConnection initialValue() {
  58                         return new DatabaseConnection();
  59                 }
  60         };
  61         public static DatabaseConnection getInstance() {
  62                 return instances.get();
  63         }
  64 }