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                                         + "?zeroDateTimeBehavior=convertToNull",
  33                                         credentials.getProperty("user"),
  34                                         credentials.getProperty("password"));
  35                 } catch (SQLException e) {
  36                         e.printStackTrace();
  37                 }
  38
  39         }
  40         public PreparedStatement prepare(String query) throws SQLException {
  41                 PreparedStatement statement = statements.get(query);
  42                 if (statement == null) {
  43                         statement = c.prepareStatement(query);
  44                         statements.put(query, statement);
  45                 }
  46                 return statement;
  47         }
  48
  49         public static int lastInsertId(PreparedStatement query) throws SQLException {
  50                 ResultSet rs = query.getGeneratedKeys();
  51                 rs.next();
  52                 int id = rs.getInt(1);
  53                 rs.close();
  54                 return id;
  55         }
  56         static ThreadLocal<DatabaseConnection> instances = new ThreadLocal<DatabaseConnection>() {
  57                 @Override
  58                 protected DatabaseConnection initialValue() {
  59                         return new DatabaseConnection();
  60                 }
  61         };
  62         public static DatabaseConnection getInstance() {
  63                 return instances.get();
  64         }
  65 }