Try to make a database connection to the given URL. The driver should return "null" if it - * realizes it is the wrong kind of driver to connect to the given URL. This will be common, as - * when the JDBC driverManager is asked to connect to a given URL, it passes the URL to each - * loaded driver in turn.
- * - *The driver should raise an SQLException if it is the right driver to connect to the given URL, - * but has trouble connecting to the database.
- * - *The java.util.Properties argument can be used to pass arbitrary string tag/value pairs as - * connection arguments.
- * - *Normally, at least "user" and "password" properties should be included in the properties. For a
- * list of supported character encoding , see
- * http://java.sun.com/products/jdk/1.2/docs/guide/internat/encoding.doc.html Note that you will
- * probably want to have set up the Postgres database itself to use the same encoding, with the
- * {@code -E
Our protocol takes the forms:
- * - *- * jdbc:postgresql://host:port/database?param1=val1&... - *- * - * @param url the URL of the database to connect to - * @param info a list of arbitrary tag/value pairs as connection arguments - * @return a connection to the URL or null if it isnt us - * @throws SQLException if a database access error occurs - * @see java.sql.Driver#connect - */ - @Override - public Connection connect(String url, Properties info) throws SQLException { - // get defaults - Properties defaults, props; - if (!url.startsWith("jdbc:postgresql:") && !url.startsWith("jdbc:dws:iam:")) { - return null; - } - try { - defaults = getDefaultProperties(); - } catch (IOException ioe) { - throw new PSQLException(GT.tr("Error loading default settings from driverconfig.properties"), PSQLState.UNEXPECTED_ERROR, ioe); + /** + *
Try to make a database connection to the given URL. The driver should return "null" if it + * realizes it is the wrong kind of driver to connect to the given URL. This will be common, as + * when the JDBC driverManager is asked to connect to a given URL, it passes the URL to each + * loaded driver in turn.
+ * + *The driver should raise an SQLException if it is the right driver to connect to the given URL, + * but has trouble connecting to the database.
+ * + *The java.util.Properties argument can be used to pass arbitrary string tag/value pairs as + * connection arguments.
+ * + *Normally, at least "user" and "password" properties should be included in the properties. For a
+ * list of supported character encoding , see
+ * http://java.sun.com/products/jdk/1.2/docs/guide/internat/encoding.doc.html Note that you will
+ * probably want to have set up the Postgres database itself to use the same encoding, with the
+ * {@code -E
Our protocol takes the forms:
+ * + *+ * jdbc:postgresql://host:port/database?param1=val1&... + *+ * + * @param url the URL of the database to connect to + * @param info a list of arbitrary tag/value pairs as connection arguments + * @return a connection to the URL or null if it isnt us + * @throws SQLException if a database access error occurs + * @see java.sql.Driver#connect + */ + @Override + public Connection connect(String url, Properties info) throws SQLException { + // get defaults + Properties defaults, props; + + String[] legalUrlTags = {"jdbc:postgresql:", "jdbc:dws:iam:"}; + boolean isUrlLegal = false; + for (String urlTag : legalUrlTags) { + if (url.startsWith(urlTag)) { + isUrlLegal = true; + } + } + if (!isUrlLegal) { + return null; + } + + try { + defaults = getDefaultProperties(); + } catch (IOException ioe) { + throw new PSQLException(GT.tr("Error loading default settings from driverconfig.properties"), + PSQLState.UNEXPECTED_ERROR, ioe); + } + + props = GetProps(defaults, info); + + // parse URL and add more properties + if ((props = parseURL(url, props)) == null) { + return null; + } + + if (!parseConnectionProperties(props)) { + return null; + } + + try { + LOGGER.debug("Connecting with URL: " + filterAuthenticationCredentials(url)); + + // Enforce login timeout, if specified, by running the connection + // attempt in a separate thread. If we hit the timeout without the + // connection completing, we abandon the connection attempt in + // the calling thread, but the separate thread will keep trying. + // Eventually, the separate thread will either fail or complete + // the connection; at that point we clean up the connection if + // we managed to establish one after all. See ConnectThread for + // more details. + long timeout = timeout(props); + if (timeout <= 0) { + Connection con = makeConnection(url, props); + return con; + } + + ConnectThread ct = new ConnectThread(url, props); + Thread thread = new Thread(ct, "PostgreSQL JDBC driver connection thread"); + thread.setDaemon(true); // Don't prevent the VM from shutting down + thread.start(); + return ct.getResult(timeout); + } catch (PSQLException ex1) { + LOGGER.debug("Connection error: ", ex1); + // re-throw the exception, otherwise it will be caught next, and a + // org.postgresql.unusual error will be returned instead. + throw ex1; + } catch (java.security.AccessControlException ace) { + throw new PSQLException(GT.tr("Your security policy has prevented the connection from being attempted. " + + "You probably need to grant the connect java.net.SocketPermission to the database server host and" + + " port that you wish to connect to."), PSQLState.UNEXPECTED_ERROR, ace); + } catch (Exception ex2) { + LOGGER.debug("Unexpected connection error: ", ex2); + throw new PSQLException(GT.tr("Something unusual has occured to cause the driver to fail. Please report " + + "this exception."), PSQLState.UNEXPECTED_ERROR, ex2); + } } - props = GetProps(defaults, info); + /** + * Parse the configuration items in the connection string. + * + * @param props Connection Properties + */ + private Boolean parseConnectionProperties(Properties props) { + Logger.setLoggerName(props.getProperty("logger")); + if (Logger.isUsingJDKLogger()) { + setupLoggerFromProperties(props); + } else { + LOGGER = Logger.getLogger("org.postgresql.Driver"); + } - // parse URL and add more properties - if ((props = parseURL(url, props)) == null) { - return null; + if (PGProperty.TRACE_INTERFACE_CLASS.get(props) != null && tracer == null) { + String traceClass = PGProperty.TRACE_INTERFACE_CLASS.get(props); + try { + if (tracerInitialized.compareAndSet(false, true)) { + tracer = Class.forName(traceClass).asSubclass(Tracer.class).newInstance(); + } + } catch (ClassCastException ce1) { + tracerInitialized.set(false); + LOGGER.warn("The currently specified class " + traceClass + ", not implement Tracer class."); + } catch (ClassNotFoundException | InstantiationException | IllegalAccessException ce2) { + tracerInitialized.set(false); + LOGGER.warn("Failed to load tracer implementation class, error: ", ce2); + } + } + Boolean parseStatus = true; + if (PGProperty.PRIORITY_SERVERS.get(props) != null) { + if (!GlobalClusterStatusTracker.isVaildPriorityServers(props)) { + parseStatus = false; + } + GlobalClusterStatusTracker.refreshProperties(props); + } + + if (MultiHostChooser.isUsingAutoLoadBalance(props)) { + if (!MultiHostChooser.isVaildPriorityLoadBalance(props)) { + parseStatus = false; + } + QueryCNListUtils.refreshProperties(props); + } + return parseStatus; } + // Used to check if the handler file is the same + private static String loggerHandlerFile; - - Logger.setLoggerName(props.getProperty("logger")); - - if(Logger.isUsingJDKLogger()) { - setupLoggerFromProperties(props); - } else { - LOGGER = Logger.getLogger("org.postgresql.Driver"); - } - - if(PGProperty.PRIORITY_SERVERS.get(props) != null){ - if(!GlobalClusterStatusTracker.isVaildPriorityServers(props)){ - return null; - } - GlobalClusterStatusTracker.refreshProperties(props); - } - - if (MultiHostChooser.isUsingAutoLoadBalance(props)) { - if(!MultiHostChooser.isVaildPriorityLoadBalance(props)){ - return null; - } - QueryCNListUtils.refreshProperties(props); - } - - try { - LOGGER.debug("Connecting with URL: " + url); - - // Enforce login timeout, if specified, by running the connection - // attempt in a separate thread. If we hit the timeout without the - // connection completing, we abandon the connection attempt in - // the calling thread, but the separate thread will keep trying. - // Eventually, the separate thread will either fail or complete - // the connection; at that point we clean up the connection if - // we managed to establish one after all. See ConnectThread for - // more details. - long timeout = timeout(props); - if (timeout <= 0) { - Connection con = makeConnection(url, props); - return con; - } - - ConnectThread ct = new ConnectThread(url, props); - Thread thread = new Thread(ct, "PostgreSQL JDBC driver connection thread"); - thread.setDaemon(true); // Don't prevent the VM from shutting down - thread.start(); - return ct.getResult(timeout); - } catch (PSQLException ex1) { - LOGGER.debug("Connection error: ",ex1); - // re-throw the exception, otherwise it will be caught next, and a - // org.postgresql.unusual error will be returned instead. - throw ex1; - } catch (java.security.AccessControlException ace) { - throw new PSQLException(GT.tr("Your security policy has prevented the connection from being attempted. You probably need to grant the connect java.net.SocketPermission to the database server host and port that you wish to connect to."), PSQLState.UNEXPECTED_ERROR, ace); - } catch (Exception ex2) { - LOGGER.debug("Unexpected connection error: ", ex2); - throw new PSQLException(GT.tr("Something unusual has occured to cause the driver to fail. Please report this exception."), PSQLState.UNEXPECTED_ERROR, ex2); - } - } - - // Used to check if the handler file is the same - private static String loggerHandlerFile; - - /** - *
Setup java.util.logging.Logger using connection properties.
- * - *See {@link PGProperty#LOGGER_FILE} and {@link PGProperty#LOGGER_FILE}
- * - * @param props Connection Properties - */ - private Boolean initLoggerProperties(String driverLogLevel) { - if (driverLogLevel == null) { - return false; // Don't mess with Logger if not set + /** + *Setup java.util.logging.Logger using connection properties.
+ * + *See {@link PGProperty#LOGGER_FILE} and {@link PGProperty#LOGGER_FILE}
+ * + * @param driverLogLevel Connection Properties + */ + private Boolean initLoggerProperties(String driverLogLevel) { + if (driverLogLevel == null) { + return false; // Don't mess with Logger if not set } if (!"OFF".equalsIgnoreCase(driverLogLevel) && !isLogFileCreated.compareAndSet(false, true)) { return false; } if ("OFF".equalsIgnoreCase(driverLogLevel)) { - PARENT_LOGGER.setLevel(Level.OFF); - return false; // Don't mess with Logger if set to OFF + PARENT_LOGGER.setLevel(Level.OFF); + return false; // Don't mess with Logger if set to OFF } else if ("DEBUG".equalsIgnoreCase(driverLogLevel)) { - PARENT_LOGGER.setLevel(Level.FINE); + PARENT_LOGGER.setLevel(Level.FINE); } else if ("TRACE".equalsIgnoreCase(driverLogLevel)) { - PARENT_LOGGER.setLevel(Level.FINEST); + PARENT_LOGGER.setLevel(Level.FINEST); } else if ("INFO".equalsIgnoreCase(driverLogLevel)) { PARENT_LOGGER.setLevel(Level.INFO); } else { - PARENT_LOGGER.setLevel(Level.OFF); + PARENT_LOGGER.setLevel(Level.OFF); } return true; - } - /** - *Setup java.util.logging.Logger using connection properties.
- * - *See {@link PGProperty#LOGGER_FILE} and {@link PGProperty#LOGGER_FILE}
- * - * @param props Connection Properties - */ - private void setupLoggerFromProperties(final Properties props) { - final String driverLogLevel = PGProperty.LOGGER_LEVEL.get(props); - if (!initLoggerProperties(driverLogLevel)) { - return; - } - - ExpressionProperties exprProps = new ExpressionProperties(props, System.getProperties()); - final String driverLogFile = PGProperty.LOGGER_FILE.get(exprProps); - if (driverLogFile != null && driverLogFile.equals(loggerHandlerFile)) { - return; // Same file output, do nothing. - } - - for (java.util.logging.Handler handlers : PARENT_LOGGER.getHandlers()) { - // Remove previously set Handlers - handlers.close(); - PARENT_LOGGER.removeHandler(handlers); - loggerHandlerFile = null; - } - - java.util.logging.Handler handler = null; - if (driverLogFile != null) { - try { - handler = new java.util.logging.FileHandler(driverLogFile); - loggerHandlerFile = driverLogFile; - } catch (Exception ex) { - System.err.println("Cannot enable FileHandler, fallback to ConsoleHandler."); - } - } - - Formatter formatter = new SimpleFormatter(); - - if ( handler == null ) { - if (DriverManager.getLogWriter() != null) { - handler = new WriterHandler(DriverManager.getLogWriter()); - } else if ( DriverManager.getLogStream() != null) { - handler = new StreamHandler(DriverManager.getLogStream(), formatter); - } else { - handler = new StreamHandler(System.err, formatter); - } - } else { - handler.setFormatter(formatter); - } - - handler.setLevel(PARENT_LOGGER.getLevel()); - PARENT_LOGGER.setUseParentHandlers(false); - PARENT_LOGGER.addHandler(handler); - } - - /** - * Perform a connect in a separate thread; supports getting the results from the original thread - * while enforcing a login timeout. - */ - private static class ConnectThread implements Runnable { - ConnectThread(String url, Properties props) { - this.url = url; - this.props = props; - } - - public void run() { - Connection conn; - Throwable error; - - try { - conn = makeConnection(url, props); - error = null; - } catch (Throwable t) { - conn = null; - error = t; - } - - synchronized (this) { - if (abandoned) { - if (conn != null) { - try { - conn.close(); - } catch (SQLException e) { - } - } - } else { - result = conn; - resultException = error; - notify(); - } - } } /** - * Get the connection result from this (assumed running) thread. If the timeout is reached - * without a result being available, a SQLException is thrown. + *Setup java.util.logging.Logger using connection properties.
* - * @param timeout timeout in milliseconds - * @return the new connection, if successful - * @throws SQLException if a connection error occurs or the timeout is reached + *See {@link PGProperty#LOGGER_FILE} and {@link PGProperty#LOGGER_FILE}
+ * + * @param props Connection Properties */ - public Connection getResult(long timeout) throws SQLException { - long expiry = System.currentTimeMillis() + timeout; - synchronized (this) { - while (true) { - if (result != null) { - return result; - } - - if (resultException != null) { - if (resultException instanceof SQLException) { - resultException.fillInStackTrace(); - throw (SQLException) resultException; - } else { - throw new PSQLException( - GT.tr( - "Something unusual has occured to cause the driver to fail. Please report this exception."), - PSQLState.UNEXPECTED_ERROR, resultException); - } - } - - long delay = expiry - System.currentTimeMillis(); - if (delay <= 0) { - abandoned = true; - throw new PSQLException(GT.tr("Connection attempt timed out."), - PSQLState.CONNECTION_UNABLE_TO_CONNECT); - } - - try { - wait(delay); - } catch (InterruptedException ie) { - - // reset the interrupt flag - Thread.currentThread().interrupt(); - abandoned = true; - - // throw an unchecked exception which will hopefully not be ignored by the calling code - throw new RuntimeException(GT.tr("Interrupted while attempting to connect.")); - } + private void setupLoggerFromProperties(final Properties props) { + final String driverLogLevel = PGProperty.LOGGER_LEVEL.get(props); + if (!initLoggerProperties(driverLogLevel)) { + return; } - } - } - private final String url; - private final Properties props; - private Connection result; - private Throwable resultException; - private boolean abandoned; - } + final String driverLogFile = PGProperty.LOGGER_FILE.getDefaultValue(); + if (driverLogFile != null && driverLogFile.equals(loggerHandlerFile)) { + return; // Same file output, do nothing. + } - /** - * Create a connection from URL and properties. Always does the connection work in the current - * thread without enforcing a timeout, regardless of any timeout specified in the properties. - * - * @param url the original URL - * @param props the parsed/defaulted connection properties - * @return a new connection - * @throws SQLException if the connection could not be made - */ - private static Connection makeConnection(String url, Properties props) throws SQLException { - PgConnection pgConnection = new PgConnection(hostSpecs(props), user(props), database(props), props, url); - GlobalConnectionTracker.possessConnectionReference(pgConnection.getQueryExecutor(), props); - return pgConnection; - } + for (java.util.logging.Handler handlers : PARENT_LOGGER.getHandlers()) { + // Remove previously set Handlers + handlers.close(); + PARENT_LOGGER.removeHandler(handlers); + loggerHandlerFile = null; + } - /** - * Returns true if the driver thinks it can open a connection to the given URL. Typically, drivers - * will return true if they understand the subprotocol specified in the URL and false if they - * don't. Our protocols start with jdbc:postgresql: - * - * @param url the URL of the driver - * @return true if this driver accepts the given URL - * @throws PSQLException - * @see java.sql.Driver#acceptsURL - */ - @Override - public boolean acceptsURL(String url) throws PSQLException { - return parseURL(url, null) != null; - } - - /** - *The getPropertyInfo method is intended to allow a generic GUI tool to discover what properties - * it should prompt a human for in order to get enough information to connect to a database.
- * - *Note that depending on the values the human has supplied so far, additional values may become - * necessary, so it may be necessary to iterate through several calls to getPropertyInfo
- * - * @param url the Url of the database to connect to - * @param info a proposed list of tag/value pairs that will be sent on connect open. - * @return An array of DriverPropertyInfo objects describing possible properties. This array may - * be an empty array if no properties are required - * @throws PSQLException - * @see java.sql.Driver#getPropertyInfo - */ - @Override - public DriverPropertyInfo[] getPropertyInfo(String url, Properties info) throws PSQLException { - Properties copy = new Properties(info); - Properties parse = parseURL(url, copy); - if (parse != null) { - copy = parse; - } - - PGProperty[] knownProperties = PGProperty.values(); - DriverPropertyInfo[] props = new DriverPropertyInfo[knownProperties.length]; - for (int i = 0; i < props.length; ++i) { - props[i] = knownProperties[i].toDriverPropertyInfo(copy); - } - - return props; - } - - @Override - public int getMajorVersion() { - return org.postgresql.util.DriverInfo.MAJOR_VERSION; - } - - @Override - public int getMinorVersion() { - return org.postgresql.util.DriverInfo.MINOR_VERSION; - } - - /** - * Returns the server version series of this driver and the specific build number. - * - * @return JDBC driver version - * @deprecated use {@link #getMajorVersion()} and {@link #getMinorVersion()} instead - */ - @Deprecated - public static String getVersion() { - return DriverInfo.DRIVER_FULL_NAME; - } - - /** - *Report whether the driver is a genuine JDBC compliant driver. A driver may only report "true" - * here if it passes the JDBC compliance tests, otherwise it is required to return false. JDBC - * compliance requires full support for the JDBC API and full support for SQL 92 Entry Level.
- * - *For PostgreSQL, this is not yet possible, as we are not SQL92 compliant (yet).
- */ - @Override - public boolean jdbcCompliant() { - return false; - } - - public static String parseIPValid(String address) { - if (address == "" || address == null) { - return address; - } - String regex = "^(1\\d{2}|2[0-4]\\d|25[0-5]|[1-9]\\d|[1-9])\\." - + "(1\\d{2}|2[0-4]\\d|25[0-5]|[1-9]\\d|\\d)\\." - + "(1\\d{2}|2[0-4]\\d|25[0-5]|[1-9]\\d|\\d)\\." - + "(1\\d{2}|2[0-4]\\d|25[0-5]|[1-9]\\d|\\d)$"; - Pattern pattern = Pattern.compile(regex); - if (!pattern.matcher(address).matches()) { - LOGGER.debug("JDBC URL invalid ip address: " + address); - } - return address; - } - - /** - * Constructs a new DriverURL, splitting the specified URL into its component parts. - * - * @param url JDBC URL to parse - * @param defaults Default properties - * @return Properties with elements added from the url - * @throws PSQLException - */ - public static Properties parseURL(String url, Properties defaults) throws PSQLException { - Properties urlProps = new Properties(defaults); - String l_urlServer = url; - String l_urlArgs = ""; - - int l_qPos = url.indexOf('?'); - if (l_qPos != -1) { - l_urlServer = url.substring(0, l_qPos); - l_urlArgs = url.substring(l_qPos + 1); - } - - if (!l_urlServer.startsWith("jdbc:postgresql:") && !l_urlServer.startsWith("jdbc:dws:iam:")) { - LOGGER.debug("JDBC URL must start with \"jdbc:postgresql:\" or \"jdbc:dws:iam:\" but was: " + url); - return null; - } - if(l_urlServer.startsWith("jdbc:postgresql:")) { - l_urlServer = l_urlServer.substring("jdbc:postgresql:".length()); - - if (l_urlServer.startsWith("//")) { - l_urlServer = l_urlServer.substring(2); - int slash = l_urlServer.indexOf('/'); - if (slash == -1) { - LOGGER.warn("JDBC URL must contain a / at the end of the host or port: " + url); - return null; + java.util.logging.Handler handler = null; + if (driverLogFile != null) { + try { + handler = new java.util.logging.FileHandler(driverLogFile); + loggerHandlerFile = driverLogFile; + } catch (Exception ex) { + System.err.println("Cannot enable FileHandler, fallback to ConsoleHandler."); } - urlProps.setProperty("PGDBNAME", URLCoder.decode(l_urlServer.substring(slash + 1))); + } + + Formatter formatter = new SimpleFormatter(); + + if (handler == null) { + if (DriverManager.getLogWriter() != null) { + handler = new WriterHandler(DriverManager.getLogWriter()); + } else if (DriverManager.getLogStream() != null) { + handler = new StreamHandler(DriverManager.getLogStream(), formatter); + } else { + handler = new StreamHandler(System.err, formatter); + } + } else { + handler.setFormatter(formatter); + } + + handler.setLevel(PARENT_LOGGER.getLevel()); + PARENT_LOGGER.setUseParentHandlers(false); + PARENT_LOGGER.addHandler(handler); + } + + /** + * Perform a connect in a separate thread; supports getting the results from the original thread + * while enforcing a login timeout. + */ + private static class ConnectThread implements Runnable { + ConnectThread(String url, Properties props) { + this.url = url; + this.props = props; + } + + public void run() { + Connection conn; + Throwable error; + + try { + conn = makeConnection(url, props); + error = null; + } catch (Throwable t) { + conn = null; + error = t; + } + + synchronized (this) { + if (abandoned) { + if (conn != null) { + try { + conn.close(); + } catch (SQLException e) { + } + } + } else { + result = conn; + resultException = error; + notify(); + } + } + } + + /** + * Get the connection result from this (assumed running) thread. If the timeout is reached + * without a result being available, a SQLException is thrown. + * + * @param timeout timeout in milliseconds + * @return the new connection, if successful + * @throws SQLException if a connection error occurs or the timeout is reached + */ + public Connection getResult(long timeout) throws SQLException { + long expiry = System.currentTimeMillis() + timeout; + synchronized (this) { + while (true) { + if (result != null) { + return result; + } + + if (resultException != null) { + if (resultException instanceof SQLException) { + resultException.fillInStackTrace(); + throw (SQLException) resultException; + } else { + throw new PSQLException( + GT.tr( + "Something unusual has occured to cause the driver to fail. Please report" + + " this exception."), + PSQLState.UNEXPECTED_ERROR, resultException); + } + } + + long delay = expiry - System.currentTimeMillis(); + if (delay <= 0) { + abandoned = true; + throw new PSQLException(GT.tr("Connection attempt timed out."), + PSQLState.CONNECTION_UNABLE_TO_CONNECT); + } - // Save the ip and port configured in the url - String[] addresses = l_urlServer.substring(0, slash).split(","); - StringBuilder hosts = new StringBuilder(); - StringBuilder ports = new StringBuilder(); - for (String address : addresses) { - int portIdx = address.lastIndexOf(':'); - if (portIdx != -1 && address.lastIndexOf(']') < portIdx) { - String portStr = address.substring(portIdx + 1); try { - int port = Integer.parseInt(portStr); - if (port < 1 || port > 65535) { - LOGGER.warn("JDBC URL port: " + portStr + " not valid (1:65535) "); + wait(delay); + } catch (InterruptedException ie) { + + // reset the interrupt flag + Thread.currentThread().interrupt(); + abandoned = true; + + // throw an unchecked exception which will hopefully not be ignored by the calling code + throw new RuntimeException(GT.tr("Interrupted while attempting to connect.")); + } + } + } + } + + private final String url; + private final Properties props; + private Connection result; + private Throwable resultException; + private boolean abandoned; + } + + /** + * Create a connection from URL and properties. Always does the connection work in the current + * thread without enforcing a timeout, regardless of any timeout specified in the properties. + * + * @param url the original URL + * @param props the parsed/defaulted connection properties + * @return a new connection + * @throws SQLException if the connection could not be made + */ + private static Connection makeConnection(String url, Properties props) throws SQLException { + PgConnection pgConnection = new PgConnection(hostSpecs(props), user(props), database(props), props, url); + GlobalConnectionTracker.possessConnectionReference(pgConnection.getQueryExecutor(), props); + return pgConnection; + } + + /** + * Returns true if the driver thinks it can open a connection to the given URL. Typically, drivers + * will return true if they understand the subprotocol specified in the URL and false if they + * don't. Our protocols start with jdbc:postgresql: + * + * @param url the URL of the driver + * @return true if this driver accepts the given URL + * @throws PSQLException + * @see java.sql.Driver#acceptsURL + */ + @Override + public boolean acceptsURL(String url) throws PSQLException { + return parseURL(url, null) != null; + } + + /** + *The getPropertyInfo method is intended to allow a generic GUI tool to discover what properties + * it should prompt a human for in order to get enough information to connect to a database.
+ * + *Note that depending on the values the human has supplied so far, additional values may become + * necessary, so it may be necessary to iterate through several calls to getPropertyInfo
+ * + * @param url the Url of the database to connect to + * @param info a proposed list of tag/value pairs that will be sent on connect open. + * @return An array of DriverPropertyInfo objects describing possible properties. This array may + * be an empty array if no properties are required + * @throws PSQLException + * @see java.sql.Driver#getPropertyInfo + */ + @Override + public DriverPropertyInfo[] getPropertyInfo(String url, Properties info) throws PSQLException { + Properties copy = new Properties(info); + Properties parse = parseURL(url, copy); + if (parse != null) { + copy = parse; + } + + PGProperty[] knownProperties = PGProperty.values(); + DriverPropertyInfo[] props = new DriverPropertyInfo[knownProperties.length]; + for (int i = 0; i < props.length; ++i) { + props[i] = knownProperties[i].toDriverPropertyInfo(copy); + } + + return props; + } + + @Override + public int getMajorVersion() { + return org.postgresql.util.DriverInfo.MAJOR_VERSION; + } + + @Override + public int getMinorVersion() { + return org.postgresql.util.DriverInfo.MINOR_VERSION; + } + + /** + * Returns the server version series of this driver and the specific build number. + * + * @return JDBC driver version + * @deprecated use {@link #getMajorVersion()} and {@link #getMinorVersion()} instead + */ + @Deprecated + public static String getVersion() { + return DriverInfo.DRIVER_FULL_NAME; + } + + /** + *Report whether the driver is a genuine JDBC compliant driver. A driver may only report "true" + * here if it passes the JDBC compliance tests, otherwise it is required to return false. JDBC + * compliance requires full support for the JDBC API and full support for SQL 92 Entry Level.
+ * + *For PostgreSQL, this is not yet possible, as we are not SQL92 compliant (yet).
+ */ + @Override + public boolean jdbcCompliant() { + return false; + } + + public static String parseIPValid(String address) { + if (address == "" || address == null) { + return address; + } + String regex = "^(1\\d{2}|2[0-4]\\d|25[0-5]|[1-9]\\d|[1-9])\\." + + "(1\\d{2}|2[0-4]\\d|25[0-5]|[1-9]\\d|\\d)\\." + + "(1\\d{2}|2[0-4]\\d|25[0-5]|[1-9]\\d|\\d)\\." + + "(1\\d{2}|2[0-4]\\d|25[0-5]|[1-9]\\d|\\d)$"; + Pattern pattern = Pattern.compile(regex); + if (!pattern.matcher(address).matches()) { + LOGGER.debug("JDBC URL invalid ip address: " + address); + } + return address; + } + + /** + * Constructs a new DriverURL, splitting the specified URL into its component parts. + * + * @param url JDBC URL to parse + * @param defaults Default properties + * @return Properties with elements added from the url + * @throws PSQLException + */ + public static Properties parseURL(String url, Properties defaults) throws PSQLException { + Properties urlProps = new Properties(defaults); + String l_urlServer = url; + String l_urlArgs = ""; + boolean isUrlLegal = false; + String[] legalUrlTags = {"jdbc:postgresql:", "jdbc:dws:iam:"}; + + int l_qPos = url.indexOf('?'); + if (l_qPos != -1) { + l_urlServer = url.substring(0, l_qPos); + l_urlArgs = url.substring(l_qPos + 1); + } + + for (String urlTag : legalUrlTags) { + if (l_urlServer.startsWith(urlTag)) { + isUrlLegal = true; + } + } + + if (!isUrlLegal) { + LOGGER.debug("JDBC URL must start with \"jdbc:postgresql:\" or \"jdbc:dws:iam:\" but was: " + + filterAuthenticationCredentials(url)); + return null; + } + + //if (l_urlServer.startsWith("jdbc:postgresql:")) { + l_urlServer = l_urlServer.substring("jdbc:postgresql:".length()); + + if (l_urlServer.startsWith("//")) { + l_urlServer = l_urlServer.substring(2); + int slash = l_urlServer.indexOf('/'); + if (slash == -1) { + LOGGER.warn("JDBC URL must contain a / at the end of the host or port: " + + filterAuthenticationCredentials(url)); + return null; + } + urlProps.setProperty("PGDBNAME", URLCoder.decode(l_urlServer.substring(slash + 1))); + + // Save the ip and port configured in the url + String[] addresses = l_urlServer.substring(0, slash).split(","); + StringBuilder hosts = new StringBuilder(); + StringBuilder ports = new StringBuilder(); + for (String address : addresses) { + int portIdx = address.lastIndexOf(':'); + if (portIdx != -1 && address.lastIndexOf(']') < portIdx) { + String portStr = address.substring(portIdx + 1); + try { + int port = Integer.parseInt(portStr); + if (port < 1 || port > 65535) { + LOGGER.warn("JDBC URL port: " + portStr + " not valid (1:65535) "); + return null; + } + } catch (NumberFormatException ignore) { + LOGGER.warn("JDBC URL invalid port number: " + portStr); return null; } - } catch (NumberFormatException ignore) { - LOGGER.warn("JDBC URL invalid port number: " + portStr); - return null; + ports.append(portStr); + hosts.append(parseIPValid((String) address.subSequence(0, portIdx))); + } else { + ports.append(DEFAULT_PORT); + hosts.append(parseIPValid(address)); } - ports.append(portStr); - hosts.append(parseIPValid((String) address.subSequence(0, portIdx))); - } else { - ports.append(DEFAULT_PORT); - hosts.append(parseIPValid(address)); + ports.append(','); + hosts.append(','); } - ports.append(','); - hosts.append(','); - } - ports.setLength(ports.length() - 1); - hosts.setLength(hosts.length() - 1); + ports.setLength(ports.length() - 1); + hosts.setLength(hosts.length() - 1); - urlProps.setProperty("PGHOST", hosts.toString()); - urlProps.setProperty("PGPORT", ports.toString()); + urlProps.setProperty("PGHOST", hosts.toString()); + urlProps.setProperty("PGPORT", ports.toString()); - //The first connection, put the host and port in the url into the props - urlProps.setProperty("PGHOSTURL", hosts.toString()); - urlProps.setProperty("PGPORTURL", ports.toString()); - } else { + //The first connection, put the host and port in the url into the props + urlProps.setProperty("PGHOSTURL", hosts.toString()); + urlProps.setProperty("PGPORTURL", ports.toString()); + } else { /* if there are no defaults set or any one of PORT, HOST, DBNAME not set then set it to default */ - if (defaults == null || !defaults.containsKey("PGPORT")) { - urlProps.setProperty("PGPORT", DEFAULT_PORT); + if (defaults == null || !defaults.containsKey("PGPORT")) { + urlProps.setProperty("PGPORT", DEFAULT_PORT); + } + if (defaults == null || !defaults.containsKey("PGHOST")) { + urlProps.setProperty("PGHOST", "localhost"); + } + if (defaults == null || !defaults.containsKey("PGDBNAME")) { + urlProps.setProperty("PGDBNAME", URLCoder.decode(l_urlServer)); + } } - if (defaults == null || !defaults.containsKey("PGHOST")) { - urlProps.setProperty("PGHOST", "localhost"); + + // parse the args part of the url + urlProps.putAll(praseParam(l_urlArgs)); + if (urlProps.getProperty("enable_ce") != null && urlProps.getProperty("enable_ce").equals("1")) { + urlProps.setProperty("CLIENTLOGIC", "1"); } - if (defaults == null || !defaults.containsKey("PGDBNAME")) { - urlProps.setProperty("PGDBNAME", URLCoder.decode(l_urlServer)); + return urlProps; + } + + /** + * Shield sensitive information in the connection string, such as password, sslpassword, iamPassword + * + * @param url Connection string. + * @return Connection string without sensitive information + */ + private static String filterAuthenticationCredentials(String url) { + String notSensitiveUrl = url; + int paramStartIndex = notSensitiveUrl.indexOf("?"); + if (paramStartIndex != -1) { + LinkedHashMapindex is out of range
*/
+
+ void setBlob(int index, InputStream stream) throws SQLException;
+
void setText(int index, InputStream stream) throws SQLException;
/**
@@ -242,4 +245,6 @@ public interface ParameterList {
* @return Object array containing the parameter values.
*/
Object[] getValues();
+
+ void bindRegisterOutParameter(int index,int oid, boolean isOracleCompatibilityFunction) throws SQLException;
}
diff --git a/pgjdbc/src/main/java/org/postgresql/core/Parser.java b/pgjdbc/src/main/java/org/postgresql/core/Parser.java
index ffe0818..368ac96 100644
--- a/pgjdbc/src/main/java/org/postgresql/core/Parser.java
+++ b/pgjdbc/src/main/java/org/postgresql/core/Parser.java
@@ -48,8 +48,6 @@ public class Parser {
boolean isBatchedReWriteConfigured,
String... returningColumnNames) throws SQLException {
int numOfOverSymble = 0;
- boolean haveProcedure = false;
- boolean haveFunction = false;
if(startWithComment(query)) {
query = removeFirstComment(query);
}
@@ -57,15 +55,7 @@ public class Parser {
String queryTemp = query.trim();
queryTemp = queryTemp.replaceAll(reg, "\0");
String[] queryArr = queryTemp.split("\0");
-
- for(int i = 0;i < queryArr.length; i++) {
- if (queryArr[i] != null && (queryArr[i].toUpperCase(Locale.ENGLISH)).equals("PROCEDURE") ) {
- haveProcedure = true;
- }
- if (queryArr[i] != null && (queryArr[i].toUpperCase(Locale.ENGLISH)).equals("FUNCTION") ) {
- haveFunction = true;
- }
- }
+ boolean haveSpecialKeyword = isContainSpecialKeyword(queryArr);
if (!withParameters && !splitStatements
&& returningColumnNames != null && returningColumnNames.length == 0) {
@@ -135,7 +125,11 @@ public class Parser {
{
break;
}
- if (i >1 && "\n".equals(String.valueOf(aChars[i - 1]))) {
+ if (!haveSpecialKeyword) {
+ break;
+ }
+
+ if (i > 1 && ("\n".equals(String.valueOf(aChars[i - 1])) || " ".equals(String.valueOf(aChars[i - 1])))) {
numOfOverSymble++;
if (inParen == 0) {
if (!whitespaceOnly) {
@@ -165,6 +159,7 @@ public class Parser {
valuesBraceClosePosition = -1;
}
+ inBeginEnd = 0;
nativeQueries.add(
new NativeQuery(
nativeSql.toString(),
@@ -236,11 +231,7 @@ public class Parser {
break;
case ';':
- if (haveProcedure || haveFunction)
- {
- break;
- }
- if (queryArr[0] !=null && ((queryArr[0].toUpperCase(Locale.ENGLISH)).equals("BEGIN") ||(queryArr[0].toUpperCase(Locale.ENGLISH)).equals("DECLARE")))
+ if (haveSpecialKeyword)
{
break;
}
@@ -268,7 +259,7 @@ public class Parser {
valuesBraceOpenPosition = -1;
valuesBraceClosePosition = -1;
}
-
+ inBeginEnd = 0;
nativeQueries.add(new NativeQuery(nativeSql.toString(),
toIntArray(bindPositions), false,
SqlCommand.createStatementTypeInfo(
@@ -304,7 +295,7 @@ public class Parser {
&& "N".equalsIgnoreCase(String.valueOf(aChars[i + 4]))
&& isSpecialCharacters(aChars[i - 1])
&& isSpecialCharacters(aChars[i + 5])) {
- inBeginEnd ++;
+ inBeginEnd++;
}
}
break;
@@ -315,7 +306,9 @@ public class Parser {
if ("N".equalsIgnoreCase(String.valueOf(aChars[i + 1]))
&& "D".equalsIgnoreCase(String.valueOf(aChars[i + 2])))
{
- inBeginEnd --;
+ int[] result = parseEnd(i, aChars, inBeginEnd);
+ i = result[0];
+ inBeginEnd = result[1];
}
}
@@ -615,6 +608,48 @@ public class Parser {
return query.length;
}
+ /**
+ * Judge whether the statement contains the keywords procedure, function, create, package and declare
+ *
+ * @param queryArr An array of strings consisting of SQL statements
+ * @return boolean
+ */
+ private static boolean isContainSpecialKeyword(final String[] queryArr) {
+ if (queryArr[0].toUpperCase(Locale.ENGLISH).equals("BEGIN")) {
+ return true;
+ }
+ boolean haveCreate = false;
+ boolean havePackage = false;
+ for (int i = 0; i < queryArr.length; i++) {
+ if (queryArr[i] == null) {
+ continue;
+ }
+ switch (queryArr[i].toUpperCase(Locale.ENGLISH)) {
+ case "PROCEDURE":
+ case "FUNCTION":
+ case "DECLARE":
+ return true;
+ case "CREATE":
+ if (i == 0) {
+ haveCreate = true;
+ }
+ break;
+ case "PACKAGE":
+ havePackage = true;
+ break;
+ case "END":
+ if (haveCreate && havePackage) {
+ // Case : create package tn end is end tn;
+ return true;
+ }
+ break;
+ default:
+ break;
+ }
+ }
+ return false;
+ }
+
/**
* Find the end of the double-quoted string starting at the given offset.
* @@ -1092,6 +1127,7 @@ public class Parser { String sql = jdbcSql; boolean isFunction = false; boolean outParamBeforeFunc = false; + boolean isOracleCompatibilityFunction = false; int len = jdbcSql.length(); int state = 1; @@ -1122,6 +1158,7 @@ public class Parser { if (ch == '?') { outParamBeforeFunc = isFunction = true; // { ? = call ... } -- function with one out parameter + isOracleCompatibilityFunction = true; ++i; ++state; } else if (ch == 'c' || ch == 'C') { // { call ... } -- proc with no out parameters @@ -1223,7 +1260,7 @@ public class Parser { if (i == len && !syntaxError) { if (state == 1) { // Not an escaped syntax. - return new JdbcCallParseInfo(sql, isFunction); + return new JdbcCallParseInfo(sql, isFunction, false); } if (state != 8) { syntaxError = true; // Ran out of query while still parsing @@ -1277,7 +1314,7 @@ public class Parser { } sql = sb.append(suffix).toString(); - return new JdbcCallParseInfo(sql, isFunction); + return new JdbcCallParseInfo(sql, isFunction, isOracleCompatibilityFunction); } /** @@ -1475,6 +1512,47 @@ public class Parser { return i; } + /** + * Judge whether the current end is the end paired with begin + */ + private static int[] parseEnd(final int offset, final char[] query, final int inBeginEnd) { + int tempOffset = offset; + int tempInBeginEnd = inBeginEnd; + tempOffset = tempOffset + 3; + while (tempOffset < query.length) { + if (isSpecialCharacters(query[tempOffset])) { + tempOffset++; + } else if ("-".equals(String.valueOf(query[tempOffset]))) { + int temp = tempOffset; + tempOffset = Parser.parseLineComment(query, tempOffset); + if (temp == tempOffset) { + tempOffset++; + } + } else if ("/".equals(String.valueOf(query[tempOffset]))) { + int temp = tempOffset; + temp = Parser.parseBlockComment(query, temp); + if (tempOffset != temp) { + tempOffset = temp + 1; + } else { + // Not a slash for a comment + tempOffset--; + tempInBeginEnd--; + break; + } + } else if (";".equals(String.valueOf(query[tempOffset])) + || (tempOffset + 1 < query.length && "$$".equals( + String.valueOf(query[tempOffset]) + query[tempOffset + 1]))) { + // This is the end paired with begin + tempInBeginEnd--; + break; + } else { + tempOffset--; + break; + } + } + return new int[]{tempOffset, tempInBeginEnd}; + } + /** * Generate sql for escaped functions. * diff --git a/pgjdbc/src/main/java/org/postgresql/core/Query.java b/pgjdbc/src/main/java/org/postgresql/core/Query.java index eeda704..fe0cef9 100644 --- a/pgjdbc/src/main/java/org/postgresql/core/Query.java +++ b/pgjdbc/src/main/java/org/postgresql/core/Query.java @@ -90,4 +90,34 @@ public interface Query { * single-statement query. */ Query[] getSubqueries(); + + /** + * Set the current statement state + * + * @param isFunction true or false + */ + void setIsFunction(boolean isFunction); + + /** + * Returns whether the current statement is a function + * + * @return true or false + * + */ + boolean getIsFunction(); + + /** + * Gets the value of nodeName + * + * @return String + */ + String getNodeName(); + + /** + * Set the value of nodeName + * + * @param nodeName DN name + * @return null + */ + void setNodeName(String nodeName); } diff --git a/pgjdbc/src/main/java/org/postgresql/core/QueryExecutor.java b/pgjdbc/src/main/java/org/postgresql/core/QueryExecutor.java index e02e9a2..794f248 100644 --- a/pgjdbc/src/main/java/org/postgresql/core/QueryExecutor.java +++ b/pgjdbc/src/main/java/org/postgresql/core/QueryExecutor.java @@ -123,6 +123,11 @@ public interface QueryExecutor extends TypeTransferModeRegistry { * Added to support queries coming from client logic (c++) and avoid a loop back */ int QUERY_EXECUTE_BYPASS_CLIENT_LOGIC = 2048; + + /** + * Execute the query again after client logic cache was reloaded + */ + int QUERY_RETRY_WITH_CLIENT_LOGIC_CACHE_RELOADS = 4096; /** * Execute several Query, passing results to a provided ResultHandler. @@ -495,5 +500,38 @@ public interface QueryExecutor extends TypeTransferModeRegistry { void setGaussdbVersion(String gaussdbVersion); + + /** + * Sets the version of the currently connected database + * + * @param workingVersionNum value of workingVersionNum + */ + void setWorkingVersionNum(String workingVersionNum); + + /** + * Get workingVersionNum + * + * @return value of workingVersionNum + */ + String getWorkingVersionNum(); + void setAvailability(boolean availability); + + String getCompatibilityMode(); + + void setCompatibilityMode(String compatibilityMode); + + /** + * Get reload status + * + * @return state true or false + */ + boolean getEnableOutparamOveride(); + + /** + * Set reload status + * + * @param enableOutparamOveride true or false + */ + void setEnableOutparamOveride(boolean enableOutparamOveride); } diff --git a/pgjdbc/src/main/java/org/postgresql/core/SocketFactoryFactory.java b/pgjdbc/src/main/java/org/postgresql/core/SocketFactoryFactory.java index 09efa75..fe56354 100644 --- a/pgjdbc/src/main/java/org/postgresql/core/SocketFactoryFactory.java +++ b/pgjdbc/src/main/java/org/postgresql/core/SocketFactoryFactory.java @@ -36,7 +36,7 @@ public class SocketFactoryFactory { return SocketFactory.getDefault(); } try { - return (SocketFactory) ObjectFactory.instantiate(socketFactoryClassName, info, true, + return ObjectFactory.instantiate(SocketFactory.class, socketFactoryClassName, info, true, PGProperty.SOCKET_FACTORY_ARG.get(info)); } catch (Exception e) { throw new PSQLException( @@ -61,7 +61,7 @@ public class SocketFactoryFactory { return new LibPQFactory(info); } try { - return (SSLSocketFactory) ObjectFactory.instantiate(classname, info, true, + return ObjectFactory.instantiate(SSLSocketFactory.class, classname, info, true, PGProperty.SSL_FACTORY_ARG.get(info)); } catch (Exception e) { throw new PSQLException( diff --git a/pgjdbc/src/main/java/org/postgresql/core/Utils.java b/pgjdbc/src/main/java/org/postgresql/core/Utils.java index 027c71a..dfbb31c 100644 --- a/pgjdbc/src/main/java/org/postgresql/core/Utils.java +++ b/pgjdbc/src/main/java/org/postgresql/core/Utils.java @@ -51,12 +51,7 @@ public class Utils { // for performance measurements. // In OracleJDK 6u65, 7u55, and 8u40 String.getBytes(Charset) is // 3 times faster than other JDK approaches. - Charset gbkCharset = Charset.forName("GBK"); - if ("GBK".equals((ConnectionFactoryImpl.CLIENT_ENCODING).toUpperCase(Locale.ENGLISH))) - { - return str.getBytes(gbkCharset); - } - return str.getBytes(utf8Charset); + return str.getBytes(Charset.forName(ConnectionFactoryImpl.CLIENT_ENCODING)); } /** diff --git a/pgjdbc/src/main/java/org/postgresql/core/v3/CompositeParameterList.java b/pgjdbc/src/main/java/org/postgresql/core/v3/CompositeParameterList.java index f5e1948..58135b9 100644 --- a/pgjdbc/src/main/java/org/postgresql/core/v3/CompositeParameterList.java +++ b/pgjdbc/src/main/java/org/postgresql/core/v3/CompositeParameterList.java @@ -47,6 +47,11 @@ class CompositeParameterList implements V3ParameterList { } + @Override + public void bindRegisterOutParameter(int index, int oid, boolean isOracleCompatibilityFunction) throws SQLException { + + } + public int getDirection(int i) { return 0; } @@ -208,5 +213,8 @@ class CompositeParameterList implements V3ParameterList { int sub = findSubParam(index); subparams[sub].setBlob(index - offsets[sub], stream, length); } - + public void setBlob(int index, InputStream stream) throws SQLException { + int sub = findSubParam(index); + subparams[sub].setBlob(index - offsets[sub], stream); + } } diff --git a/pgjdbc/src/main/java/org/postgresql/core/v3/CompositeQuery.java b/pgjdbc/src/main/java/org/postgresql/core/v3/CompositeQuery.java index 77809e4..62ae348 100644 --- a/pgjdbc/src/main/java/org/postgresql/core/v3/CompositeQuery.java +++ b/pgjdbc/src/main/java/org/postgresql/core/v3/CompositeQuery.java @@ -76,6 +76,26 @@ class CompositeQuery implements Query { return subqueries; } + @Override + public void setIsFunction(boolean isFunction) { + + } + + @Override + public boolean getIsFunction() { + return false; + } + + @Override + public String getNodeName() { + return ""; + } + + @Override + public void setNodeName(String nodeName) { + + } + public boolean isStatementDescribed() { for (SimpleQuery subquery : subqueries) { if (!subquery.isStatementDescribed()) { diff --git a/pgjdbc/src/main/java/org/postgresql/core/v3/ConnectionFactoryImpl.java b/pgjdbc/src/main/java/org/postgresql/core/v3/ConnectionFactoryImpl.java index 9bd230d..debfa0b 100644 --- a/pgjdbc/src/main/java/org/postgresql/core/v3/ConnectionFactoryImpl.java +++ b/pgjdbc/src/main/java/org/postgresql/core/v3/ConnectionFactoryImpl.java @@ -36,6 +36,7 @@ import java.util.Map; import java.util.Properties; import java.util.TimeZone; import java.util.UUID; +import java.util.concurrent.ConcurrentHashMap; import javax.net.SocketFactory; @@ -62,16 +63,31 @@ public class ConnectionFactoryImpl extends ConnectionFactory { public static String USE_BOOLEAN = "false"; private static final int AUTH_REQ_SHA256 = 10; private static final int AUTH_REQ_MD5_SHA256 = 11; + private static final int AUTH_REQ_SHA256_RFC = 12; private static final int AUTH_REQ_SM3 = 13; private static final int PLAIN_PASSWORD = 0; private static final int MD5_PASSWORD = 1; private static final int SHA256_PASSWORD = 2; private static final int SM3_PASSWORD = 3; private static final int ERROR_PASSWORD = 4; + private static final int SHA256_PASSWORD_RFC = 6; private static final int PROTOCOL_VERSION_351 = 351; private static final int PROTOCOL_VERSION_350 = 350; + private static final int PROTOCOL_VERSION_1 = 1; private int protocolVerion = PROTOCOL_VERSION_351; private String connectInfo = ""; + + /** + * Whitelist of supported client_encoding + */ + public static final HashMapSupplement to synchronization of public methods on current QueryExecutor.
* @@ -1507,11 +1554,29 @@ public class QueryExecutorImpl extends QueryExecutorBase { pendingDescribePortalQueue.add(sync); } + private void sendTraceId() throws IOException { + String traceId; + if ((traceId = Driver.getTracer()) == null) { + return; + } + if (LOGGER.isTraceEnabled()) { + LOGGER.trace("[" + socketAddress + "] " + "trace :" + traceId); + } + byte[] encodedTraceId = Utils.encodeUTF8(traceId); + int encodedSize = 4 + + encodedTraceId.length + 1; + pgStream.sendChar('J'); + pgStream.sendInteger4(encodedSize); + pgStream.send(encodedTraceId); + pgStream.sendChar(0); + } + private void sendParse(SimpleQuery query, SimpleParameterList params, boolean oneShot) throws IOException { // Already parsed, or we have a Parse pending and the types are right? int[] typeOIDs = params.getTypeOIDs(); - if (query.isPreparedFor(typeOIDs, deallocateEpoch)) { + if ((query.getNodeName() == null || query.getNodeName().isEmpty()) + && query.isPreparedFor(typeOIDs, deallocateEpoch)) { return; } @@ -1541,6 +1606,35 @@ public class QueryExecutorImpl extends QueryExecutorBase { byte[] encodedStatementName = query.getEncodedStatementName(); String nativeSql = query.getNativeSql(); + if (query.getNodeName() != null && !query.getNodeName().isEmpty()) { + try { + nativeSql = HintNodeName.addNodeName(nativeSql, query.getNodeName(), this); + } catch (SQLException e) { + throw new IOException(e.getMessage()); + } + } + + boolean sendFunctionParamType = isSendFunctionParamType(query); + char[] paramFlags = new char[params.getFlags().length]; + + if (sendFunctionParamType) { + byte[] flags = params.getFlags(); + for (int j = 0; j < flags.length; j++) { + switch (flags[j]) { + case 1: + paramFlags[j] = 'i'; + break; + case 2: + paramFlags[j] = 'o'; + break; + case 3: + paramFlags[j] = 'b'; + break; + default: + paramFlags[j] = 'i'; + } + } + } if (LOGGER.isTraceEnabled()) { StringBuilder sbuf = new StringBuilder(" FE=> Parse(stmt=" + statementName + ",query=\""); sbuf.append(nativeSql); @@ -1551,6 +1645,15 @@ public class QueryExecutorImpl extends QueryExecutorBase { } sbuf.append(params.getTypeOID(i)); } + if (sendFunctionParamType) { + sbuf.append("}\",flags={"); + for (int i = 1; i <= paramFlags.length; ++i) { + if (i != 1) { + sbuf.append(","); + } + sbuf.append(paramFlags[i - 1]); + } + } sbuf.append("})"); LOGGER.trace("[" + socketAddress + "] " + sbuf.toString()); } @@ -1565,10 +1668,15 @@ public class QueryExecutorImpl extends QueryExecutorBase { // + N + 1 (statement name, zero-terminated) // + N + 1 (query, zero terminated) // + 2 (parameter count) + N * 4 (parameter types) + // + 2 or 0 (if overload + 2 else + 0) + // + params.getFlags().length or 0 (if oracle compatibilityMode & overload & sql is function or procedure send + // length else nosend) int encodedSize = 4 + (encodedStatementName == null ? 0 : encodedStatementName.length) + 1 + queryUtf8.length + 1 - + 2 + 4 * params.getParameterCount(); + + 2 + 4 * params.getParameterCount() + + (this.isDescribeOutparam() ? 2 : 0) + + (sendFunctionParamType ? params.getFlags().length : 0); pgStream.sendChar('P'); // Parse pgStream.sendInteger4(encodedSize); @@ -1582,7 +1690,16 @@ public class QueryExecutorImpl extends QueryExecutorBase { for (int i = 1; i <= params.getParameterCount(); ++i) { pgStream.sendInteger4(params.getTypeOID(i)); } - + if (this.isDescribeOutparam()) { + if (sendFunctionParamType) { + pgStream.sendInteger2(params.getParameterCount()); + for (int i = 1; i <= paramFlags.length; ++i) { + pgStream.sendChar(paramFlags[i - 1]); + } + } else { + pgStream.sendInteger2(0); + } + } pendingParseQueue.add(query); } @@ -2059,6 +2176,7 @@ public class QueryExecutorImpl extends QueryExecutorBase { rows = fetchSize; // maxRows > fetchSize } + sendTraceId(); sendParse(query, params, oneShot); // Must do this after sendParse to pick up any changes to the @@ -2160,6 +2278,7 @@ public class QueryExecutorImpl extends QueryExecutorBase { V3ParameterList parameters = (V3ParameterList) parameterLists[0]; SimpleParameterList params = (SimpleParameterList) parameters; + sendTraceId(); sendParse(query, params, oneShot); // Must do this after sendParse to pick up any changes to the @@ -3039,6 +3158,12 @@ public class QueryExecutorImpl extends QueryExecutorBase { throw new PSQLException(GT.tr("Protocol error. Session setup failed."), PSQLState.PROTOCOL_VIOLATION); } + } else if ("behavior_compat_options".equals(name)) { + if (value != null && value.contains("proc_outparam_override")) { + setEnableOutparamOveride(true); + } else { + setEnableOutparamOveride(false); + } } } @@ -3099,6 +3224,23 @@ public class QueryExecutorImpl extends QueryExecutorBase { useBinarySendForOids.addAll(oids); } + /** + * In oracle compatibility mode, turn on output parameter overload (enable_outparam_overide=true) and + * sql is a function or a stored procedure. At this time, need to send the type of function and stored + * procedure parameters (in or out) + * + * @param query Current SQL + * @return true or false + */ + private boolean isSendFunctionParamType(Query query) { + if (("ORA".equals(this.getCompatibilityMode()) || "A".equals(this.getCompatibilityMode())) + && this.getEnableOutparamOveride() && query.getIsFunction()) { + return true; + } else { + return false; + } + } + private void setIntegerDateTimes(boolean state) { integerDateTimes = state; } diff --git a/pgjdbc/src/main/java/org/postgresql/core/v3/SimpleParameterList.java b/pgjdbc/src/main/java/org/postgresql/core/v3/SimpleParameterList.java index 12bdd83..3c022ec 100644 --- a/pgjdbc/src/main/java/org/postgresql/core/v3/SimpleParameterList.java +++ b/pgjdbc/src/main/java/org/postgresql/core/v3/SimpleParameterList.java @@ -49,6 +49,8 @@ class SimpleParameterList implements V3ParameterList { this.encoded = new byte[paramCount][]; this.flags = new byte[paramCount]; this.transferModeRegistry = transferModeRegistry; + this.compatibilityModes = new String[paramCount]; + this.isOracleCompatibilityFunctions = new boolean[paramCount]; } @Override @@ -63,6 +65,19 @@ class SimpleParameterList implements V3ParameterList { flags[index - 1] |= OUT; } + @Override + public void bindRegisterOutParameter(int index, int oid, boolean isOracleCompatibilityFunction) throws SQLException { + if (index < 1 || index > paramValues.length) { + throw new PSQLException( + GT.tr("The column index is out of range: {0}, number of columns: {1}.", + index, paramValues.length), + PSQLState.INVALID_PARAMETER_VALUE); + } + paramTypes[index - 1] = oid; + compatibilityModes[index - 1] = "ORA"; + isOracleCompatibilityFunctions[index - 1] = isOracleCompatibilityFunction; + } + private void bind(int index, Object value, int oid, byte binary) throws SQLException { if (index < 1 || index > paramValues.length) { throw new PSQLException( @@ -181,18 +196,11 @@ class SimpleParameterList implements V3ParameterList { } public void setBlob(int index, InputStream stream, int length) throws SQLException { - try { - int i = Math.min(stream.available(), length); - byte[] tmp = new byte[i]; - int len = stream.read(tmp); - // In the condition of empty Blob. Like byte[] b = {}; new ByteArrayInputStream(b); - if(len == -1){ - LOGGER.trace("Failed to read the inputstream:", new SQLException("Failed to read the inputstream")); - } - setBlob(index, tmp, 0, tmp.length); - } catch (IOException e) { - throw new SQLException(e.getMessage()); - } + bind(index, new StreamWrapper(stream, length), Oid.BLOB, BINARY); + } + + public void setBlob(int index, InputStream stream) throws SQLException { + bind(index, new StreamWrapper(stream), Oid.BLOB, BINARY); } @Override @@ -314,8 +322,16 @@ class SimpleParameterList implements V3ParameterList { public void convertFunctionOutParameters() { for (int i = 0; i < paramTypes.length; ++i) { if (direction(i) == OUT) { - paramTypes[i] = Oid.VOID; - paramValues[i] = "null"; + if(compatibilityModes[i] != null && compatibilityModes[i].equalsIgnoreCase("ORA")){ + // function return value as void. + if (isOracleCompatibilityFunctions[i] == true && i == 0) { + paramTypes[i] = Oid.VOID; + } + paramValues[i] = "null"; + }else{ + paramTypes[i] = Oid.VOID; + paramValues[i] = "null"; + } } } } @@ -514,6 +530,8 @@ class SimpleParameterList implements V3ParameterList { private final byte[] flags; private final byte[][] encoded; private final TypeTransferModeRegistry transferModeRegistry; + private final boolean[] isOracleCompatibilityFunctions; + private final String[] compatibilityModes; /** * Marker object representing NULL; this distinguishes "parameter never set" from "parameter set diff --git a/pgjdbc/src/main/java/org/postgresql/core/v3/SimpleQuery.java b/pgjdbc/src/main/java/org/postgresql/core/v3/SimpleQuery.java index 9d464f1..66161ed 100644 --- a/pgjdbc/src/main/java/org/postgresql/core/v3/SimpleQuery.java +++ b/pgjdbc/src/main/java/org/postgresql/core/v3/SimpleQuery.java @@ -31,6 +31,7 @@ import java.util.Map; */ class SimpleQuery implements Query { private static Log LOGGER = Logger.getLogger(SimpleQuery.class.getName()); + private String nodeName = null; SimpleQuery(SimpleQuery src) { this(src.nativeQuery, src.transferModeRegistry, src.sanitiserDisabled); @@ -67,6 +68,16 @@ class SimpleQuery implements Query { return null; } + @Override + public void setIsFunction(boolean isFunction) { + this.isFunction = isFunction; + } + + @Override + public boolean getIsFunction() { + return this.isFunction; + } + /** *Return maximum size in bytes that each result row from this query may return. Mainly used for * batches that return results.
@@ -327,6 +338,14 @@ class SimpleQuery implements Query { return nativeQuery.bindPositions.length * getBatchSize(); } + public String getNodeName() { + return this.nodeName; + } + + public void setNodeName(String nodeName) { + this.nodeName = nodeName; + } + private Map