DBConnectionManager.java
上传用户:yuyunping
上传日期:2013-03-21
资源大小:1844k
文件大小:9k
源码类别:

Java书籍

开发平台:

Java

  1. package net.acai.database;
  2. import java.io.*;
  3. import java.sql.*;
  4. import java.util.*;
  5. import java.util.Date;
  6. /**
  7.  * 管理类DBConnectionManager支持对一个或多个由属性文件定义的数据库连接
  8.  * 池的访问.客户程序可以调用getInstance()方法访问本类的唯一实例.
  9.  * Title:        清清网络
  10.  * Description:
  11.  * Copyright:    Copyright (c) 2002
  12.  * Company:      www.SuperSpace.com
  13.  * @author:       SuperSpace
  14.  * @version 1.0
  15.  */
  16. public class DBConnectionManager {
  17.  static private DBConnectionManager instance; // 唯一实例
  18.  static private int clients;
  19.  private Vector drivers = new Vector();
  20.  private PrintWriter log;
  21.  private Hashtable pools = new Hashtable();
  22.  /**
  23.   * 返回唯一实例.如果是第一次调用此方法,则创建实例
  24.   *
  25.   * @return DBConnectionManager 唯一实例
  26.   */
  27.  static synchronized public DBConnectionManager getInstance() {
  28.    if (instance == null) {
  29.      instance = new DBConnectionManager();
  30.    }
  31.    clients++;
  32.    return instance;
  33.  }
  34.  /**
  35.   * 建构函数私有以防止其它对象创建本类实例
  36.   */
  37.  private DBConnectionManager() {
  38.    init();
  39.  }
  40.  /**
  41.   * 将连接对象返回给由名字指定的连接池
  42.   *
  43.   * @param name 在属性文件中定义的连接池名字
  44.   * @param con 连接对象
  45.   */
  46.  public void freeConnection(String name, Connection con) {
  47.    DBConnectionPool pool = (DBConnectionPool) pools.get(name);
  48.    if (pool != null) {
  49.      pool.freeConnection(con);
  50.    }
  51.  }
  52.  /**
  53.   * 获得一个可用的(空闲的)连接.如果没有可用连接,且已有连接数小于最大连接数
  54.   * 限制,则创建并返回新连接
  55.   *
  56.   * @param name 在属性文件中定义的连接池名字
  57.   * @return Connection 可用连接或null
  58.   */
  59.  public Connection getConnection(String name) {
  60.    DBConnectionPool pool = (DBConnectionPool) pools.get(name);
  61.    if (pool != null) {
  62.      return pool.getConnection();
  63.    }
  64.    return null;
  65.  }
  66.  /**
  67.   * 获得一个可用连接.若没有可用连接,且已有连接数小于最大连接数限制,
  68.   * 则创建并返回新连接.否则,在指定的时间内等待其它线程释放连接.
  69.   *
  70.   * @param name 连接池名字
  71.   * @param time 以毫秒计的等待时间
  72.   * @return Connection 可用连接或null
  73.   */
  74.  public Connection getConnection(String name, long time) {
  75.    DBConnectionPool pool = (DBConnectionPool) pools.get(name);
  76.    if (pool != null) {
  77.      return pool.getConnection(time);
  78.    }
  79.    return null;
  80.  }
  81.  /**
  82.   * 关闭所有连接,撤销驱动程序的注册
  83.   */
  84.  public synchronized void release() {
  85.    // 等待直到最后一个客户程序调用
  86.    if (--clients != 0) {
  87.      return;
  88.    }
  89.    Enumeration allPools = pools.elements();
  90.    while (allPools.hasMoreElements()) {
  91.      DBConnectionPool pool = (DBConnectionPool) allPools.nextElement();
  92.      pool.release();
  93.    }
  94.    Enumeration allDrivers = drivers.elements();
  95.    while (allDrivers.hasMoreElements()) {
  96.      Driver driver = (Driver) allDrivers.nextElement();
  97.      try {
  98.        DriverManager.deregisterDriver(driver);
  99.        log("撤销JDBC驱动程序 " + driver.getClass().getName()+"的注册");
  100.      }
  101.      catch (SQLException e) {
  102.        log(e, "无法撤销下列JDBC驱动程序的注册: " + driver.getClass().getName());
  103.      }
  104.    }
  105.  }
  106.  /**
  107.   * 根据指定属性创建连接池实例.
  108.   *
  109.   * @param props 连接池属性
  110.   */
  111.  private void createPools(Properties props) {
  112.    Enumeration propNames = props.propertyNames();
  113.    while (propNames.hasMoreElements()) {
  114.    String name = (String) propNames.nextElement();
  115.    if (name.endsWith(".url")) {
  116.      String poolName = name.substring(0, name.lastIndexOf("."));
  117.      String url = props.getProperty(poolName + ".url");
  118.      if (url == null) {
  119.        log("没有为连接池" + poolName + "指定URL");
  120.        continue;
  121.      }
  122.    String user = props.getProperty(poolName + ".user");
  123.    String password = props.getProperty(poolName + ".password");
  124.    String maxconn = props.getProperty(poolName + ".maxconn", "0");
  125.    int max;
  126.    try {
  127.    max = Integer.valueOf(maxconn).intValue();
  128.    }
  129.    catch (NumberFormatException e) {
  130.      log("错误的最大连接数限制: " + maxconn + " .连接池: " + poolName);
  131.      max = 0;
  132.    }
  133.    DBConnectionPool pool =
  134.    new DBConnectionPool(poolName, url, user, password, max);
  135.    pools.put(poolName, pool);
  136.    log("成功创建连接池" + poolName);
  137.   }
  138.  }
  139.  }
  140.  /**
  141.   * 读取属性完成初始化
  142.   */
  143.   private void init() {
  144.   InputStream is = getClass().getResourceAsStream("/db.properties");
  145.   Properties dbProps = new Properties();
  146.   try {
  147.     dbProps.load(is);
  148.   }
  149.   catch (Exception e) {
  150.     System.err.println("不能读取属性文件. " +
  151.    "请确保db.properties在CLASSPATH指定的路径中");
  152.     return;
  153.   }
  154.   String logFile = dbProps.getProperty("logfile", "DBConnectionManager.log");
  155.   try {
  156.     log = new PrintWriter(new FileWriter(logFile, true), true);
  157.   }
  158.   catch (IOException e) {
  159.     System.err.println("无法打开日志文件: " + logFile);
  160.     log = new PrintWriter(System.err);
  161.   }
  162.   loadDrivers(dbProps);
  163.   createPools(dbProps);
  164.  }
  165.  /**
  166.   * 装载和注册所有JDBC驱动程序
  167.   *
  168.   * @param props 属性
  169.   */
  170.  private void loadDrivers(Properties props) {
  171.    String driverClasses = props.getProperty("drivers");
  172.    StringTokenizer st = new StringTokenizer(driverClasses);
  173.    while (st.hasMoreElements()) {
  174.    String driverClassName = st.nextToken().trim();
  175.    try {
  176.      Driver driver = (Driver)
  177.      Class.forName(driverClassName).newInstance();
  178.      DriverManager.registerDriver(driver);
  179.      drivers.addElement(driver);
  180.      log("成功注册JDBC驱动程序" + driverClassName);
  181.    }
  182.    catch (Exception e) {
  183.      log("无法注册JDBC驱动程序: " +
  184.      driverClassName + ", 错误: " + e);
  185.    }
  186.  }
  187.  }
  188.  /**
  189.   * 将文本信息写入日志文件
  190.   */
  191.  private void log(String msg) {
  192.    log.println(new Date() + ": " + msg);
  193.  }
  194.  /**
  195.   * 将文本信息与异常写入日志文件
  196.   */
  197.  private void log(Throwable e, String msg) {
  198.    log.println(new Date() + ": " + msg);
  199.    e.printStackTrace(log);
  200.  }
  201.  /**
  202.   * 此内部类定义了一个连接池.它能够根据要求创建新连接,直到预定的最
  203.   * 大连接数为止.在返回连接给客户程序之前,它能够验证连接的有效性.
  204.   */
  205.  class DBConnectionPool {
  206.    private int checkedOut;
  207.    private Vector freeConnections = new Vector();
  208.    private int maxConn;
  209.    private String name;
  210.    private String password;
  211.    private String URL;
  212.    private String user;
  213.  /**
  214.   * 创建新的连接池
  215.   *
  216.   * @param name 连接池名字
  217.   * @param URL 数据库的JDBC URL
  218.   * @param user 数据库帐号,或 null
  219.   * @param password 密码,或 null
  220.   * @param maxConn 此连接池允许建立的最大连接数
  221.   */
  222.  public DBConnectionPool(String name, String URL, String user, String password,
  223.    int maxConn) {
  224.    this.name = name;
  225.    this.URL = URL;
  226.    this.user = user;
  227.    this.password = password;
  228.    this.maxConn = maxConn;
  229.    }
  230.   /**
  231.    * 将不再使用的连接返回给连接池
  232.    *
  233.    * @param con 客户程序释放的连接
  234.    */
  235.  public synchronized void freeConnection(Connection con) {
  236.    // 将指定连接加入到向量末尾
  237.    freeConnections.addElement(con);
  238.    checkedOut--;
  239.    notifyAll();
  240.  }
  241.  /**
  242.   * 从连接池获得一个可用连接.如没有空闲的连接且当前连接数小于最大连接
  243.   * 数限制,则创建新连接.如原来登记为可用的连接不再有效,则从向量删除之,
  244.   * 然后递归调用自己以尝试新的可用连接.
  245.   */
  246.  public synchronized Connection getConnection() {
  247.    Connection con = null;
  248.    if (freeConnections.size() > 0) {
  249.      // 获取向量中第一个可用连接
  250.      con = (Connection) freeConnections.firstElement();
  251.      freeConnections.removeElementAt(0);
  252.      try {
  253.        if (con.isClosed()) {
  254.          log("从连接池" + name+"删除一个无效连接");
  255.          // 递归调用自己,尝试再次获取可用连接
  256.          con = getConnection();
  257.        }
  258.      }
  259.      catch (SQLException e) {
  260.        log("从连接池" + name+"删除一个无效连接");
  261.        // 递归调用自己,尝试再次获取可用连接
  262.        con = getConnection();
  263.      }
  264.     }
  265.     else if (maxConn == 0 || checkedOut < maxConn) {
  266. con = newConnection();
  267. }
  268. if (con != null) {
  269. checkedOut++;
  270. }
  271. return con;
  272. }
  273. /**
  274. * 从连接池获取可用连接.可以指定客户程序能够等待的最长时间
  275. * 参见前一个getConnection()方法.
  276. *
  277. * @param timeout 以毫秒计的等待时间限制
  278. */
  279. public synchronized Connection getConnection(long timeout) {
  280. long startTime = new Date().getTime();
  281. Connection con;
  282. while ((con = getConnection()) == null) {
  283. try {
  284. wait(timeout);
  285. }
  286. catch (InterruptedException e) {}
  287. if ((new Date().getTime() - startTime) >= timeout) {
  288.         // wait()返回的原因是超时
  289.         return null;
  290.       }
  291.     }
  292.     return con;
  293.   }
  294.  /**
  295.   * 关闭所有连接
  296.   */
  297.  public synchronized void release() {
  298.    Enumeration allConnections = freeConnections.elements();
  299.    while (allConnections.hasMoreElements()) {
  300.      Connection con = (Connection) allConnections.nextElement();
  301.      try {
  302.        con.close();
  303.        log("关闭连接池" + name+"中的一个连接");
  304.      }
  305.      catch (SQLException e) {
  306.        log(e, "无法关闭连接池" + name+"中的连接");
  307.      }
  308.    }
  309.    freeConnections.removeAllElements();
  310.  }
  311.  /**
  312.   * 创建新的连接
  313.   */
  314.  private Connection newConnection() {
  315.    Connection con = null;
  316.    try {
  317.      if (user == null) {
  318.        con = DriverManager.getConnection(URL);
  319.      }
  320.      else {
  321.        con = DriverManager.getConnection(URL, user, password);
  322.      }
  323.      log("连接池" + name+"创建一个新的连接");
  324.    }
  325.    catch (SQLException e) {
  326.      log(e, "无法创建下列URL的连接: " + URL);
  327.      return null;
  328.    }
  329.    return con;
  330.  }
  331.  }
  332.  }