org.springframework.data.domain.AuditorAware#org.springframework.orm.jpa.vendor.Database源码实例Demo

@Bean
public LocalContainerEntityManagerFactoryBean entityManagerFactory() {

	HibernateJpaVendorAdapter vendorAdapter = new HibernateJpaVendorAdapter();
	vendorAdapter.setDatabase(Database.HSQL);
	vendorAdapter.setGenerateDdl(true);

	LocalContainerEntityManagerFactoryBean factory = new LocalContainerEntityManagerFactoryBean();
	factory.setJpaVendorAdapter(vendorAdapter);
	factory.setPackagesToScan(packagesToScan());
	factory.setPersistenceUnitName("spring-data-rest-webmvc");
	factory.setDataSource(dataSource());
	factory.afterPropertiesSet();

	return factory;
}
 

/**
 * 获取数据库类型
 */
public static Database getDBType(DataSource dataSource){
	String jdbcUrl = getJdbcUrlFromDataSource(dataSource);
	
	if (StringUtils.contains(jdbcUrl, ":h2:")) {
		return Database.H2 ;
	} else if (StringUtils.contains(jdbcUrl, ":mysql:")) {
		return Database.MYSQL ;
	} else if (StringUtils.contains(jdbcUrl, ":oracle:")) {
		return Database.ORACLE ;
	} else if (StringUtils.contains(jdbcUrl, ":postgresql:")) {
		return Database.POSTGRESQL ;
	} else if (StringUtils.contains(jdbcUrl, ":sqlserver:")) {
		return Database.SQL_SERVER ;
	}
	throw new IllegalArgumentException("Unknown Database of " + jdbcUrl);
}
 

/**
 * Gets the Hibernate JPA vendor adapter needed by the entity manager.
 *
 * @return the Hibernate JPA vendor adapter.
 */
private JpaVendorAdapter getHibernateJpaVendorAdapter()
{
    HibernateJpaVendorAdapter hibernateJpaVendorAdapter = new HibernateJpaVendorAdapter();
    // Set the database type.
    String databaseType = configurationHelper.getProperty(ConfigurationValue.DATABASE_TYPE);
    if (StringUtils.isBlank(databaseType))
    {
        throw new IllegalStateException(
            String.format("No database type found. Ensure the \"%s\" configuration entry is configured.", ConfigurationValue.DATABASE_TYPE.getKey()));
    }
    Database database = Database.valueOf(databaseType);
    LOGGER.info("jpaTargetDatabase={}", database);
    hibernateJpaVendorAdapter.setDatabase(database);
    hibernateJpaVendorAdapter.setGenerateDdl(false);
    return hibernateJpaVendorAdapter;
}
 

public JpaTargetManagement(final EntityManager entityManager, final QuotaManagement quotaManagement,
        final TargetRepository targetRepository, final TargetMetadataRepository targetMetadataRepository,
        final RolloutGroupRepository rolloutGroupRepository,
        final DistributionSetRepository distributionSetRepository,
        final TargetFilterQueryRepository targetFilterQueryRepository,
        final TargetTagRepository targetTagRepository, final NoCountPagingRepository criteriaNoCountDao,
        final EventPublisherHolder eventPublisherHolder, final TenantAware tenantAware,
        final AfterTransactionCommitExecutor afterCommit, final VirtualPropertyReplacer virtualPropertyReplacer,
        final Database database) {
    this.entityManager = entityManager;
    this.quotaManagement = quotaManagement;
    this.targetRepository = targetRepository;
    this.targetMetadataRepository = targetMetadataRepository;
    this.rolloutGroupRepository = rolloutGroupRepository;
    this.distributionSetRepository = distributionSetRepository;
    this.targetFilterQueryRepository = targetFilterQueryRepository;
    this.targetTagRepository = targetTagRepository;
    this.criteriaNoCountDao = criteriaNoCountDao;
    this.eventPublisherHolder = eventPublisherHolder;
    this.tenantAware = tenantAware;
    this.afterCommit = afterCommit;
    this.virtualPropertyReplacer = virtualPropertyReplacer;
    this.database = database;
}
 

