Java DataSourceFactory.setValidationQuery方法代码示例

import io.dropwizard.db.DataSourceFactory; //导入方法依赖的package包/类
@Before
public void setup() throws Exception
{
    DBIFactory factory = new DBIFactory();
    Environment environment = new Environment("test", new ObjectMapper(), null, new MetricRegistry(), ClassLoader.getSystemClassLoader());
    DataSourceFactory dataSourceFactory = new DataSourceFactory();
    dataSourceFactory.setUrl("jdbc:hsqldb:mem:soa-jdbi;shutdown=true");
    dataSourceFactory.setDriverClass("org.hsqldb.jdbc.JDBCDriver");
    dataSourceFactory.setLogValidationErrors(true);
    dataSourceFactory.setUser("SA");
    dataSourceFactory.setValidationQuery("SELECT * FROM INFORMATION_SCHEMA.SYSTEM_TABLES");
    DBI jdbi = factory.build(environment, dataSourceFactory, "test");
    dynamicAttributes = new JdbiDynamicAttributes(jdbi, Collections.singletonList("test"));

    dynamicAttributes.getDao().createTable();
    dynamicAttributes.start();
}
 

import io.dropwizard.db.DataSourceFactory; //导入方法依赖的package包/类
private DataSourceFactory configureDataSourceFactory(MappedJsonConfiguration configuration) {
    DataSourceFactory dataSourceFactory = new DataSourceFactory();
    DatabaseType databaseType = configuration.getDatabaseConfiguration().getDbms();
    String connectionString = configuration.getDatabaseConfiguration().getConnection();

    if (!StringUtils.startsWithIgnoreCase(connectionString, databaseType.getConnectionPrefix())) {
        String msg = String.format("Invalid database connection URL: \"%s\" - must start with \"%s\"", connectionString, databaseType.getConnectionPrefix());
        LOGGER.error(msg);
        throw new MetadictRuntimeException(msg);
    }

    dataSourceFactory.setDriverClass(databaseType.getJdbcDriver());
    dataSourceFactory.setUrl(configuration.getDatabaseConfiguration().getConnection());
    dataSourceFactory.setValidationQuery(VALIDATION_QUERY_COMMENT + databaseType.getValidationQuery());

    LOGGER.info("Configured new DataSourceFactory of type {} to {}", databaseType, connectionString);

    return dataSourceFactory;
}
 

import io.dropwizard.db.DataSourceFactory; //导入方法依赖的package包/类
@Before
public void setUp() throws Exception {
    final Environment environment = mock(Environment.class);
    when(environment.lifecycle()).thenReturn(mock(LifecycleEnvironment.class));
    when(environment.metrics()).thenReturn(new MetricRegistry());

    when(this.bundle.getSessionHolders()).thenReturn(this.sessionHolders);

    final DataSourceFactory dataSourceFactory = new DataSourceFactory();
    dataSourceFactory.setUrl("jdbc:hsqldb:mem:unit-of-work-" + UUID.randomUUID().toString());
    dataSourceFactory.setUser("sa");
    dataSourceFactory.setDriverClass("org.hsqldb.jdbcDriver");
    dataSourceFactory.setValidationQuery("SELECT 1 FROM INFORMATION_SCHEMA.SYSTEM_USERS");
    dataSourceFactory.setProperties(ImmutableMap.of("hibernate.dialect", "org.hibernate.dialect.HSQLDialect"));
    dataSourceFactory.setMinSize(1);

    this.sessionFactory = new SessionFactoryFactory()
            .build(this.bundle, environment, dataSourceFactory, ImmutableList.<Class<?>>of(), RemoteCredentialHibernateBundle.DEFAULT_NAME);
    when(this.bundle.getSessionFactory()).thenReturn(this.sessionFactory);
    final Session session = this.sessionFactory.openSession();
    try {
        session.createSQLQuery("create table user_sessions (token varchar(64) primary key, username varchar(16))")
                .executeUpdate();
        session.createSQLQuery("insert into user_sessions values ('67ab89d', 'jeff_28')")
                .executeUpdate();
    } finally {
        session.close();
    }
}
 

