Druid/src/main/java/com/alibaba/druid/mock/MockDriver.java

/*
 * Copyright 2011 Alibaba Group.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.alibaba.druid.mock;

import java.sql.Blob;
import java.sql.Clob;
import java.sql.Connection;
import java.sql.Driver;
import java.sql.DriverManager;
import java.sql.DriverPropertyInfo;
import java.sql.NClob;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.SQLXML;
import java.util.Properties;

public class MockDriver implements Driver {
	private String prefix = "jdbc:fake:";
	private String mockPrefix = "jdbc:mock:";

	private final static MockDriver instance = new MockDriver();

	static {
		registerDriver(instance);
	}

	public static boolean registerDriver(Driver driver) {
		try {
			DriverManager.registerDriver(driver);

			return true;
		} catch (Exception e) {
			e.printStackTrace();
		}

		return false;
	}

	@Override
	public Connection connect(String url, Properties info) throws SQLException {
		MockConnection conn = new MockConnection(this);

		if (url == null) {
			return conn;
		}

		if (url.startsWith(prefix)) {
			String catalog = url.substring(prefix.length());
			conn.setCatalog(catalog);

			return conn;
		}
		
		if (url.startsWith(mockPrefix)) {
			String catalog = url.substring(mockPrefix.length());
			conn.setCatalog(catalog);
			
			return conn;
		}

		return null;
	}

	@Override
	public boolean acceptsURL(String url) throws SQLException {
		return url.startsWith(prefix);
	}

	@Override
	public DriverPropertyInfo[] getPropertyInfo(String url, Properties info) throws SQLException {
		return null;
	}

	@Override
	public int getMajorVersion() {
		return 0;
	}

	@Override
	public int getMinorVersion() {
		return 0;
	}

	@Override
	public boolean jdbcCompliant() {
		return true;
	}

	protected ResultSet createResultSet(MockStatement stmt, String sql) {
		MockResultSet rs = new MockResultSet(stmt);

		if ("SELECT 1".equalsIgnoreCase(sql)) {
			rs.getRows().add(new Object[] { 1 });
		} else if ("SELECT NOW()".equalsIgnoreCase(sql)) {
			rs.getRows().add(new Object[] { new java.sql.Timestamp(System.currentTimeMillis()) });
		} else if ("SELECT value FROM _int_1000_".equalsIgnoreCase(sql)) {
			for (int i = 0; i < 1000; ++i) {
				rs.getRows().add(new Object[] { i });
			}
		}

		return rs;
	}

	protected ResultSet createResultSet(MockPreparedStatement stmt) {
		MockResultSet rs = new MockResultSet(stmt);

		String sql = stmt.getSql();

		if ("SELECT 1".equalsIgnoreCase(sql)) {
			rs.getRows().add(new Object[] { 1 });
		} else if ("SELECT NOW()".equalsIgnoreCase(sql)) {
			rs.getRows().add(new Object[] { new java.sql.Timestamp(System.currentTimeMillis()) });
		} else if ("SELECT ?".equalsIgnoreCase(sql)) {
			rs.getRows().add(new Object[] { stmt.getParameters().get(0) });
		}

		return rs;
	}

	protected Clob createClob(MockConnection conn) throws SQLException {
		return new MockClob();
	}

	protected Blob createBlob(MockConnection conn) throws SQLException {
		return new MockBlob();
	}

	protected NClob createNClob(MockConnection conn) throws SQLException {
		return new MockNClob();
	}

	protected SQLXML createSQLXML(MockConnection conn) throws SQLException {
		return new MockSQLXML();
	}
}