public JpaSoftwareModuleManagement(final EntityManager entityManager,
        final DistributionSetRepository distributionSetRepository,
        final SoftwareModuleRepository softwareModuleRepository,
        final SoftwareModuleMetadataRepository softwareModuleMetadataRepository,
        final SoftwareModuleTypeRepository softwareModuleTypeRepository,
        final NoCountPagingRepository criteriaNoCountDao, final AuditorAware<String> auditorProvider,
        final ArtifactManagement artifactManagement, final QuotaManagement quotaManagement,
        final VirtualPropertyReplacer virtualPropertyReplacer, final Database database) {
    this.entityManager = entityManager;
    this.distributionSetRepository = distributionSetRepository;
    this.softwareModuleRepository = softwareModuleRepository;
    this.softwareModuleMetadataRepository = softwareModuleMetadataRepository;
    this.softwareModuleTypeRepository = softwareModuleTypeRepository;
    this.criteriaNoCountDao = criteriaNoCountDao;
    this.auditorProvider = auditorProvider;
    this.artifactManagement = artifactManagement;
    this.quotaManagement = quotaManagement;
    this.virtualPropertyReplacer = virtualPropertyReplacer;
    this.database = database;
}
 

@Test
public void correctRsqlBuildsLikePredicateWithPercentage() {
    reset(baseSoftwareModuleRootMock, criteriaQueryMock, criteriaBuilderMock);
    final String correctRsql = "name==a%";
    when(baseSoftwareModuleRootMock.get("name")).thenReturn(baseSoftwareModuleRootMock);
    when(baseSoftwareModuleRootMock.getJavaType()).thenReturn((Class) SoftwareModule.class);
    when(criteriaBuilderMock.like(any(Expression.class), anyString(), eq('\\'))).thenReturn(mock(Predicate.class));
    when(criteriaBuilderMock.<String> greaterThanOrEqualTo(any(Expression.class), any(String.class)))
            .thenReturn(mock(Predicate.class));
    when(criteriaBuilderMock.upper(eq(pathOfString(baseSoftwareModuleRootMock))))
            .thenReturn(pathOfString(baseSoftwareModuleRootMock));
    // test
    RSQLUtility.parse(correctRsql, SoftwareModuleFields.class, null, Database.H2)
            .toPredicate(baseSoftwareModuleRootMock, criteriaQueryMock, criteriaBuilderMock);

    // verification
    verify(criteriaBuilderMock, times(1)).and(any(Predicate.class));
    verify(criteriaBuilderMock, times(1)).like(eq(pathOfString(baseSoftwareModuleRootMock)),
            eq("a\\%".toUpperCase()), eq('\\'));
}
 

@Test
public void correctRsqlBuildsLikePredicateWithPercentageSQLServer() {
    reset(baseSoftwareModuleRootMock, criteriaQueryMock, criteriaBuilderMock);
    final String correctRsql = "name==a%";
    when(baseSoftwareModuleRootMock.get("name")).thenReturn(baseSoftwareModuleRootMock);
    when(baseSoftwareModuleRootMock.getJavaType()).thenReturn((Class) SoftwareModule.class);
    when(criteriaBuilderMock.upper(eq(pathOfString(baseSoftwareModuleRootMock))))
            .thenReturn(pathOfString(baseSoftwareModuleRootMock));
    when(criteriaBuilderMock.like(any(Expression.class), anyString(), eq('\\'))).thenReturn(mock(Predicate.class));
    when(criteriaBuilderMock.<String> greaterThanOrEqualTo(any(Expression.class), any(String.class)))
            .thenReturn(mock(Predicate.class));

    // test
    RSQLUtility.parse(correctRsql, SoftwareModuleFields.class, null, Database.SQL_SERVER)
            .toPredicate(baseSoftwareModuleRootMock, criteriaQueryMock, criteriaBuilderMock);

    // verification
    verify(criteriaBuilderMock, times(1)).and(any(Predicate.class));
    verify(criteriaBuilderMock, times(1)).like(eq(pathOfString(baseSoftwareModuleRootMock)),
            eq("a[%]".toUpperCase()), eq('\\'));
}
 

