xxq250
/
homework-jianmu
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

adapt the calcite for customer GLD

This commit is contained in:
zyyang 2020-09-14 17:48:27 +08:00
parent 21584e3069
commit 99741286ce
5 changed files with 510 additions and 417 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

227
src/connector/jdbc/pom.xml
View File

@ -1,106 +1,129 @@
<project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" <project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd"> xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
<modelVersion>4.0.0</modelVersion> <modelVersion>4.0.0</modelVersion>
<groupId>com.taosdata.jdbc</groupId> <groupId>com.taosdata.jdbc</groupId>
<artifactId>taos-jdbcdriver</artifactId> <artifactId>taos-jdbcdriver</artifactId>
<version>2.0.0</version> <version>2.0.0</version>
<packaging>jar</packaging> <packaging>jar</packaging>
<name>JDBCDriver</name> <name>JDBCDriver</name>
<url>https://github.com/taosdata/TDengine/tree/master/src/connector/jdbc</url>
<description>TDengine JDBC Driver</description>
<licenses>
<license>
<name>GNU AFFERO GENERAL PUBLIC LICENSE Version 3</name>
<url>https://github.com/taosdata/TDengine/blob/master/LICENSE</url>
<distribution>repo</distribution>
</license>
</licenses>
<scm>
<connection>scm:git:git://github.com/taosdata/TDengine.git</connection>
<developerConnection>scm:git:git@github.com:taosdata/TDengine.git</developerConnection>
<url>https://github.com/taosdata/TDengine/tree/master/src/connector/jdbc</url> <url>https://github.com/taosdata/TDengine/tree/master/src/connector/jdbc</url>
<tag>HEAD</tag> <description>TDengine JDBC Driver</description>
</scm> <licenses>
<developers> <license>
<developer> <name>GNU AFFERO GENERAL PUBLIC LICENSE Version 3</name>
<name>taosdata</name> <url>https://github.com/taosdata/TDengine/blob/master/LICENSE</url>
<email>support@taosdata.com</email> <distribution>repo</distribution>
<organization>https://www.taosdata.com/</organization> </license>
<organizationUrl>https://www.taosdata.com/</organizationUrl> </licenses>
</developer> <scm>
</developers> <connection>scm:git:git://github.com/taosdata/TDengine.git</connection>
<properties> <developerConnection>scm:git:git@github.com:taosdata/TDengine.git</developerConnection>
<project.build.sourceEncoding>UTF-8</project.build.sourceEncoding> <url>https://github.com/taosdata/TDengine/tree/master/src/connector/jdbc</url>
<java.version>1.8</java.version> <tag>HEAD</tag>
<maven-compiler-plugin.version>3.6.0</maven-compiler-plugin.version> </scm>
<commons-logging.version>1.1.2</commons-logging.version> <developers>
<commons-lang3.version>3.5</commons-lang3.version> <developer>
</properties> <name>taosdata</name>
<dependencies> <email>support@taosdata.com</email>
<dependency> <organization>https://www.taosdata.com/</organization>
<groupId>commons-logging</groupId> <organizationUrl>https://www.taosdata.com/</organizationUrl>
<artifactId>commons-logging</artifactId> </developer>
<version>${commons-logging.version}</version> </developers>
<exclusions> <properties>
<exclusion> <project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
<groupId>*</groupId> <java.version>1.8</java.version>
<artifactId>*</artifactId> <maven-compiler-plugin.version>3.6.0</maven-compiler-plugin.version>
</exclusion> <commons-logging.version>1.1.2</commons-logging.version>
</exclusions> <commons-lang3.version>3.5</commons-lang3.version>
</dependency> </properties>
<dependency> <dependencies>
<groupId>org.apache.commons</groupId> <dependency>
<artifactId>commons-lang3</artifactId> <groupId>commons-logging</groupId>
<version>${commons-lang3.version}</version> <artifactId>commons-logging</artifactId>
</dependency> <version>${commons-logging.version}</version>
<dependency> <exclusions>
<groupId>junit</groupId> <exclusion>
<artifactId>junit</artifactId> <groupId>*</groupId>
<version>4.13</version> <artifactId>*</artifactId>
<scope>test</scope> </exclusion>
</dependency> </exclusions>
</dependencies> </dependency>
<build> <dependency>
<plugins> <groupId>org.apache.commons</groupId>
<plugin> <artifactId>commons-lang3</artifactId>
<groupId>org.apache.maven.plugins</groupId> <version>${commons-lang3.version}</version>
<artifactId>maven-assembly-plugin</artifactId> </dependency>
<version>3.0.0</version> <dependency>
<configuration> <groupId>junit</groupId>
<descriptors> <artifactId>junit</artifactId>
<descriptor>src/main/assembly/assembly-jar.xml</descriptor> <version>4.13</version>
</descriptors> <scope>test</scope>
</configuration> </dependency>
<executions>
<execution> <!-- calcite -->
<id>make-assembly</id> <dependency>
<phase>package</phase> <groupId>org.apache.calcite</groupId>
<goals> <artifactId>calcite-core</artifactId>
<goal>single</goal> <version>1.23.0</version>
</goals> </dependency>
</execution> <dependency>
</executions> <groupId>org.apache.commons</groupId>
</plugin> <artifactId>commons-dbcp2</artifactId>
<plugin> <version>2.7.0</version>
<groupId>org.apache.maven.plugins</groupId> </dependency>
<artifactId>maven-compiler-plugin</artifactId> <dependency>
<version>${maven-compiler-plugin.version}</version> <groupId>org.apache.calcite.avatica</groupId>
<configuration> <artifactId>avatica-core</artifactId>
<encoding>UTF-8</encoding> <version>1.17.0</version>
<source>${java.version}</source> </dependency>
<target>${java.version}</target> <dependency>
<debug>true</debug> <groupId>mysql</groupId>
<showDeprecation>true</showDeprecation> <artifactId>mysql-connector-java</artifactId>
</configuration> <version>5.1.47</version>
</plugin> </dependency>
<plugin>
<groupId>org.apache.maven.plugins</groupId> </dependencies>
<artifactId>maven-surefire-plugin</artifactId> <build>
<version>2.12.4</version> <plugins>
<configuration> <plugin>
<testFailureIgnore>true</testFailureIgnore> <groupId>org.apache.maven.plugins</groupId>
</configuration> <artifactId>maven-assembly-plugin</artifactId>
</plugin> <version>3.0.0</version>
</plugins> <configuration>
</build> <descriptors>
<descriptor>src/main/assembly/assembly-jar.xml</descriptor>
</descriptors>
</configuration>
<executions>
<execution>
<id>make-assembly</id>
<phase>package</phase>
<goals>
<goal>single</goal>
</goals>
</execution>
</executions>
</plugin>
<plugin>
<groupId>org.apache.maven.plugins</groupId>
<artifactId>maven-compiler-plugin</artifactId>
<version>${maven-compiler-plugin.version}</version>
<configuration>
<encoding>UTF-8</encoding>
<source>${java.version}</source>
<target>${java.version}</target>
<debug>true</debug>
<showDeprecation>true</showDeprecation>
</configuration>
</plugin>
<plugin>
<groupId>org.apache.maven.plugins</groupId>
<artifactId>maven-surefire-plugin</artifactId>
<version>2.12.4</version>
<configuration>
<testFailureIgnore>true</testFailureIgnore>
</configuration>
</plugin>
</plugins>
</build>
</project> </project>

