All Downloads are FREE. Search and download functionalities are using the official Maven repository.

cubrid.jdbc.driver.CUBRIDDriver Maven / Gradle / Ivy

The newest version!
/*
 * Copyright (C) 2008 Search Solution Corporation. All rights reserved by Search Solution.
 *
 * Redistribution and use in source and binary forms, with or without modification,
 * are permitted provided that the following conditions are met:
 *
 * - Redistributions of source code must retain the above copyright notice,
 *   this list of conditions and the following disclaimer.
 *
 * - Redistributions in binary form must reproduce the above copyright notice,
 *   this list of conditions and the following disclaimer in the documentation
 *   and/or other materials provided with the distribution.
 *
 * - Neither the name of the  nor the names of its contributors
 *   may be used to endorse or promote products derived from this software without
 *   specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
 * IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT,
 * INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
 * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA,
 * OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
 * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY
 * OF SUCH DAMAGE.
 *
 */

package cubrid.jdbc.driver;

import java.io.File;
import java.io.FileNotFoundException;
import java.io.PrintStream;
import java.sql.Connection;
import java.sql.Driver;
import java.sql.DriverManager;
import java.sql.DriverPropertyInfo;
import java.sql.SQLException;
import java.sql.Timestamp;
import java.text.SimpleDateFormat;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import java.util.Properties;
import java.util.StringTokenizer;
import java.util.concurrent.CopyOnWriteArrayList;
import java.util.logging.Logger;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

import cubrid.jdbc.jci.BrokerHealthCheck;
import cubrid.jdbc.jci.UConnection;
import cubrid.jdbc.jci.UJCIManager;
import cubrid.jdbc.jci.UJCIUtil;

/**
 * Title: CUBRID JDBC Driver Description:
 * 
 * @version 2.0
 */

public class CUBRIDDriver implements Driver {
	// version
	public static final String version_string = "10.1.1.7691";
	public static final int major_version;
	public static final int minor_version;
	public static final int patch_version;
	static {
		StringTokenizer st = new StringTokenizer(version_string, ".");
		if (st.countTokens() != 4) {
			throw new RuntimeException("Could not parse version_string: "
					+ version_string);
		}
		major_version = Integer.parseInt(st.nextToken());
		minor_version = Integer.parseInt(st.nextToken());
		patch_version = Integer.parseInt(st.nextToken());
	}

	// default connection informations
	public static final String default_hostname = "localhost";
	public static final int default_port = 30000;
	public static final String default_user = "public";
	public static final String default_password = "";

	private final static String URL_PATTERN =
	    "jdbc:cubrid(-oracle|-mysql)?:([a-zA-Z_0-9\\.-]*):([0-9]*):([^:]+):([^:]*):([^:]*):(\\?[a-zA-Z_0-9]+=[^&=?]+(&[a-zA-Z_0-9]+=[^&=?]+)*)?";
	private final static String CUBRID_JDBC_URL_HEADER = "jdbc:cubrid";
	private final static String JDBC_DEFAULT_CONNECTION = "jdbc:default:connection";

	static {
		try {
			DriverManager.registerDriver(new CUBRIDDriver());
		} catch (SQLException e) {
		}
	}

	private static PrintStream debugOutput;
	
	static {
		if (UJCIUtil.isConsoleDebug()) {
			try {
				debugOutput = new PrintStream(new File("cubrid.log"));
			} catch (FileNotFoundException e) {
				debugOutput = System.out;
			}
		}
		Thread brokerHealthCheck = new Thread(new BrokerHealthCheck());
		brokerHealthCheck.setDaemon(true);
		brokerHealthCheck.setContextClassLoader(null);
		brokerHealthCheck.start();
	}

	public static void printDebug(String msg) {
		Timestamp timestamp = new Timestamp(System.currentTimeMillis());
		SimpleDateFormat fmt = new SimpleDateFormat("MM-dd hh:mm:ss.SSS");

		String line = String.format("%s %s", fmt.format(timestamp), msg);
		debugOutput.println(line);
	}
	
	/*
	 * java.sql.Driver interface
	 */