@Bean
public JpaVendorAdapter jpaVendorAdapter() {
  HibernateJpaVendorAdapter adapter = new HibernateJpaVendorAdapter();
  adapter.setDatabase(Database.H2);
  adapter.setShowSql(true);
  adapter.setGenerateDdl(false);
  adapter.setDatabasePlatform("org.hibernate.dialect.H2Dialect");
  return adapter;
}
 

@Bean
public HibernateJpaVendorAdapter jpaVendorAdapter() {
  HibernateJpaVendorAdapter adapter = new HibernateJpaVendorAdapter();
  adapter.setDatabase(Database.H2);
  adapter.setShowSql(false);
  adapter.setGenerateDdl(true);
  return adapter;
}
 

@Bean
public JpaVendorAdapter jpaVendorAdapter() {
  HibernateJpaVendorAdapter adapter = new HibernateJpaVendorAdapter();
  adapter.setDatabase(Database.H2);
  adapter.setShowSql(true);
  adapter.setGenerateDdl(false);
  adapter.setDatabasePlatform("org.hibernate.dialect.H2Dialect");
  return adapter;
}
 

@Bean
public JpaVendorAdapter jpaVendorAdapter() {
	final HibernateJpaVendorAdapter hibernateJpaVendorAdapter = new HibernateJpaVendorAdapter();
	hibernateJpaVendorAdapter.setShowSql(true);
	hibernateJpaVendorAdapter.setGenerateDdl(true);
	hibernateJpaVendorAdapter.setDatabase(Database.H2);
	return hibernateJpaVendorAdapter;
}
 

@Bean
public LocalContainerEntityManagerFactoryBean entityManagerFactory() {
  final HibernateJpaVendorAdapter vendorAdapter = new HibernateJpaVendorAdapter();
  vendorAdapter.setDatabase(Database.H2);
  vendorAdapter.setGenerateDdl(true);

  final LocalContainerEntityManagerFactoryBean em = new LocalContainerEntityManagerFactoryBean();
  em.setDataSource(dataSource());
  em.setPackagesToScan("io.opentracing.contrib.specialagent.test.spring.data");
  em.setJpaVendorAdapter(vendorAdapter);
  return em;
}
 

@Bean(name = "entityManagerFactory")
public EntityManagerFactory entityManagerFactory() {
  LocalContainerEntityManagerFactoryBean entityManagerFactoryBean = new LocalContainerEntityManagerFactoryBean();
  entityManagerFactoryBean.setDataSource(dataSource());
  entityManagerFactoryBean.setPackagesToScan("org.activiti.rest.api.jpa.model");
  entityManagerFactoryBean.setPersistenceUnitName("test");
  HibernateJpaVendorAdapter hibernateJpaVendorAdapter = new HibernateJpaVendorAdapter();
  hibernateJpaVendorAdapter.setShowSql(false);
  hibernateJpaVendorAdapter.setGenerateDdl(true);
  hibernateJpaVendorAdapter.setDatabase(Database.H2);
  entityManagerFactoryBean.setJpaVendorAdapter(hibernateJpaVendorAdapter);
  entityManagerFactoryBean.afterPropertiesSet();
  return entityManagerFactoryBean.getObject();
}
 

@Bean
public JpaVendorAdapter jpaVendorAdapter() {
  AbstractJpaVendorAdapter adapter = new HibernateJpaVendorAdapter();
  adapter.setShowSql(jpaProperties.isShowSql());
  adapter.setDatabase(Database.MYSQL);
  adapter.setDatabasePlatform(jpaProperties.getDatabasePlatform());
  adapter.setGenerateDdl(jpaProperties.isGenerateDdl());
  return adapter;
}
 

@Bean
LocalContainerEntityManagerFactoryBean entityManagerFactory(DataSource dataSource) {

	HibernateJpaVendorAdapter adapter = new HibernateJpaVendorAdapter();
	adapter.setDatabase(Database.HSQL);
	adapter.setGenerateDdl(true);

	LocalContainerEntityManagerFactoryBean factoryBean = new LocalContainerEntityManagerFactoryBean();
	factoryBean.setDataSource(dataSource);
	factoryBean.setJpaVendorAdapter(adapter);
	factoryBean.setPackagesToScan("org.springframework.events.jpa");

	return factoryBean;
}
 

