src.main.java.org.kawanfw.sql.tomcat.TomcatStarterUtil Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of aceql-http Show documentation
Show all versions of aceql-http Show documentation
AceQL HTTP is a framework of REST like http APIs that allow to access to remote SQL databases over http from any device that supports http.
AceQL HTTP is provided with three client SDK:
- The AceQL C# Client SDK allows to wrap the HTTP APIs using Microsoft SQL Server like calls in their code, just like they would for a local database.
- The AceQL Java Client JDBC Driver allows to wrap the HTTP APIs using JDBC calls in their code, just like they would for a local database.
- The AceQL Python Client SDK allows SQL calls to be encoded with standard unmodified DB-API 2.0 syntax
/*
* This file is part of AceQL HTTP.
* AceQL HTTP: SQL Over HTTP
* Copyright (C) 2020, KawanSoft SAS
* (http://www.kawansoft.com). All rights reserved.
*
* AceQL HTTP is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public
* License as published by the Free Software Foundation; either
* version 2.1 of the License, or (at your option) any later version.
*
* AceQL HTTP is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with this library; if not, write to the Free Software
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
* 02110-1301 USA
*
* Any modifications to this file must keep this entire header
* intact.
*/
package org.kawanfw.sql.tomcat;
import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.lang.reflect.Constructor;
import java.net.DatagramSocket;
import java.net.ServerSocket;
import java.sql.Connection;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.HashSet;
import java.util.List;
import java.util.Objects;
import java.util.Properties;
import java.util.Set;
import javax.servlet.Servlet;
import org.apache.catalina.Context;
import org.apache.catalina.Wrapper;
import org.apache.catalina.startup.Tomcat;
import org.apache.commons.lang3.SystemUtils;
import org.apache.tomcat.jdbc.pool.DataSource;
import org.apache.tomcat.jdbc.pool.PoolProperties;
import org.kawanfw.sql.api.server.DatabaseConfigurationException;
import org.kawanfw.sql.servlet.ServerSqlManager;
import org.kawanfw.sql.servlet.connection.RollbackUtil;
import org.kawanfw.sql.tomcat.util.LinkedProperties;
import org.kawanfw.sql.util.SqlTag;
/**
* @author Nicolas de Pomereu
*
* Utility classes called at Tomcat startup
*/
public class TomcatStarterUtil {
/** Universal and clean line separator */
static String CR_LF = System.getProperty("line.separator");
@SuppressWarnings("unused")
private static final String ERROR_MESSAGE = "D" + "b" + " V" + "e" + "n" + "d" + "or" + " is" + " " + "no" + "t"
+ " sup" + "po" + "rt" + "ed" + " in" + " this " + "ver" + "si" + "on " + "fo" + "r Dr" + "iv" + "er: ";
/**
* protected constructor
*/
protected TomcatStarterUtil() {
}
/**
* If the user has created a driverClassName property in the properties file: we
* create a Tomcat JDBC Pool from the properties
*
* @param properties properties extracted from the properties file
* @throws DatabaseConfigurationException
*/
public static void createAndStoreDataSources(Properties properties) throws DatabaseConfigurationException {
if (properties == null) {
throw new IllegalArgumentException("properties is null");
}
Set databases = getDatabaseNames(properties);
for (String database : databases) {
createAndStoreDataSource(properties, database.trim());
}
}
public static void addServlets(Properties properties, Context rootCtx) {
if (properties == null) {
throw new IllegalArgumentException("properties is null");
}
Set servlets = getServlets(properties);
if (servlets.isEmpty()) {
return;
}
System.out.println(SqlTag.SQL_PRODUCT_START + " Loading servlets:");
for (String servlet : servlets) {
String servletClassName = properties.getProperty(servlet + "." + "class");
if (servletClassName == null || servletClassName.isEmpty()) {
throw new IllegalArgumentException(servlet + ".class" + " property not found for servlet " + servlet
+ ". " + SqlTag.PLEASE_CORRECT);
}
servletClassName = servletClassName.trim();
String servletUrl = properties.getProperty(servlet + "." + "url-pattern");
if (servletUrl == null || servletUrl.isEmpty()) {
throw new IllegalArgumentException(servlet + ".url-pattern" + " property not found for servlet "
+ servlet + ". " + SqlTag.PLEASE_CORRECT);
}
servletUrl = servletUrl.trim();
Class c = null;
Servlet servletInstance = null;
try {
c = Class.forName(servletClassName);
// servletInstance = (Servlet) c.newInstance();
Constructor constructor = c.getConstructor();
servletInstance = (Servlet) constructor.newInstance();
} catch (Exception e) {
throw new IllegalArgumentException("Exception when loading " + servletClassName + " (servlet " + servlet
+ "): " + e.toString() + ". " + SqlTag.PLEASE_CORRECT, e);
}
System.out.println(SqlTag.SQL_PRODUCT_START + " -> Servlet " + servlet + " [url-pattern: " + servletUrl
+ "] successfully loaded.");
@SuppressWarnings("unused")
Wrapper wrapper = Tomcat.addServlet(rootCtx, servlet, servletInstance);
rootCtx.addServletMappingDecoded(servletUrl, servlet);
}
}
/**
* Returns the servlets names from the properties
*
* @param properties
* @return servlets names
*/
public static Set getServlets(Properties properties) {
String servlets = properties.getProperty("servlets");
if (servlets == null || servlets.isEmpty()) {
return new HashSet<>();
}
String[] servletArray = servlets.split(",");
Set servletSet = new HashSet<>();
for (int i = 0; i < servletArray.length; i++) {
servletSet.add(servletArray[i].trim());
}
return servletSet;
}
/**
* Returns the database names from the properties
*
* @param properties
* @return the database names
* @throws DatabaseConfigurationException
*/
public static Set getDatabaseNames(Properties properties) throws DatabaseConfigurationException {
if (properties == null) {
throw new IllegalArgumentException("properties is null");
}
String databases = properties.getProperty("databases");
if (databases == null || databases.isEmpty()) {
throw new DatabaseConfigurationException(
"the databases property is not set in properties file. " + SqlTag.PLEASE_CORRECT);
}
String[] databaseArray = databases.split(",");
Set databaseSet = new HashSet<>();
for (int i = 0; i < databaseArray.length; i++) {
databaseSet.add(databaseArray[i].trim());
}
return databaseSet;
}
/**
* If the user has created a driverClassName property in the properties file: we
* create a Tomcat JDBC Pool from the properties
*
* @param properties properties extracted from the properties file
* @param database the database name for which to set the properties
* @throws DatabaseConfigurationException
*/
public static void createAndStoreDataSource(Properties properties, String database)
throws DatabaseConfigurationException {
Objects.requireNonNull(properties, "properties cannot be null!");
Objects.requireNonNull(database, "database cannot be null!");
String driverClassName = properties.getProperty(database + "." + "driverClassName");
String url = properties.getProperty(database + "." + "url");
String username = properties.getProperty(database + "." + "username");
if (driverClassName == null || driverClassName.isEmpty()) {
System.err.println(SqlTag.SQL_PRODUCT_START + " WARNING: driverClassName"
+ " property not found for database " + database + "! ");
System.err.println(SqlTag.SQL_PRODUCT_START
+ " Connection management must be defined in DatabaseConfigurator.getConnection(String database)");
return;
}
checkParameters(properties, database, driverClassName, url, username);
PoolProperties poolProperties = createPoolProperties(properties, database);
DataSource dataSource = new DataSource();
dataSource.setPoolProperties(poolProperties);
Connection connection = null;
try {
System.out.println(
SqlTag.SQL_PRODUCT_START + " Testing DataSource.getConnection() for " + database + " database:");
connection = dataSource.getConnection();
if (connection == null) {
throw new DatabaseConfigurationException(
"Connection is null. Please verify all the values in properties file.");
}
// Future usage: Checks that DB Vendor is supported
System.out.println(SqlTag.SQL_PRODUCT_START + " -> Connection OK!");
} catch (SQLException e) {
RollbackUtil.rollback(connection);
throw new DatabaseConfigurationException(e.getMessage() + " " + e.getCause());
} finally {
if (connection != null) {
try {
connection.close();
} catch (Exception e) {
e.printStackTrace();
}
}
}
TomcatSqlModeStore.setDataSource(database, dataSource);
}
/**
* @param properties
* @param database
* @return
* @throws DatabaseConfigurationException
*/
private static PoolProperties createPoolProperties(Properties properties, String database)
throws DatabaseConfigurationException {
// OK! create and test the DataSource
PoolPropertiesCreator poolPropertiesCreator = new PoolPropertiesCreator(properties, database);
PoolProperties poolProperties = null;
try {
poolProperties = poolPropertiesCreator.create();
} catch (Exception e) {
throw new DatabaseConfigurationException(e.getMessage());
}
return poolProperties;
}
/**
* @param properties
* @param database
* @param driverClassName
* @param url
* @param username
* @throws DatabaseConfigurationException
*/
private static void checkParameters(Properties properties, String database, String driverClassName, String url,
String username) throws DatabaseConfigurationException {
if ((url == null) || url.isEmpty()) {
throw new DatabaseConfigurationException(
"the url property is not set in properties file for driverClassName " + driverClassName + ". "
+ SqlTag.PLEASE_CORRECT);
}
if ((username == null) || username.isEmpty()) {
throw new DatabaseConfigurationException(
"the username property is not set in properties file for driverClassName " + driverClassName + ". "
+ SqlTag.PLEASE_CORRECT);
}
String password = properties.getProperty(database + "." + "password");
if ((password == null) || password.isEmpty()) {
throw new DatabaseConfigurationException(
"the password property is not set in properties file for driverClassName " + driverClassName + ". "
+ SqlTag.PLEASE_CORRECT);
}
System.out.println(
SqlTag.SQL_PRODUCT_START + " Setting Tomcat JDBC Pool attributes for " + database + " database:");
}
/**
* Returns the Properties extracted from a file.
*
* @param file the file containing the properties
* @return the Properties extracted from the file
*
* @throws IOException
* @throws DatabaseConfigurationException
*/
public static Properties getProperties(File file) throws IOException, DatabaseConfigurationException {
if (file == null) {
throw new IllegalArgumentException("file can not be null!");
}
if (!file.exists()) {
throw new DatabaseConfigurationException("properties file not found: " + file);
}
// Get the properties with order of position in file:
Set linkedProperties = LinkedProperties.getLinkedPropertiesName(file);
// Create the ordered properties
Properties properties;
try (InputStream in = new FileInputStream(file);) {
properties = new LinkedProperties(linkedProperties);
properties.load(in);
}
return properties;
}
/**
* Set the servlet parameters store with the values extracted from the
* Properties.
*
* @param properties Properties extracted from the server-sql.properties files
* @throws IllegalArgumentException
*/
public static void setInitParametersInStore(Properties properties) throws IllegalArgumentException {
ServletParametersStore.init(); // Set back to null static values
String aceQLManagerServletCallName = TomcatStarterUtil.getAceQLManagerSevletName(properties);
ServletParametersStore.setServletName(aceQLManagerServletCallName);
Set databases = getDatabaseNames(properties);
ServletParametersStore.setDatabaseNames(databases);
String userAuthenticatorClassName = TomcatStarterUtil
.trimSafe(properties.getProperty(ServerSqlManager.USER_AUTHENTICATOR_CLASS_NAME));
if (userAuthenticatorClassName != null && !userAuthenticatorClassName.isEmpty()) {
ServletParametersStore.setUserAuthenticatorClassName(userAuthenticatorClassName);
}
for (String database : databases) {
// Set the configurator to use for this database
String databaseConfiguratorClassName = TomcatStarterUtil.trimSafe(
properties.getProperty(database + "." + ServerSqlManager.DATABASE_CONFIGURATOR_CLASS_NAME));
if (databaseConfiguratorClassName != null && !databaseConfiguratorClassName.isEmpty()) {
ServletParametersStore.setInitParameter(database, new InitParamNameValuePair(
ServerSqlManager.DATABASE_CONFIGURATOR_CLASS_NAME, databaseConfiguratorClassName));
}
String sqlFirewallClassNameArray = TomcatStarterUtil.trimSafe(
properties.getProperty(database + "." + ServerSqlManager.SQL_FIREWALL_MANAGER_CLASS_NAMES));
if (sqlFirewallClassNameArray != null && !sqlFirewallClassNameArray.isEmpty()) {
List sqlFirewallClassNames = TomcatStarterUtilFirewall.getList(sqlFirewallClassNameArray);
ServletParametersStore.setSqlFirewallClassNames(database, sqlFirewallClassNames);
} else {
ServletParametersStore.setSqlFirewallClassNames(database, new ArrayList());
}
}
String blobDownloadConfiguratorClassName = TomcatStarterUtil
.trimSafe(properties.getProperty(ServerSqlManager.BLOB_DOWNLOAD_CONFIGURATOR_CLASS_NAME));
ServletParametersStore.setBlobDownloadConfiguratorClassName(blobDownloadConfiguratorClassName);
String blobUploadConfiguratorClassName = TomcatStarterUtil
.trimSafe(properties.getProperty(ServerSqlManager.BLOB_UPLOAD_CONFIGURATOR_CLASS_NAME));
ServletParametersStore.setBlobUploadConfiguratorClassName(blobUploadConfiguratorClassName);
String sessionConfiguratorClassName = TomcatStarterUtil
.trimSafe(properties.getProperty(ServerSqlManager.SESSION_CONFIGURATOR_CLASS_NAME));
ServletParametersStore.setSessionConfiguratorClassName(sessionConfiguratorClassName);
String jwtSessionConfiguratorSecretValue = TomcatStarterUtil
.trimSafe(properties.getProperty(ServerSqlManager.JWT_SESSION_CONFIGURATOR_SECRET));
ServletParametersStore.setJwtSessionConfiguratorSecretValue(jwtSessionConfiguratorSecretValue);
}
/**
* Safely trim a String
*
* @param s the String to trim
* @return
*/
public static String trimSafe(final String s) {
String sNew = s;
if (sNew != null) {
sNew = sNew.trim();
}
return sNew;
}
/**
* Checks to see if a specific port is available.
*
* @param port the port to check for availability
*/
public static boolean available(int port) {
ServerSocket ss = null;
DatagramSocket ds = null;
try {
ss = new ServerSocket(port);
ss.setReuseAddress(true);
ds = new DatagramSocket(port);
ds.setReuseAddress(true);
return true;
} catch (IOException e) {
// e.printStackTrace();
} finally {
if (ds != null) {
ds.close();
}
if (ss != null) {
try {
ss.close();
} catch (IOException e) {
/* should not be thrown */
}
}
}
return false;
}
/**
* Returns the Java Info at startup
* @return the Java info
*/
public static String getJavaInfo() {
return SqlTag.SQL_PRODUCT_START + " Java Info: " + SystemUtils.JAVA_VENDOR + " / " + SystemUtils.JAVA_RUNTIME_NAME + " / " + SystemUtils.JAVA_VERSION;
}
static String getAceQLManagerSevletName(Properties properties) {
String aceQLManagerServletCallName = properties.getProperty("aceQLManagerServletCallName");
// Support old name:
if (aceQLManagerServletCallName == null || aceQLManagerServletCallName.isEmpty()) {
aceQLManagerServletCallName = properties.getProperty("serverSqlManagerServletName");
}
if (aceQLManagerServletCallName == null || aceQLManagerServletCallName.isEmpty()) {
throw new DatabaseConfigurationException(
"aceQLManagerServletCallName property is null. " + SqlTag.PLEASE_CORRECT);
}
if (aceQLManagerServletCallName.contains("/")) {
throw new DatabaseConfigurationException(
"aceQLManagerServletCallName property can not contain \"/\" separator. " + SqlTag.PLEASE_CORRECT);
}
aceQLManagerServletCallName = aceQLManagerServletCallName.trim();
return aceQLManagerServletCallName;
}
}