com.introproventures.graphql.jpa.query.example.starwars.StarwarsSchemaConfiguration Maven / Gradle / Ivy

package com.introproventures.graphql.jpa.query.example.starwars;

import java.util.HashMap;
import java.util.Map;

import javax.persistence.EntityManager;
import javax.persistence.EntityManagerFactory;
import javax.sql.DataSource;

import com.introproventures.graphql.jpa.query.autoconfigure.GraphQLSchemaConfigurer;
import com.introproventures.graphql.jpa.query.autoconfigure.GraphQLShemaRegistration;
import com.introproventures.graphql.jpa.query.schema.impl.GraphQLJpaSchemaBuilder;
import org.hibernate.cfg.AvailableSettings;
import org.hibernate.dialect.H2Dialect;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.boot.context.properties.ConfigurationProperties;
import org.springframework.boot.jdbc.DataSourceBuilder;
import org.springframework.boot.orm.jpa.EntityManagerFactoryBuilder;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Primary;
import org.springframework.core.io.DefaultResourceLoader;
import org.springframework.core.io.ResourceLoader;
import org.springframework.jdbc.datasource.init.DataSourceInitializer;
import org.springframework.jdbc.datasource.init.ResourceDatabasePopulator;
import org.springframework.orm.jpa.LocalContainerEntityManagerFactoryBean;

@Configuration
public class StarwarsSchemaConfiguration {

    @Bean
    @Primary
    @ConfigurationProperties(prefix = "starwars")    
    @Qualifier("starWarsDataSource")     
    public DataSource starWarsDataSource() {
        return DataSourceBuilder.create().build();
    }    
     
    @Bean
    public DataSourceInitializer starWarsDataSourceInitializer(@Qualifier("starWarsDataSource") DataSource starWarsDataSource) {
        DataSourceInitializer dataSourceInitializer = new DataSourceInitializer();
        ResourceLoader resourceLoader = new DefaultResourceLoader();
        
        ResourceDatabasePopulator databasePopulator = new ResourceDatabasePopulator();
        databasePopulator.addScript(resourceLoader.getResource("starwars.sql"));
        
        dataSourceInitializer.setDataSource(starWarsDataSource);
        dataSourceInitializer.setDatabasePopulator(databasePopulator);
        
        return dataSourceInitializer;
    }
    
    
    @Bean
    @Primary
    @Qualifier("starWarsEntityManager")
    public LocalContainerEntityManagerFactoryBean starWarsEntityManagerFactory(
            EntityManagerFactoryBuilder builder) {
        
        Map properties = new HashMap<>();
        properties.put(AvailableSettings.HBM2DDL_AUTO, "create-drop");
        properties.put(AvailableSettings.HBM2DLL_CREATE_SCHEMAS, "true");
        properties.put(AvailableSettings.DIALECT, H2Dialect.class.getName());
        properties.put(AvailableSettings.SHOW_SQL, "true");
        properties.put(AvailableSettings.FORMAT_SQL, "true");
        
        return builder
                .dataSource(starWarsDataSource())
                .packages(Character.class)
                .persistenceUnit("starwars")
                .properties(properties)
                .build();
    }    

    @Configuration
    public static class GraphQLJpaQuerySchemaConfigurer implements GraphQLSchemaConfigurer {

        private final EntityManager entityManager;

        public GraphQLJpaQuerySchemaConfigurer(@Qualifier("starWarsEntityManager") EntityManagerFactory entityManager) {
            this.entityManager = entityManager.createEntityManager();
        }

        @Override
        public void configure(GraphQLShemaRegistration registry) {

            registry.register(new GraphQLJpaSchemaBuilder(entityManager).name("GraphQLStarWars").build());
        }
    }
    
}