import io.dropwizard.db.DataSourceFactory; //导入方法依赖的package包/类
private DataSourceFactory createConfig(String dbName) {
    Map<String, String> properties = Maps.newHashMap();
    properties.put("hibernate.dialect", "org.hibernate.dialect.H2Dialect");
    properties.put("hibernate.hbm2ddl.auto", "create");

    DataSourceFactory shard = new DataSourceFactory();
    shard.setDriverClass("org.h2.Driver");
    shard.setUrl("jdbc:h2:mem:" + dbName);
    shard.setValidationQuery("select 1");
    shard.setProperties(properties);

    return shard;
}
 

import io.dropwizard.db.DataSourceFactory; //导入方法依赖的package包/类
@Before
public void setUp() throws Exception {
    when(bundle.name()).thenReturn(getClass().getSimpleName() + "-bundle");
    when(environment.metrics()).thenReturn(metricRegistry);
    when(environment.lifecycle()).thenReturn(lifecycleEnvironment);

    config = new DataSourceFactory();
    config.setUrl("jdbc:hsqldb:mem:DbTest-" + System.currentTimeMillis());
    config.setUser("sa");
    config.setDriverClass("org.hsqldb.jdbcDriver");
    config.setValidationQuery("SELECT 1 FROM INFORMATION_SCHEMA.SYSTEM_USERS");
}
 

import io.dropwizard.db.DataSourceFactory; //导入方法依赖的package包/类
@BeforeClass
public static void setUp() throws Exception {
    final EntityManagerBundle<?> bundle = mock(EntityManagerBundle.class);
    final Environment environment = mock(Environment.class);
    when(bundle.name()).thenReturn("test-bundle");
    when(environment.lifecycle()).thenReturn(mock(LifecycleEnvironment.class));
    when(environment.metrics()).thenReturn(new MetricRegistry());

    final DataSourceFactory dataSourceFactory = new DataSourceFactory();
    dataSourceFactory.setUrl("jdbc:hsqldb:mem:unit-of-work-" + UUID.randomUUID().toString());
    dataSourceFactory.setUser("sa");
    dataSourceFactory.setDriverClass("org.hsqldb.jdbcDriver");
    dataSourceFactory.setValidationQuery("SELECT 1 FROM INFORMATION_SCHEMA.SYSTEM_USERS");
    dataSourceFactory.setProperties(ImmutableMap.of("hibernate.dialect", "org.hibernate.dialect.HSQLDialect"));
    dataSourceFactory.setInitialSize(1);
    dataSourceFactory.setMinSize(1);

    entityManagerFactory = new EntityManagerFactoryFactory()
            .build(bundle, environment, dataSourceFactory, ImmutableList.<Class<?>>of());

    final EntityManager entityManager = entityManagerFactory.createEntityManager();
    try {
        EntityTransaction entityTransaction = entityManager.getTransaction();
        entityTransaction.begin();

        entityManager.createNativeQuery("create table user_sessions (token varchar(64) primary key, username varchar(16))")
                .executeUpdate();
        entityManager.createNativeQuery("insert into user_sessions values ('67ab89d', 'jeff_28')")
                .executeUpdate();

        entityTransaction.commit();
    } finally {
        entityManager.close();
    }

    final EntityManagerContext entityManagerContext = new EntityManagerContext(entityManagerFactory);
    sharedEntityManager = new SharedEntityManagerFactory().build(entityManagerContext);
}
 

