com.yahoo.elide.standalone.Util Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of elide-standalone Show documentation
Elide Standalone Application
The newest version!
/*
 * Copyright 2017, Yahoo Inc.
 * Licensed under the Apache License, Version 2.0
 * See LICENSE file in project root for terms.
 */
package com.yahoo.elide.standalone;

import com.yahoo.elide.async.models.AsyncQuery;
import com.yahoo.elide.core.utils.ClassScanner;
import com.yahoo.elide.datastores.jpa.PersistenceUnitInfoImpl;
import com.zaxxer.hikari.HikariConfig;
import com.zaxxer.hikari.HikariDataSource;
import org.hibernate.jpa.boot.internal.EntityManagerFactoryBuilderImpl;
import org.hibernate.jpa.boot.internal.PersistenceUnitInfoDescriptor;

import jakarta.persistence.Entity;
import jakarta.persistence.EntityManagerFactory;
import jakarta.persistence.spi.PersistenceUnitInfo;

import java.util.HashMap;
import java.util.List;
import java.util.Properties;
import java.util.stream.Collectors;

import javax.sql.DataSource;

/**
 * Util.
 */
public class Util {

    public static EntityManagerFactory getEntityManagerFactory(
            ClassScanner scanner,
            String modelPackageName,
            boolean includeAsyncModel,
            Properties options
    ) {

        // Configure default options for example service
        populateDefaultOptions(options);

        ClassLoader classLoader = null;

        PersistenceUnitInfo persistenceUnitInfo = new PersistenceUnitInfoImpl("elide-stand-alone",
                combineModelEntities(scanner, modelPackageName, includeAsyncModel),
                options, classLoader);

        return new EntityManagerFactoryBuilderImpl(
                new PersistenceUnitInfoDescriptor(persistenceUnitInfo), new HashMap<>(), classLoader)
                .build();
    }

    /**
     * Creates {@link DataSource} object from Database Connection Properties.
     * @param options Database Connection Properties.
     * @return DataSource Object.
     */
    public static DataSource getDataSource(Properties options) {

        // Configure default options for example service
        populateDefaultOptions(options);

        HikariConfig config = new HikariConfig();
        config.setJdbcUrl(options.getProperty("jakarta.persistence.jdbc.url"));
        config.setUsername(options.getProperty("jakarta.persistence.jdbc.user"));
        config.setPassword(options.getProperty("jakarta.persistence.jdbc.password"));
        config.setDriverClassName(options.getProperty("jakarta.persistence.jdbc.driver"));
        if (options.getProperty("hibernate.hikari.connectionTimeout") != null) {
            config.setConnectionTimeout(new Long(options.getProperty("hibernate.hikari.connectionTimeout")));
        }
        if (options.getProperty("hibernate.hikari.idleTimeout") != null) {
            config.setIdleTimeout(new Long(options.getProperty("hibernate.hikari.idleTimeout")));
        }
        if (options.getProperty("hibernate.hikari.maximumPoolSize") != null) {
            config.setMaximumPoolSize(new Integer(options.getProperty("hibernate.hikari.maximumPoolSize")));
        }

        return new HikariDataSource(config);
    }

    private static void populateDefaultOptions(Properties options) {
        if (options.isEmpty()) {
            options.put("hibernate.show_sql", "true");
            options.put("hibernate.hbm2ddl.auto", "create");
            options.put("hibernate.dialect", "org.hibernate.dialect.MySQL5Dialect");
            options.put("hibernate.current_session_context_class", "thread");
            options.put("hibernate.jdbc.use_scrollable_resultset", "true");

            // Collection Proxy & JDBC Batching
            options.put("hibernate.jdbc.batch_size", "50");
            options.put("hibernate.jdbc.fetch_size", "50");
            options.put("hibernate.default_batch_fetch_size", "100");

            // Hikari Connection Pool Settings
            options.putIfAbsent("hibernate.connection.provider_class",
                    "com.zaxxer.hikari.hibernate.HikariConnectionProvider");
            options.putIfAbsent("hibernate.hikari.connectionTimeout", "20000");
            options.putIfAbsent("hibernate.hikari.maximumPoolSize", "30");
            options.putIfAbsent("hibernate.hikari.idleTimeout", "30000");

            options.put("jakarta.persistence.jdbc.driver", "com.mysql.jdbc.Driver");
            options.put("jakarta.persistence.jdbc.url", "jdbc:mysql://localhost/elide?serverTimezone=UTC");
            options.put("jakarta.persistence.jdbc.user", "elide");
            options.put("jakarta.persistence.jdbc.password", "elide123");
        }
    }

    /**
     * Combine the model entities with Async and Dynamic models.
     *
     * @param scanner Class scanner
     * @param modelPackageName Package name
     * @param includeAsyncModel Include Async model package Name
     * @return All entities combined from both package.
     */
    public static List combineModelEntities(ClassScanner scanner,
                                                    String modelPackageName, boolean includeAsyncModel) {

        List modelEntities = getAllEntities(scanner, modelPackageName);

        if (includeAsyncModel) {
            modelEntities.addAll(getAllEntities(scanner, AsyncQuery.class.getPackage().getName()));
        }
        return modelEntities;
    }

    /**
     * Get all the entities in a package.
     *
     * @param packageName Package name
     * @return All entities found in package.
     */
    public static List getAllEntities(ClassScanner scanner, String packageName) {
        return scanner.getAnnotatedClasses(packageName, Entity.class).stream()
                .map(Class::getName)
                .collect(Collectors.toList());
    }
}