nosi.base.ActiveRecord.HibernateUtils Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of igrp-core Show documentation
Show all versions of igrp-core Show documentation
IGRP Framework is a powerful and highly customizable platform developed by the Operational Nucleus for the Information Society (NOSi) to create web applications, it provides out of box, several modules to make easy to create stand-alone, production-grade web applications: authentication and access-control, business processes automation, reporting, page builder with automatic code generation and incorporation of the Once-Only-Principle, written in Java. IGRP Framework WAR - Contains some keys resources that give UI to IGRP Framework and others supports files.
package nosi.base.ActiveRecord;
import nosi.core.config.ConfigApp;
import nosi.core.config.ConfigCommonMainConstants;
import nosi.core.webapp.Core;
import nosi.core.webapp.Igrp;
import nosi.webapps.igrp.dao.User;
import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;
import org.hibernate.HibernateException;
import org.hibernate.Session;
import org.hibernate.SessionFactory;
import org.hibernate.boot.registry.StandardServiceRegistryBuilder;
import org.hibernate.cfg.Configuration;
import org.hibernate.service.ServiceRegistry;
import java.io.IOException;
import java.io.InputStream;
import java.sql.*;
import java.util.*;
import java.util.concurrent.ConcurrentHashMap;
/**
* Emanuel
* 29 May 2018
*/
public class HibernateUtils {
private static final Logger LOG = LogManager.getLogger(HibernateUtils.class);
private static final Map SESSION_FACTORY = new ConcurrentHashMap<>();
private static final SessionFactory SESSION_FACTORY_IGRP;
public static final StandardServiceRegistryBuilder REGISTRY_BUILDER_IGRP;
public static final String SUFIX_HIBERNATE_CONFIG = ".cfg.xml";
private HibernateUtils() {
throw new IllegalStateException("Utility class");
}
static {
final String cfgName = ConfigApp.getInstance().getBaseConnection() + SUFIX_HIBERNATE_CONFIG;
final Properties properties = getProperties(ConfigApp.getInstance().getBaseConnection() + ".properties");
REGISTRY_BUILDER_IGRP = new StandardServiceRegistryBuilder().applySettings(properties).configure(cfgName);
SESSION_FACTORY_IGRP = buildSessionFactory(cfgName);
}
public static SessionFactory getSessionFactory(String connectionName) {
return getSessionFactory(connectionName, Core.getCurrentDadParam());
}
public static SessionFactory getSessionFactory(String connectionName, String dad) {
if (connectionName != null && connectionName.equalsIgnoreCase(ConfigApp.getInstance().getBaseConnection()))
return SESSION_FACTORY_IGRP;
final String fileName = (dad != null && !dad.isEmpty()) ? connectionName + "." + dad : connectionName;
return SESSION_FACTORY.compute(connectionName, (key, existingSessionFactory) -> {
if (existingSessionFactory == null || !existingSessionFactory.isOpen()) {
return buildSessionFactory(fileName + SUFIX_HIBERNATE_CONFIG);
}
return existingSessionFactory;
});
}
public static Session getSession(String connectionName) {
SessionFactory sessionFactory = getSessionFactory(connectionName);
if (sessionFactory != null) {
if (sessionFactory.isOpen() && sessionFactory.getCurrentSession() != null
&& sessionFactory.getCurrentSession().isOpen()) {
return sessionFactory.getCurrentSession();
}
sessionFactory.close();
removeSessionFactory(connectionName);
sessionFactory = getSessionFactory(connectionName);
if (sessionFactory != null)
return sessionFactory.getCurrentSession();
}
throw new HibernateException(Core.gt("Problema de conexão. Por favor verifica o seu ficheiro hibernate."));
}
private static SessionFactory buildSessionFactory(String cfgName) {
try {
ServiceRegistry serviceRegistry;
if (cfgName.startsWith(ConfigApp.getInstance().getBaseConnection()))
serviceRegistry = REGISTRY_BUILDER_IGRP.build();
else
serviceRegistry = getServiceRegistryBuilder(cfgName).build();
return new Configuration().buildSessionFactory(serviceRegistry);
} catch (Exception ex) {
LOG.error("Initial SessionFactory creation failed.", ex);
throw new ExceptionInInitializerError(ex);
}
}
public static Map getSettings() {
return HibernateUtils.REGISTRY_BUILDER_IGRP.getSettings();
}
private static StandardServiceRegistryBuilder getServiceRegistryBuilder(String cfgName) {
return new StandardServiceRegistryBuilder().applySettings(getProperties(cfgName.replace(SUFIX_HIBERNATE_CONFIG, ".properties"))).configure(cfgName);
}
public static Map getSettings(String cfgName) {
return getServiceRegistryBuilder(cfgName).getSettings();
}
public static synchronized void unregisterAllDrivers() {
final Enumeration drivers = DriverManager.getDrivers();
while (drivers.hasMoreElements()) {
try {
DriverManager.deregisterDriver(drivers.nextElement());
} catch (SQLException e) {
e.printStackTrace();
}
}
}
public static void closeAllConnection() {
SESSION_FACTORY.values().forEach(SessionFactory::close);
SESSION_FACTORY.clear();
if (SESSION_FACTORY_IGRP != null)
SESSION_FACTORY_IGRP.close();
}
public static void removeSessionFactory(String connectionName) {
final SessionFactory sessionFactory = SESSION_FACTORY.remove(connectionName);
if (null != sessionFactory && sessionFactory.isOpen())
sessionFactory.close();
}
private static Properties getProperties(String fileName) {
final Properties properties = new Properties();
try (InputStream inputStream = HibernateUtils.class.getClassLoader().getResourceAsStream(fileName)) {
if (Objects.nonNull(inputStream)) {
properties.load(inputStream);
processEnvironmentVariables(properties);
}
} catch (IOException e) {
e.printStackTrace();
}
return properties;
}
/**
* This method processes environment variables for database access.
* It checks if the system environment variable 'ENABLE_ENV_VARIABLE_SCANNING' is set to true,
* and if so, it loads environment variables specified in the properties object
* by replacing placeholders with corresponding environment variable values.
*
* @param properties The properties object containing key-value pairs with potential environment variable placeholders.
*/
private static void processEnvironmentVariables(Properties properties) {
if (!ConfigCommonMainConstants.isEnvironmentVariableScanActive())
return;
LOG.info("Loading environment variables for database access...");
for (var propertyKey : properties.stringPropertyNames()) {
var propertyValue = properties.getProperty(propertyKey);
if (isEnvVariableFormat(propertyValue)) {
var environmentVariableName = propertyValue.substring(2, propertyValue.length() - 1);
var environmentVariableValue = System.getenv(environmentVariableName);
if (environmentVariableValue != null)
properties.setProperty(propertyKey, environmentVariableValue.trim());
else
LOG.error("Environment Variable: {} not found", environmentVariableName);
}
}
}
/**
* This method checks if the given property value is in the format of an environment variable placeholder.
*
* @param propertyValue The value of a property to be checked.
* @return True if the property value is in the format "${...}", false otherwise.
*/
private static boolean isEnvVariableFormat(String propertyValue) {
return propertyValue != null &&
propertyValue.startsWith("${") &&
propertyValue.endsWith("}");
}
public static void setSessionAudit(Session s) {
final String hcdc = (String) s.getSessionFactory().getProperties().get("hibernate.connection.driver_class");
if(hcdc.contains("postgresql")){
final User currentUser = Core.getCurrentUser();
// Set the session variables
if(currentUser!=null)
s.doWork(connection -> {
setAuditContext(connection, "audit.AUDIT_USER_CONTEXT", currentUser.getEmail());
setAuditContext(connection, "audit.AUDIT_USER_ID", String.valueOf(currentUser.getId()));
setAuditContext(connection, "audit.AUDIT_USER_IP", Igrp.getInstance().getRequest().getRemoteAddr());
});
}
}
public static void setSessionAudit(Connection connection, StringBuilder parsedQuery) {
try {
if (connection.getClientInfo("ApplicationName").contains("postgresql")){
final User currentUser = Core.getCurrentUser();
if(currentUser!=null){
parsedQuery.append(String.format("SET session audit.AUDIT_USER_CONTEXT = '%s'; ", currentUser.getEmail()));
parsedQuery.append(String.format("SET session audit.AUDIT_USER_ID = '%s'; ", String.valueOf(currentUser.getId())));
parsedQuery.append(String.format("SET session audit.AUDIT_USER_IP = '%s'; ", Igrp.getInstance().getRequest().getRemoteAddr()));
}
}
} catch (SQLException e) {
throw new RuntimeException(e);
}
}
private static void setAuditContext(Connection connection, String settingName, String settingValue) throws SQLException {
String sql = String.format("SET session %s = '%s'", settingName, settingValue);
try (PreparedStatement stmt = connection.prepareStatement(sql)) {
stmt.execute();
}
}
}