import io.dropwizard.db.DataSourceFactory; //导入方法依赖的package包/类
@Override
protected Application configure() {
    this.forceSet(TestProperties.CONTAINER_PORT, "0");

    final MetricRegistry metricRegistry = new MetricRegistry();
    final SessionFactoryFactory factory = new SessionFactoryFactory();
    final DataSourceFactory dbConfig = new DataSourceFactory();
    this.bundle = mock(RemoteCredentialHibernateBundle.class);

    final SessionHolders sessionHolders = mock(SessionHolders.class);
    when(this.bundle.getSessionHolders()).thenReturn(sessionHolders);

    final Environment environment = mock(Environment.class);
    final LifecycleEnvironment lifecycleEnvironment = mock(LifecycleEnvironment.class);
    when(environment.lifecycle()).thenReturn(lifecycleEnvironment);
    when(environment.metrics()).thenReturn(metricRegistry);

    dbConfig.setUrl("jdbc:hsqldb:mem:DbTest-" + System.nanoTime()
            + "?hsqldb.translate_dti_types=false");
    dbConfig.setUser("sa");
    dbConfig.setDriverClass("org.hsqldb.jdbcDriver");
    dbConfig.setValidationQuery("SELECT 1 FROM INFORMATION_SCHEMA.SYSTEM_USERS");

    this.sessionFactory = factory.build(this.bundle,
            environment,
            dbConfig,
            ImmutableList.<Class<?>> of(Person.class),
            RemoteCredentialHibernateBundle.DEFAULT_NAME);
    when(this.bundle.getSessionFactory()).thenReturn(this.sessionFactory);
    when(this.bundle.getCurrentThreadSessionFactory()).thenReturn(this.sessionFactory);

    final Session session = this.sessionFactory.openSession();
    try {
        session.createSQLQuery("DROP TABLE people IF EXISTS").executeUpdate();
        session.createSQLQuery(
                "CREATE TABLE people (name varchar(100) primary key, email varchar(16), birthday timestamp with time zone)")
                .executeUpdate();
        session.createSQLQuery(
                "INSERT INTO people VALUES ('Coda', '[email protected]', '1979-01-02 00:22:00+0:00')")
                .executeUpdate();
    } finally {
        session.close();
    }

    final DropwizardResourceConfig config = DropwizardResourceConfig
            .forTesting(new MetricRegistry());
    config.register(new UnitOfWorkApplicationListener("hr-db", this.bundle));
    config.register(new PersonResource(new PersonDAO(this.bundle)));
    config.register(new JacksonMessageBodyProvider(Jackson.newObjectMapper(),
            Validators.newValidator()));
    config.register(new DataExceptionMapper());

    return config;
}
 

import io.dropwizard.db.DataSourceFactory; //导入方法依赖的package包/类
@Override
protected Application configure() {
    final MetricRegistry metricRegistry = new MetricRegistry();
    final DataSourceFactory dbConfig = new DataSourceFactory();
    final Environment environment = mock(Environment.class);
    final LifecycleEnvironment lifecycleEnvironment = mock(LifecycleEnvironment.class);
    when(environment.lifecycle()).thenReturn(lifecycleEnvironment);
    when(environment.metrics()).thenReturn(metricRegistry);

    String url = "jdbc:hsqldb:mem:dwtest" + System.nanoTime();

    Map<String, String> props = new HashMap<String, String>();
    props.put("username", "sa");
    props.put("password", "");
    props.put("url", url);

    try {
        HSQLDBInit.initPublic(props);
    } catch (Exception e) {
        throw new RuntimeException(e);
    }

    dbConfig.setUrl(props.get("url"));
    dbConfig.setUser(props.get("user"));
    dbConfig.setDriverClass("org.hsqldb.jdbcDriver");
    dbConfig.setValidationQuery("SELECT 1 FROM INFORMATION_SCHEMA.SYSTEM_USERS");

    final DropwizardResourceConfig config = DropwizardResourceConfig.forTesting(new MetricRegistry());

    DataSource dataSource = dbConfig.build(metricRegistry, "jooq");
    config.register(JooqTransactionalApplicationListener.class);

    Configuration configuration = new DefaultConfiguration().set(SQLDialect.HSQLDB);
    configuration.set(new DataSourceConnectionProvider(dataSource));

    config.register(new ConfigurationFactoryProvider.Binder(configuration, dataSource,
            new TestTenantConnectionProvider(dbConfig, metricRegistry, url)));
    config.register(ExampleResource.class);
    config.register(new JacksonMessageBodyProvider(Jackson.newObjectMapper(),
            Validation.buildDefaultValidatorFactory().getValidator()));
    return config;
}