DBConnectionManager.java

Go to the documentation of this file.

package cedar.jetweb.db;

import java.sql.*;
import java.util.Vector;
import java.util.HashMap;

public class DBConnectionManager {

    private HashMap<String, HashMap< String, String> > URLs
    = new HashMap< String, HashMap< String, String> >();
    private HashMap< String, String> defaultUsers_ 
    = new HashMap< String, String>();
    private HashMap<String, String> adminUsers_
    = new HashMap<String, String>();
    private HashMap<String, HashMap<String, Connection> > logConn 
    = new HashMap<String, HashMap<String, Connection> >();

    boolean driverLoaded=false;


    public DBConnectionManager init(String[] args){

    if(!driverLoaded){
        driverLoaded = true;
        String driverName = System.getProperty("driverName");
        System.out.println("The sql driver name is "+driverName);
        
        try{
        Class.forName(driverName).newInstance();
        System.out.println("Loaded " + driverName);
        }catch(Exception E){
        System.err.println("Unable to load driver " + driverName);
        E.printStackTrace();
        }
    }

    for(String arg: args){
        
        //for(int ii=0; ii!= args.length; ++ii){
        //if(!URLs.containsKey(args[ii])){
        if(!URLs.containsKey(arg)){
        String URL = System.getProperty(arg);
        
        Vector<String> users = new Vector<String>();
        String allUsers = System.getProperty(arg+"_users");

        System.out.println("allUsers = "+allUsers);

        String userType = System.getProperty(arg+"_default_user");

        defaultUsers_.put(arg, userType);
        
        userType = System.getProperty(arg + "_admin_user");
        
        adminUsers_.put(arg, userType);
        
        int ii = allUsers.indexOf(" ");
        while(ii>-1){
            System.out.println("ii = "+ii);
            String user = allUsers.substring(0, ii).trim();
            allUsers = allUsers.substring(ii+1).trim();
            ii = allUsers.indexOf(" ");
            if(user.length()>0)users.add(user);
        }

        users.add(allUsers.trim());

        HashMap<String, String> passwords = new HashMap();
                
        HashMap<String, String> userURLs = 
            new HashMap<String, String>();
        URLs.put(arg, userURLs);
        
        String userList = "";

        for(String user: users){
            String password = 
            System.getProperty(user + "_" + arg + "_password");
            if(password==null) password = "";

            userList = userList + " " + user;

            String fullURL = URL + "?user=" + user;
            if(!password.equals(""))
            fullURL = fullURL + "&password="+password;

            userURLs.put(user, fullURL);
            openConnection(arg, user);
        }

        System.out.println("Using database " + 
                   URL + 
                   " with users: " + userList);
        }
    }
        
    return(this);
    };


    public DBConnectionManager init(String arg){
    String[] args = {arg};
    return(init(args));
    };

    public DBConnectionManager init(){
    return(init(DBConfig.JETWEB));
    };


    public synchronized Connection getConnection(String dbName, String user){

    HashMap<String, Connection> connList = logConn.get(dbName);
    Connection con = null;
    if(connList !=null){
        con = connList.get(user);
    }

    boolean conExists = !(con==null);
    if(conExists){
        try{
        conExists = !(con.isClosed());
        }catch(SQLException e){
        System.out.println
            ("Unable to determine state of connection to "+dbName);
        System.out.println("SQLException: "+e.getMessage());
        }
    } 

    if(!conExists){

        System.out.println("dbName = "+dbName);
        System.out.println("user = "+user);

        System.out.println("URLs = "+URLs);

        System.out.println
        ("Reconnecting to database "+URLs.get(dbName).get(user));
        openConnection(dbName, user);
    }
    return(con);
    };

    public synchronized Connection getConnectionAsAdmin(String dbName){
    String userName = adminUsers_.get(dbName);
    return(getConnection(dbName, userName));
    }

    public synchronized Connection getConnection(String dbName){
    String userName = defaultUsers_.get(dbName);

    return(getConnection(dbName, userName));
    }

    public synchronized Connection getConnection(){

    return(getConnection
           (DBConfig.JETWEB, defaultUsers_.get(DBConfig.JETWEB)));
    };

    public synchronized Connection getConnectionAsAdmin(){
    return(getConnection
           (DBConfig.JETWEB, adminUsers_.get(DBConfig.JETWEB)));
    }

    private void openConnection(String dbName, String user){

    try{

        HashMap<String, Connection> existingUsers = 
        logConn.get(dbName);
        if(existingUsers == null){ 
        existingUsers = 
            new HashMap<String, Connection>();
        logConn.put(dbName, existingUsers);
        }

        existingUsers.put
        (user, DriverManager.getConnection
         (URLs.get(dbName).get(user)));
        System.out.println("Connected to "+URLs.get(dbName).get(user));
        //System.out.println(logConn.get(arg).toString());

    }catch(SQLException E) {
        System.out.println("SQLException: " + E.getMessage());
        System.out.println("SQLState:     " + E.getSQLState());
        System.out.println("VendorError:  " + E.getErrorCode());
    }
    return;
    };
}

Generated Wed Jan 17 09:14:27 GMT 2007