54
src/connector/jdbc/src/main/java/com/taosdata/jdbc/TSDBConnection.java
View File

@ -53,66 +53,12 @@ public class TSDBConnection implements Connection {
public TSDBConnection(Properties info, TSDBDatabaseMetaData meta) throws SQLException { public TSDBConnection(Properties info, TSDBDatabaseMetaData meta) throws SQLException {
this.dbMetaData = meta; this.dbMetaData = meta;
//load taos.cfg start
File cfgDir = loadConfigDir(info.getProperty(TSDBDriver.PROPERTY_KEY_CONFIG_DIR));
File cfgFile = cfgDir.listFiles((dir, name) -> "taos.cfg".equalsIgnoreCase(name))[0];
List<String> endpoints = loadConfigEndpoints(cfgFile);
if (!endpoints.isEmpty()) {
info.setProperty(TSDBDriver.PROPERTY_KEY_HOST, endpoints.get(0).split(":")[0]);
info.setProperty(TSDBDriver.PROPERTY_KEY_PORT, endpoints.get(0).split(":")[1]);
}
//load taos.cfg end
connect(info.getProperty(TSDBDriver.PROPERTY_KEY_HOST), connect(info.getProperty(TSDBDriver.PROPERTY_KEY_HOST),
Integer.parseInt(info.getProperty(TSDBDriver.PROPERTY_KEY_PORT, "0")), Integer.parseInt(info.getProperty(TSDBDriver.PROPERTY_KEY_PORT, "0")),
info.getProperty(TSDBDriver.PROPERTY_KEY_DBNAME), info.getProperty(TSDBDriver.PROPERTY_KEY_USER), info.getProperty(TSDBDriver.PROPERTY_KEY_DBNAME), info.getProperty(TSDBDriver.PROPERTY_KEY_USER),
info.getProperty(TSDBDriver.PROPERTY_KEY_PASSWORD)); info.getProperty(TSDBDriver.PROPERTY_KEY_PASSWORD));
} }
private List<String> loadConfigEndpoints(File cfgFile) {
List<String> endpoints = new ArrayList<>();
try (BufferedReader reader = new BufferedReader(new FileReader(cfgFile))) {
String line = null;
while ((line = reader.readLine()) != null) {
if (line.trim().startsWith("firstEp") || line.trim().startsWith("secondEp")) {
endpoints.add(line.substring(line.indexOf('p') + 1).trim());
}
if (endpoints.size() > 1)
break;
}
} catch (FileNotFoundException e) {
e.printStackTrace();
} catch (IOException e) {
e.printStackTrace();
}
return endpoints;
}
/**
* @param cfgDirPath
* @return return the config dir
**/
private File loadConfigDir(String cfgDirPath) {
if (cfgDirPath == null)
return loadDefaultConfigDir();
File cfgDir = new File(cfgDirPath);
if (!cfgDir.exists())
return loadDefaultConfigDir();
return cfgDir;
}
/**
* @return search the default config dir, if the config dir is not exist will return null
*/
private File loadDefaultConfigDir() {
File cfgDir;
File cfgDir_linux = new File("/etc/taos");
cfgDir = cfgDir_linux.exists() ? cfgDir_linux : null;
File cfgDir_windows = new File("C:\\TDengine\\cfg");
cfgDir = (cfgDir == null && cfgDir_windows.exists()) ? cfgDir_windows : cfgDir;
return cfgDir;
}
private void connect(String host, int port, String dbName, String user, String password) throws SQLException { private void connect(String host, int port, String dbName, String user, String password) throws SQLException {
this.connector = new TSDBJNIConnector(); this.connector = new TSDBJNIConnector();
this.connector.connect(host, port, dbName, user, password); this.connector.connect(host, port, dbName, user, password);

6
src/connector/jdbc/src/main/java/com/taosdata/jdbc/TSDBDatabaseMetaData.java
View File

@ -68,15 +68,15 @@ public class TSDBDatabaseMetaData implements java.sql.DatabaseMetaData {
} }
public boolean nullsAreSortedLow() throws SQLException { public boolean nullsAreSortedLow() throws SQLException {
return false; return !nullsAreSortedHigh();
} }
public boolean nullsAreSortedAtStart() throws SQLException { public boolean nullsAreSortedAtStart() throws SQLException {
return false; return true;
} }
public boolean nullsAreSortedAtEnd() throws SQLException { public boolean nullsAreSortedAtEnd() throws SQLException {
return false; return !nullsAreSortedAtStart();
} }
public String getDatabaseProductName() throws SQLException { public String getDatabaseProductName() throws SQLException {

560
src/connector/jdbc/src/main/java/com/taosdata/jdbc/TSDBDriver.java
View File

@ -16,24 +16,27 @@ package com.taosdata.jdbc;
import org.apache.commons.lang3.StringUtils; import org.apache.commons.lang3.StringUtils;
import java.io.*;
import java.sql.*; import java.sql.*;
import java.util.ArrayList;
import java.util.List;
import java.util.Properties; import java.util.Properties;
import java.util.logging.Logger; import java.util.logging.Logger;
/** /**
* The Java SQL framework allows for multiple database drivers. Each driver * The Java SQL framework allows for multiple database drivers. Each driver
* should supply a class that implements the Driver interface * should supply a class that implements the Driver interface
* *
* <p> * <p>
* The DriverManager will try to load as many drivers as it can find and then * The DriverManager will try to load as many drivers as it can find and then
* for any given connection request, it will ask each driver in turn to try to * for any given connection request, it will ask each driver in turn to try to
* connect to the target URL. * connect to the target URL.
* *
* <p> * <p>
* It is strongly recommended that each Driver class should be small and stand * It is strongly recommended that each Driver class should be small and stand
* alone so that the Driver class can be loaded and queried without bringing in * alone so that the Driver class can be loaded and queried without bringing in
* vast quantities of supporting code. * vast quantities of supporting code.
* *
* <p> * <p>
* When a Driver class is loaded, it should create an instance of itself and * When a Driver class is loaded, it should create an instance of itself and
* register it with the DriverManager. This means that a user can load and * register it with the DriverManager. This means that a user can load and
@ -41,39 +44,39 @@ import java.util.logging.Logger;
*/ */
public class TSDBDriver implements java.sql.Driver { public class TSDBDriver implements java.sql.Driver {
@Deprecated @Deprecated
private static final String URL_PREFIX1 = "jdbc:TSDB://"; private static final String URL_PREFIX1 = "jdbc:TSDB://";
private static final String URL_PREFIX = "jdbc:TAOS://"; private static final String URL_PREFIX = "jdbc:TAOS://";
/** /**
* Key used to retrieve the database value from the properties instance passed * Key used to retrieve the database value from the properties instance passed
* to the driver. * to the driver.
*/ */
public static final String PROPERTY_KEY_DBNAME = "dbname"; public static final String PROPERTY_KEY_DBNAME = "dbname";
/** /**
* Key used to retrieve the host value from the properties instance passed to * Key used to retrieve the host value from the properties instance passed to
* the driver. * the driver.
*/ */
public static final String PROPERTY_KEY_HOST = "host"; public static final String PROPERTY_KEY_HOST = "host";
/** /**
* Key used to retrieve the password value from the properties instance passed * Key used to retrieve the password value from the properties instance passed
* to the driver. * to the driver.
*/ */
public static final String PROPERTY_KEY_PASSWORD = "password"; public static final String PROPERTY_KEY_PASSWORD = "password";
/** /**
* Key used to retrieve the port number value from the properties instance * Key used to retrieve the port number value from the properties instance
* passed to the driver. * passed to the driver.
*/ */
public static final String PROPERTY_KEY_PORT = "port"; public static final String PROPERTY_KEY_PORT = "port";
/** /**
* Key used to retrieve the user value from the properties instance passed to * Key used to retrieve the user value from the properties instance passed to
* the driver. * the driver.
*/ */
public static final String PROPERTY_KEY_USER = "user"; public static final String PROPERTY_KEY_USER = "user";
/** /**
* Key for the configuration file directory of TSDB client in properties instance * Key for the configuration file directory of TSDB client in properties instance
@ -98,277 +101,318 @@ public class TSDBDriver implements java.sql.Driver {
public static final String PROPERTY_KEY_PROTOCOL = "protocol"; public static final String PROPERTY_KEY_PROTOCOL = "protocol";
/** /**
* Index for port coming out of parseHostPortPair(). * Index for port coming out of parseHostPortPair().
*/ */
public final static int PORT_NUMBER_INDEX = 1; public final static int PORT_NUMBER_INDEX = 1;
/** /**
* Index for host coming out of parseHostPortPair(). * Index for host coming out of parseHostPortPair().
*/ */
public final static int HOST_NAME_INDEX = 0; public final static int HOST_NAME_INDEX = 0;
private TSDBDatabaseMetaData dbMetaData = null; private TSDBDatabaseMetaData dbMetaData = null;
static { static {
try { try {
java.sql.DriverManager.registerDriver(new TSDBDriver()); java.sql.DriverManager.registerDriver(new TSDBDriver());
} catch (SQLException E) { } catch (SQLException E) {
throw new RuntimeException(TSDBConstants.WrapErrMsg("can't register tdengine jdbc driver!")); throw new RuntimeException(TSDBConstants.WrapErrMsg("can't register tdengine jdbc driver!"));
} }
} }
public Connection connect(String url, Properties info) throws SQLException { private List<String> loadConfigEndpoints(File cfgFile) {
if (url == null) { List<String> endpoints = new ArrayList<>();
throw new SQLException(TSDBConstants.WrapErrMsg("url is not set!")); try (BufferedReader reader = new BufferedReader(new FileReader(cfgFile))) {
} String line = null;
while ((line = reader.readLine()) != null) {
if (line.trim().startsWith("firstEp") || line.trim().startsWith("secondEp")) {
endpoints.add(line.substring(line.indexOf('p') + 1).trim());
}
if (endpoints.size() > 1)
break;
}
} catch (FileNotFoundException e) {
e.printStackTrace();
} catch (IOException e) {
e.printStackTrace();
}
return endpoints;
}
Properties props = null; /**
* @param cfgDirPath
* @return return the config dir
**/
private File loadConfigDir(String cfgDirPath) {
if (cfgDirPath == null)
return loadDefaultConfigDir();
File cfgDir = new File(cfgDirPath);
if (!cfgDir.exists())
return loadDefaultConfigDir();
return cfgDir;
}
if ((props = parseURL(url, info)) == null) { /**
return null; * @return search the default config dir, if the config dir is not exist will return null
} */
private File loadDefaultConfigDir() {
File cfgDir;
File cfgDir_linux = new File("/etc/taos");
cfgDir = cfgDir_linux.exists() ? cfgDir_linux : null;
File cfgDir_windows = new File("C:\\TDengine\\cfg");
cfgDir = (cfgDir == null && cfgDir_windows.exists()) ? cfgDir_windows : cfgDir;
return cfgDir;
}
try { public Connection connect(String url, Properties info) throws SQLException {
TSDBJNIConnector.init((String) props.get(PROPERTY_KEY_CONFIG_DIR), (String) props.get(PROPERTY_KEY_LOCALE), (String) props.get(PROPERTY_KEY_CHARSET), if (url == null) {
(String) props.get(PROPERTY_KEY_TIME_ZONE)); throw new SQLException(TSDBConstants.WrapErrMsg("url is not set!"));
Connection newConn = new TSDBConnection(props, this.dbMetaData); }
return newConn;
} catch (SQLWarning sqlWarning) {
sqlWarning.printStackTrace();
Connection newConn = new TSDBConnection(props, this.dbMetaData);
return newConn;
} catch (SQLException sqlEx) {
throw sqlEx;
} catch (Exception ex) {
SQLException sqlEx = new SQLException("SQLException:" + ex.toString());
sqlEx.initCause(ex);
throw sqlEx;
}
}
/** Properties props = null;
* Parses hostPortPair in the form of [host][:port] into an array, with the if ((props = parseURL(url, info)) == null) {
* element of index HOST_NAME_INDEX being the host (or null if not specified), return null;
* and the element of index PORT_NUMBER_INDEX being the port (or null if not }
* specified).
*
* @param hostPortPair
* host and port in form of of [host][:port]
*
* @return array containing host and port as Strings
*
* @throws SQLException
* if a parse error occurs
*/
protected static String[] parseHostPortPair(String hostPortPair) throws SQLException {
String[] splitValues = new String[2];
int portIndex = hostPortPair.indexOf(":"); //load taos.cfg start
if (info.getProperty(TSDBDriver.PROPERTY_KEY_HOST) == null && info.getProperty(TSDBDriver.PROPERTY_KEY_PORT) == null){
File cfgDir = loadConfigDir(info.getProperty(TSDBDriver.PROPERTY_KEY_CONFIG_DIR));
File cfgFile = cfgDir.listFiles((dir, name) -> "taos.cfg".equalsIgnoreCase(name))[0];
List<String> endpoints = loadConfigEndpoints(cfgFile);
if (!endpoints.isEmpty()) {
info.setProperty(TSDBDriver.PROPERTY_KEY_HOST, endpoints.get(0).split(":")[0]);
info.setProperty(TSDBDriver.PROPERTY_KEY_PORT, endpoints.get(0).split(":")[1]);
}
}
String hostname = null; try {
TSDBJNIConnector.init((String) props.get(PROPERTY_KEY_CONFIG_DIR), (String) props.get(PROPERTY_KEY_LOCALE), (String) props.get(PROPERTY_KEY_CHARSET),
(String) props.get(PROPERTY_KEY_TIME_ZONE));
Connection newConn = new TSDBConnection(props, this.dbMetaData);
return newConn;
} catch (SQLWarning sqlWarning) {
sqlWarning.printStackTrace();
Connection newConn = new TSDBConnection(props, this.dbMetaData);
return newConn;
} catch (SQLException sqlEx) {
throw sqlEx;
} catch (Exception ex) {
SQLException sqlEx = new SQLException("SQLException:" + ex.toString());
sqlEx.initCause(ex);
throw sqlEx;
}
}
if (portIndex != -1) { /**
if ((portIndex + 1) < hostPortPair.length()) { * Parses hostPortPair in the form of [host][:port] into an array, with the
String portAsString = hostPortPair.substring(portIndex + 1); * element of index HOST_NAME_INDEX being the host (or null if not specified),
hostname = hostPortPair.substring(0, portIndex); * and the element of index PORT_NUMBER_INDEX being the port (or null if not
* specified).
*
* @param hostPortPair host and port in form of of [host][:port]
* @return array containing host and port as Strings
* @throws SQLException if a parse error occurs
*/
protected static String[] parseHostPortPair(String hostPortPair) throws SQLException {
String[] splitValues = new String[2];
splitValues[HOST_NAME_INDEX] = hostname; int portIndex = hostPortPair.indexOf(":");
splitValues[PORT_NUMBER_INDEX] = portAsString; String hostname = null;
} else {
throw new SQLException(TSDBConstants.WrapErrMsg("port is not proper!"));
}
} else {
splitValues[HOST_NAME_INDEX] = hostPortPair;
splitValues[PORT_NUMBER_INDEX] = null;
}
return splitValues; if (portIndex != -1) {
} if ((portIndex + 1) < hostPortPair.length()) {
String portAsString = hostPortPair.substring(portIndex + 1);
hostname = hostPortPair.substring(0, portIndex);
public boolean acceptsURL(String url) throws SQLException { splitValues[HOST_NAME_INDEX] = hostname;
return StringUtils.isNotBlank(url) && url.startsWith(URL_PREFIX);
}
public DriverPropertyInfo[] getPropertyInfo(String url, Properties info) throws SQLException { splitValues[PORT_NUMBER_INDEX] = portAsString;
if (info == null) { } else {
info = new Properties(); throw new SQLException(TSDBConstants.WrapErrMsg("port is not proper!"));
} }
} else {
splitValues[HOST_NAME_INDEX] = hostPortPair;
splitValues[PORT_NUMBER_INDEX] = null;
}
if ((url != null) && (url.startsWith(URL_PREFIX) || url.startsWith(URL_PREFIX1))) { return splitValues;
info = parseURL(url, info); }
}
DriverPropertyInfo hostProp = new DriverPropertyInfo(PROPERTY_KEY_HOST, info.getProperty(PROPERTY_KEY_HOST)); public boolean acceptsURL(String url) throws SQLException {
hostProp.required = true; return StringUtils.isNotBlank(url) && url.startsWith(URL_PREFIX);
}
DriverPropertyInfo portProp = new DriverPropertyInfo(PROPERTY_KEY_PORT, public DriverPropertyInfo[] getPropertyInfo(String url, Properties info) throws SQLException {
info.getProperty(PROPERTY_KEY_PORT, TSDBConstants.DEFAULT_PORT)); if (info == null) {
portProp.required = false; info = new Properties();
}
DriverPropertyInfo dbProp = new DriverPropertyInfo(PROPERTY_KEY_DBNAME, info.getProperty(PROPERTY_KEY_DBNAME)); if ((url != null) && (url.startsWith(URL_PREFIX) || url.startsWith(URL_PREFIX1))) {
dbProp.required = false; info = parseURL(url, info);
dbProp.description = "Database name"; }
DriverPropertyInfo userProp = new DriverPropertyInfo(PROPERTY_KEY_USER, info.getProperty(PROPERTY_KEY_USER)); DriverPropertyInfo hostProp = new DriverPropertyInfo(PROPERTY_KEY_HOST, info.getProperty(PROPERTY_KEY_HOST));
userProp.required = true; hostProp.required = true;
DriverPropertyInfo passwordProp = new DriverPropertyInfo(PROPERTY_KEY_PASSWORD, DriverPropertyInfo portProp = new DriverPropertyInfo(PROPERTY_KEY_PORT, info.getProperty(PROPERTY_KEY_PORT, TSDBConstants.DEFAULT_PORT));
info.getProperty(PROPERTY_KEY_PASSWORD)); portProp.required = false;
passwordProp.required = true;
DriverPropertyInfo[] propertyInfo = new DriverPropertyInfo[5]; DriverPropertyInfo dbProp = new DriverPropertyInfo(PROPERTY_KEY_DBNAME, info.getProperty(PROPERTY_KEY_DBNAME));
propertyInfo[0] = hostProp; dbProp.required = false;
propertyInfo[1] = portProp; dbProp.description = "Database name";
propertyInfo[2] = dbProp;
propertyInfo[3] = userProp;
propertyInfo[4] = passwordProp;
return propertyInfo; DriverPropertyInfo userProp = new DriverPropertyInfo(PROPERTY_KEY_USER, info.getProperty(PROPERTY_KEY_USER));
} userProp.required = true;
/** DriverPropertyInfo passwordProp = new DriverPropertyInfo(PROPERTY_KEY_PASSWORD, info.getProperty(PROPERTY_KEY_PASSWORD));
* example: jdbc:TSDB://127.0.0.1:0/db?user=root&password=your_password passwordProp.required = true;
*/
public Properties parseURL(String url, Properties defaults) throws java.sql.SQLException { DriverPropertyInfo[] propertyInfo = new DriverPropertyInfo[5];
Properties urlProps = (defaults != null) ? defaults : new Properties(); propertyInfo[0] = hostProp;
if (url == null) { propertyInfo[1] = portProp;
return null; propertyInfo[2] = dbProp;
} propertyInfo[3] = userProp;
propertyInfo[4] = passwordProp;
if (!StringUtils.startsWithIgnoreCase(url, URL_PREFIX) && !StringUtils.startsWithIgnoreCase(url, URL_PREFIX1)) { return propertyInfo;
return null; }
}
String urlForMeta = url; /**
* example: jdbc:TSDB://127.0.0.1:0/db?user=root&password=your_password
*/
public Properties parseURL(String url, Properties defaults) throws java.sql.SQLException {
Properties urlProps = (defaults != null) ? defaults : new Properties();
if (url == null) {
return null;
}
String dbProductName = url.substring(url.indexOf(":") + 1); if (!StringUtils.startsWithIgnoreCase(url, URL_PREFIX) && !StringUtils.startsWithIgnoreCase(url, URL_PREFIX1)) {
dbProductName = dbProductName.substring(0, dbProductName.indexOf(":")); return null;
int beginningOfSlashes = url.indexOf("//"); }
url = url.substring(beginningOfSlashes + 2);
String host = url.substring(0, url.indexOf(":")); String urlForMeta = url;
url = url.substring(url.indexOf(":") + 1);
urlProps.setProperty(PROPERTY_KEY_HOST, host);
String port = url.substring(0, url.indexOf("/")); String dbProductName = url.substring(url.indexOf(":") + 1);
urlProps.setProperty(PROPERTY_KEY_PORT, port); dbProductName = dbProductName.substring(0, dbProductName.indexOf(":"));
url = url.substring(url.indexOf("/") + 1); int beginningOfSlashes = url.indexOf("//");
url = url.substring(beginningOfSlashes + 2);
if (url.indexOf("?") != -1) { String host = url.substring(0, url.indexOf(":"));
String dbName = url.substring(0, url.indexOf("?")); url = url.substring(url.indexOf(":") + 1);
urlProps.setProperty(PROPERTY_KEY_DBNAME, dbName); urlProps.setProperty(PROPERTY_KEY_HOST, host);
url = url.trim().substring(url.indexOf("?") + 1);
} else {
// without user & password so return
if(!url.trim().isEmpty()) {
String dbName = url.trim();
urlProps.setProperty(PROPERTY_KEY_DBNAME, dbName);
}
this.dbMetaData = new TSDBDatabaseMetaData(dbProductName, urlForMeta, urlProps.getProperty("user"));
return urlProps;
}
String user = ""; String port = url.substring(0, url.indexOf("/"));
urlProps.setProperty(PROPERTY_KEY_PORT, port);
url = url.substring(url.indexOf("/") + 1);
if (url.indexOf("&") == -1) { if (url.indexOf("?") != -1) {
String[] kvPair = url.trim().split("="); String dbName = url.substring(0, url.indexOf("?"));
if (kvPair.length == 2) { urlProps.setProperty(PROPERTY_KEY_DBNAME, dbName);
setPropertyValue(urlProps, kvPair); url = url.trim().substring(url.indexOf("?") + 1);
return urlProps; } else {
} // without user & password so return
} if (!url.trim().isEmpty()) {
String dbName = url.trim();
urlProps.setProperty(PROPERTY_KEY_DBNAME, dbName);
}
this.dbMetaData = new TSDBDatabaseMetaData(dbProductName, urlForMeta, urlProps.getProperty("user"));
return urlProps;
}
String[] queryStrings = url.trim().split("&"); String user = "";
for (String queryStr : queryStrings) {
String[] kvPair = queryStr.trim().split("=");
if (kvPair.length < 2){
continue;
}
setPropertyValue(urlProps, kvPair);
}
user = urlProps.getProperty(PROPERTY_KEY_USER).toString(); if (url.indexOf("&") == -1) {
this.dbMetaData = new TSDBDatabaseMetaData(dbProductName, urlForMeta, user); String[] kvPair = url.trim().split("=");
if (kvPair.length == 2) {
setPropertyValue(urlProps, kvPair);
return urlProps;
}
}
return urlProps; String[] queryStrings = url.trim().split("&");
} for (String queryStr : queryStrings) {
String[] kvPair = queryStr.trim().split("=");
if (kvPair.length < 2) {
continue;
}
setPropertyValue(urlProps, kvPair);
}
public void setPropertyValue(Properties property, String[] keyValuePair) { user = urlProps.getProperty(PROPERTY_KEY_USER).toString();
switch (keyValuePair[0].toLowerCase()) { this.dbMetaData = new TSDBDatabaseMetaData(dbProductName, urlForMeta, user);
case PROPERTY_KEY_USER:
property.setProperty(PROPERTY_KEY_USER, keyValuePair[1]); return urlProps;
break; }
case PROPERTY_KEY_PASSWORD:
property.setProperty(PROPERTY_KEY_PASSWORD, keyValuePair[1]); public void setPropertyValue(Properties property, String[] keyValuePair) {
break; switch (keyValuePair[0].toLowerCase()) {
case PROPERTY_KEY_TIME_ZONE: case PROPERTY_KEY_USER:
property.setProperty(PROPERTY_KEY_TIME_ZONE, keyValuePair[1]); property.setProperty(PROPERTY_KEY_USER, keyValuePair[1]);
break; break;
case PROPERTY_KEY_LOCALE: case PROPERTY_KEY_PASSWORD:
property.setProperty(PROPERTY_KEY_LOCALE, keyValuePair[1]); property.setProperty(PROPERTY_KEY_PASSWORD, keyValuePair[1]);
break; break;
case PROPERTY_KEY_CHARSET: case PROPERTY_KEY_TIME_ZONE:
property.setProperty(PROPERTY_KEY_CHARSET, keyValuePair[1]); property.setProperty(PROPERTY_KEY_TIME_ZONE, keyValuePair[1]);
break; break;
case PROPERTY_KEY_CONFIG_DIR: case PROPERTY_KEY_LOCALE:
property.setProperty(PROPERTY_KEY_CONFIG_DIR, keyValuePair[1]); property.setProperty(PROPERTY_KEY_LOCALE, keyValuePair[1]);
break; break;
} case PROPERTY_KEY_CHARSET:
} property.setProperty(PROPERTY_KEY_CHARSET, keyValuePair[1]);
break;
case PROPERTY_KEY_CONFIG_DIR:
property.setProperty(PROPERTY_KEY_CONFIG_DIR, keyValuePair[1]);
break;
}
}
public int getMajorVersion() { public int getMajorVersion() {
return 1; return 1;
} }
public int getMinorVersion() { public int getMinorVersion() {
return 1; return 1;
} }
public boolean jdbcCompliant() { public boolean jdbcCompliant() {
return false; return false;
} }
public Logger getParentLogger() throws SQLFeatureNotSupportedException { public Logger getParentLogger() throws SQLFeatureNotSupportedException {
return null; return null;
} }
/** /**
* Returns the host property * Returns the host property
* *
* @param props * @param props the java.util.Properties instance to retrieve the hostname from.
* the java.util.Properties instance to retrieve the hostname from. * @return the host
* */
* @return the host public String host(Properties props) {
*/ return props.getProperty(PROPERTY_KEY_HOST, "localhost");
public String host(Properties props) { }
return props.getProperty(PROPERTY_KEY_HOST, "localhost");
}
/** /**
* Returns the port number property * Returns the port number property
* *
* @param props * @param props the properties to get the port number from
* the properties to get the port number from * @return the port number
* */
* @return the port number public int port(Properties props) {
*/ return Integer.parseInt(props.getProperty(PROPERTY_KEY_PORT, TSDBConstants.DEFAULT_PORT));
public int port(Properties props) { }
return Integer.parseInt(props.getProperty(PROPERTY_KEY_PORT, TSDBConstants.DEFAULT_PORT));
}
/** /**
* Returns the database property from <code>props</code> * Returns the database property from <code>props</code>
* *
* @param props * @param props the Properties to look for the database property.
* the Properties to look for the database property. * @return the database name.
* */
* @return the database name. public String database(Properties props) {
*/ return props.getProperty(PROPERTY_KEY_DBNAME);
public String database(Properties props) { }
return props.getProperty(PROPERTY_KEY_DBNAME);
}
} }

80
src/connector/jdbc/src/main/java/com/taosdata/jdbc/cases/CalciteTest.java Normal file
View File

@ -0,0 +1,80 @@
package com.taosdata.jdbc.cases;
import org.apache.calcite.adapter.jdbc.JdbcSchema;
import org.apache.calcite.jdbc.CalciteConnection;
import org.apache.calcite.schema.SchemaPlus;
import org.apache.calcite.sql.parser.SqlParseException;
import org.apache.commons.dbcp2.BasicDataSource;
import java.sql.*;
import java.util.HashMap;
import java.util.Map;
import java.util.Properties;
public class CalciteTest {
public static void main(String[] args) throws SqlParseException, ClassNotFoundException, SQLException {
//创建Calcite Connection对象
Class.forName("org.apache.calcite.jdbc.Driver");
Properties info = new Properties();
info.setProperty("caseSensitive", "false");
Connection connection = DriverManager.getConnection("jdbc:calcite:", info);
CalciteConnection calciteConnection = connection.unwrap(CalciteConnection.class);
SchemaPlus rootSchema = calciteConnection.getRootSchema();
// JDBC adapter
Class.forName("com.mysql.jdbc.Driver");
BasicDataSource dataSource = new BasicDataSource();
dataSource.setUrl("jdbc:mysql://192.168.56.101:3306");
dataSource.setUsername("root");
dataSource.setPassword("123456");
Map<String, String> map = new HashMap<>();
JdbcSchema schema = JdbcSchema.create(rootSchema, "hr", dataSource, null, null);
rootSchema.add("hr", schema);
Statement statement = calciteConnection.createStatement();
ResultSet resultSet = statement.executeQuery("select * from hr.depts");
while (resultSet.next()) {
ResultSetMetaData metaData = resultSet.getMetaData();
for (int i = 1; i <= metaData.getColumnCount(); i++) {
String columnLabel = metaData.getColumnLabel(i);
System.out.println(columnLabel + " : " + resultSet.getString(i));
}
}
resultSet.close();
statement.close();
connection.close();
//创建TDengine的数据源schema
// Class.forName("com.taosdata.jdbc.TSDBDriver");
// String url = "jdbc:TAOS://127.0.0.1:6030/hdb";
// dataSource.setUrl(url);
// dataSource.setUsername("root");
// dataSource.setPassword("taosdata");
// Class.forName("com.mysql.jdbc.Driver");
// String url = "jdbc:mysql://localhost:3306/hdb";
// BasicDataSource dataSource = new BasicDataSource();
// dataSource.setUrl(url);
// dataSource.setUsername("root");
// dataSource.setPassword("123456");
//这里hdb是在tdengine中创建的数据库名
// JdbcSchema schema = JdbcSchema.create(rootSchema, "test", dataSource, null, "test");
// Schema schema = JdbcSchema.create(rootSchema, "test", dataSource, "hdb", null);
//创建新的schema自动映射到原来的hdb数据库
// rootSchema.add("test", schema);
// Statement stmt = calciteConnection.createStatement();
//查询schema test中的表，表名是tdengine中的表
// ResultSet rs = stmt.executeQuery("select * from test.t");
// for (int i = 1; i <= rs.getMetaData().getColumnCount(); i++) {
// System.out.println(rs.getMetaData().getColumnName(i));
// }
// while (rs.next()) {
// System.out.println(rs.getObject(1));
// }
}
}