@Bean
protected EclipseLinkJpaVendorAdapter jpaVendorAdapter() {
    EclipseLinkJpaVendorAdapter adapter = new EclipseLinkJpaVendorAdapter();
    adapter.setDatabase(Database.MYSQL);
    adapter.setShowSql(true);
    adapter.setGenerateDdl(false);
    return adapter;
}
 

@Bean
public LocalContainerEntityManagerFactoryBean entityManagerFactory() {
	HibernateJpaVendorAdapter vendorAdapter = new HibernateJpaVendorAdapter();
	vendorAdapter.setDatabase(Database.H2);
	vendorAdapter.setGenerateDdl(true);

	LocalContainerEntityManagerFactoryBean factory = new LocalContainerEntityManagerFactoryBean();
	factory.setJpaVendorAdapter(vendorAdapter);
	factory.setPackagesToScan("org.springframework.content.cmis.integration");
	factory.setDataSource(dataSource());

	return factory;
}
 

@Bean
public LocalContainerEntityManagerFactoryBean entityManagerFactory() {
	HibernateJpaVendorAdapter vendorAdapter = new HibernateJpaVendorAdapter();
	vendorAdapter.setDatabase(Database.H2);
	vendorAdapter.setGenerateDdl(true);

	LocalContainerEntityManagerFactoryBean factory = new LocalContainerEntityManagerFactoryBean();
	factory.setJpaVendorAdapter(vendorAdapter);
	factory.setPackagesToScan("org.springframework.content.cmis.support");
	factory.setDataSource(dataSource());

	return factory;
}
 

@Bean
public LocalContainerEntityManagerFactoryBean entityManagerFactory() {
    HibernateJpaVendorAdapter vendorAdapter = new HibernateJpaVendorAdapter();
    vendorAdapter.setDatabase(Database.H2);
    vendorAdapter.setGenerateDdl(true);

    LocalContainerEntityManagerFactoryBean factory = new LocalContainerEntityManagerFactoryBean();
    factory.setJpaVendorAdapter(vendorAdapter);
    factory.setPackagesToScan("internal.org.springframework.renditions.context");
    factory.setDataSource(dataSource());

    return factory;
}
 

@Bean
public LocalContainerEntityManagerFactoryBean entityManagerFactory() {
    HibernateJpaVendorAdapter vendorAdapter = new HibernateJpaVendorAdapter();
    vendorAdapter.setDatabase(Database.H2);
    vendorAdapter.setGenerateDdl(true);

    LocalContainerEntityManagerFactoryBean factory = new LocalContainerEntityManagerFactoryBean();
    factory.setJpaVendorAdapter(vendorAdapter);
    factory.setPackagesToScan("org.springframework.content.elasticsearch");
    factory.setDataSource(dataSource());

    return factory;
}
 

@Bean
public LocalContainerEntityManagerFactoryBean entityManagerFactory() {
	HibernateJpaVendorAdapter vendorAdapter = new HibernateJpaVendorAdapter();
	vendorAdapter.setDatabase(Database.HSQL);
	vendorAdapter.setGenerateDdl(true);

	LocalContainerEntityManagerFactoryBean factory = new LocalContainerEntityManagerFactoryBean();
	factory.setJpaVendorAdapter(vendorAdapter);
	factory.setPackagesToScan(getClass().getPackage().getName());
	factory.setDataSource(dataSource());

	return factory;
}
 

@Bean
public JpaVendorAdapter jpaVendorAdapter() {
    HibernateJpaVendorAdapter adapter = new HibernateJpaVendorAdapter();
    adapter.setShowSql(true);
    adapter.setGenerateDdl(true);
    adapter.setDatabase(Database.H2);
    adapter.setDatabasePlatform("org.hibernate.dialect.H2Dialect");
    return adapter;
}
 