	public Connection connect(String url, Properties info) throws SQLException {
	    if (!acceptsURL(url)) {
		return null;
	    }

	    if (url.toLowerCase().startsWith(JDBC_DEFAULT_CONNECTION)) {
		return defaultConnection();
	    }

	    Pattern pattern = Pattern.compile(URL_PATTERN, Pattern.CASE_INSENSITIVE);
	    Matcher matcher = pattern.matcher(url);
	    if (!matcher.find()) {
		throw new CUBRIDException(CUBRIDJDBCErrorCode.invalid_url, url, null);
	    }
	    
	    String match = matcher.group();
	    if (!match.equals(url)) {
	      throw new CUBRIDException(CUBRIDJDBCErrorCode.invalid_url, url, null);
	    }

	    String dummy;
	    String host = matcher.group(2);
	    String portString = matcher.group(3);
	    String db = matcher.group(4);
	    String user;
	    String pass;
	    String prop = matcher.group(7);
	    int port = default_port;

	    UConnection u_con;
	    String resolvedUrl;
	    ConnectionProperties connProperties;

	    if (host == null || host.length() == 0) {
		host = default_hostname;
	    }

	    if (portString == null || portString.length() == 0) {
		port = default_port;
	    } else {
		port = Integer.parseInt(portString);
	    }

	    connProperties = new ConnectionProperties();
	    connProperties.setProperties(prop);

	    user = info.getProperty("user");
	    if (user == null) {
		user = matcher.group(5);
	    }

	    pass = info.getProperty("password");
	    if (pass == null) {
		pass = matcher.group(6);
	    }

	    resolvedUrl = "jdbc:cubrid:" + host + ":" + port + ":" + db + ":" + user + ":********:";
	    if (prop != null) {
		resolvedUrl += prop;
	    }

	    connProperties.setProperties(info);

	    dummy = connProperties.getAltHosts();
	    if (dummy != null) {
		ArrayList altHostList = new ArrayList();
		altHostList.add(host + ":" + port);

		StringTokenizer st = new StringTokenizer(dummy, ",", false);
		while (st.hasMoreTokens()) {
		    altHostList.add(st.nextToken());
		}
		
		if (connProperties.getConnLoadBal()) {
			Collections.shuffle(altHostList);
		}
		try {
		    u_con = UJCIManager.connect(altHostList, db, user, pass, resolvedUrl);
		} catch (CUBRIDException e) {
		    throw e;
		}
	    } else {
		try {
		    u_con = UJCIManager.connect(host, port, db, user, pass, resolvedUrl);
		} catch (CUBRIDException e) {
		    throw e;
		}
	    }

	    u_con.setCharset(connProperties.getCharSet());
	    u_con.setZeroDateTimeBehavior(connProperties.getZeroDateTimeBehavior());
            u_con.setResultWithCUBRIDTypes(connProperties.getResultWithCUBRIDTypes());

	    u_con.setConnectionProperties(connProperties);
	    u_con.tryConnect();
	    return new CUBRIDConnection(u_con, url, user);
	}

	public Connection defaultConnection() throws SQLException {
		if (UJCIUtil.isServerSide()) {
			Thread t = Thread.currentThread();
			Connection c = (Connection) UJCIUtil.invoke(
					"com.cubrid.jsp.ExecuteThread", "getJdbcConnection", null,
					t, null);
			if (c != null) {
				return c;
			}

			UConnection u_con = UJCIManager.connectDefault();
			CUBRIDConnection con = new CUBRIDConnection(u_con,
					"jdbc:default:connection:", "default", true);
			UJCIUtil.invoke("com.cubrid.jsp.ExecuteThread",
					"setJdbcConnection", new Class[] { Connection.class }, t,
					new Object[] { con });
			return con;
		} else {
			return null;
		}
	}

	public boolean acceptsURL(String url) throws SQLException {
	    if (url == null) {
	    	return false;
	    }
	    
	    String urlHeader = CUBRID_JDBC_URL_HEADER;
	    String className = CUBRIDDriver.class.getName();
	    if (className.matches(".*mysql.*")) {
	    	urlHeader += "-mysql:";
	    } else if (className.matches(".*oracle.*")) {
	    	urlHeader += "-oracle:";
	    }
	    else {
	    	urlHeader += ":";
	    }
	    
	    
	    if (url.toLowerCase().startsWith(urlHeader)) {
	    	return true;
	    }
	    
	    if (url.toLowerCase().startsWith(JDBC_DEFAULT_CONNECTION)) {
	    	return true;
	    }

	    return false;
	}

	public DriverPropertyInfo[] getPropertyInfo(String url, Properties info)
			throws SQLException {
		return new DriverPropertyInfo[0];
	}

	public int getMajorVersion() {
		return major_version;
	}

	public int getMinorVersion() {
		return minor_version;
	}

	public boolean jdbcCompliant() {
		return true;
	}

	/* JDK 1.7 */
	public Logger getParentLogger() {
		throw new java.lang.UnsupportedOperationException();
	}
}




© 2015 - 2024 Weber Informatics LLC | Privacy Policy