@Bean
public JpaVendorAdapter jpaVendorAdapter() {
    HibernateJpaVendorAdapter adapter = new HibernateJpaVendorAdapter();
    adapter.setShowSql(true);
    adapter.setGenerateDdl(true);
    adapter.setDatabase(Database.MYSQL);
    return adapter;
}
 

@Bean(name = "entityManagerFactory")
public EntityManagerFactory entityManagerFactory(DataSource dataSource) {
    LocalContainerEntityManagerFactoryBean entityManagerFactoryBean = new LocalContainerEntityManagerFactoryBean();
    entityManagerFactoryBean.setDataSource(dataSource);
    entityManagerFactoryBean.setPackagesToScan("org.flowable.rest.api.jpa.model");
    entityManagerFactoryBean.setPersistenceUnitName("test");
    HibernateJpaVendorAdapter hibernateJpaVendorAdapter = new HibernateJpaVendorAdapter();
    hibernateJpaVendorAdapter.setShowSql(false);
    hibernateJpaVendorAdapter.setGenerateDdl(true);
    hibernateJpaVendorAdapter.setDatabase(Database.H2);
    entityManagerFactoryBean.setJpaVendorAdapter(hibernateJpaVendorAdapter);
    entityManagerFactoryBean.afterPropertiesSet();
    return entityManagerFactoryBean.getObject();
}
 

@Bean
public LocalContainerEntityManagerFactoryBean entityManagerFactory() {

  LocalContainerEntityManagerFactoryBean factory = new LocalContainerEntityManagerFactoryBean();
  factory.setDataSource(dataSource());
  factory.setMappingResources("META-INF/orm.xml");
  factory.setPackagesToScan("com.cosium.spring.data.jpa.entity.graph");

  HibernateJpaVendorAdapter jpaVendorAdapter = new HibernateJpaVendorAdapter();
  jpaVendorAdapter.setDatabase(Database.H2);
  jpaVendorAdapter.setGenerateDdl(true);
  factory.setJpaVendorAdapter(jpaVendorAdapter);

  return factory;
}
 

@Bean
public JpaVendorAdapter jpaVendorAdapter() {
	HibernateJpaVendorAdapter hibernateJpaVendorAdapter = new HibernateJpaVendorAdapter();
	hibernateJpaVendorAdapter.setShowSql(true);
	hibernateJpaVendorAdapter.setGenerateDdl(true);
	hibernateJpaVendorAdapter.setDatabase(Database.MYSQL);
	return hibernateJpaVendorAdapter;
}
 

public JpaTargetTagManagement(final TargetTagRepository targetTagRepository,
        final TargetRepository targetRepository, final VirtualPropertyReplacer virtualPropertyReplacer,
        final Database database) {
    this.targetTagRepository = targetTagRepository;
    this.targetRepository = targetRepository;
    this.virtualPropertyReplacer = virtualPropertyReplacer;
    this.database = database;
}
 

private RSQLSpecification(final String rsql, final Class<A> enumType,
        final VirtualPropertyReplacer virtualPropertyReplacer, final Database database) {
    this.rsql = rsql;
    this.enumType = enumType;
    this.virtualPropertyReplacer = virtualPropertyReplacer;
    this.database = database;
}
 

private JpqQueryRSQLVisitor(final Root<T> root, final CriteriaBuilder cb, final Class<A> enumType,
        final VirtualPropertyReplacer virtualPropertyReplacer, final Database database,
        final CriteriaQuery<?> query) {
    this.root = root;
    this.cb = cb;
    this.query = query;
    this.enumType = enumType;
    this.virtualPropertyReplacer = virtualPropertyReplacer;
    this.simpleTypeConverter = new SimpleTypeConverter();
    this.database = database;
    this.joinsNeeded = false;
}
 

private String toSQL(final String transformedValue) {
    final String escaped;

    if (database == Database.SQL_SERVER) {
        escaped = transformedValue.replace("%", "[%]").replace("_", "[_]");
    } else {
        escaped = transformedValue.replace("%", ESCAPE_CHAR + "%").replace("_", ESCAPE_CHAR + "_");
    }

    return escaped.replace(LIKE_WILDCARD, '%').toUpperCase();
}