使用Spring 3.1,Hibernate 4.1,Dbunit等提高数据库测试的性能

时间:2012-10-13 19:33:06

标签: performance spring hibernate unit-testing dbunit

我目前正在开始一个新项目,我有大约190个存储库测试。我注意到的一件事 - 我并不完全确定为什么会发生这种情况 - 是针对HSQLDB(2.2.8)的集成测试运行速度比我想象的要慢得多。

我认为我已经跟踪了每次测试前插入数据的瓶颈。对于大多数测试,仅设置数据库的范围为.15到.38秒。这是无法接受的。我原以为内存数据库要快得多:(

以下是我的所有存储库测试扩展的数据库测试类:

@ContextConfiguration(locations = {"classpath:applicationContext.xml"})
@RunWith(SpringJUnit4ClassRunner.class)
@TransactionConfiguration(defaultRollback=true)
@Transactional
public abstract class DatabaseTest {

    public static final String TEST_RESOURCES = "src/test/resources/";

    @Autowired
    protected SessionFactory sessionFactory;

    @Autowired
    protected UserRepository userRepository;

    @Autowired
    protected DataSource dataSource;

    protected IDatabaseTester databaseTester;

    protected Map<String, Object> jdbcMap;
    protected JdbcTemplate jdbcTemplate;

    @PostConstruct
    public void initialize() throws SQLException, IOException, DataSetException {
        jdbcTemplate = new JdbcTemplate(dataSource);

        setupHsqlDb();

        databaseTester = new DataSourceDatabaseTester(dataSource);
        databaseTester.setSetUpOperation(DatabaseOperation.CLEAN_INSERT);
        databaseTester.setTearDownOperation(DatabaseOperation.NONE);
        databaseTester.setDataSet(getDataSet());
    }

    @Before
    public void insertDbUnitData() throws Exception {
        long time = System.currentTimeMillis();

        databaseTester.onSetup();

        long elapsed = System.currentTimeMillis() - time;
        System.out.println(getClass() + " Insert DB Unit Data took: " + elapsed);
    }

    @After
    public void cleanDbUnitData() throws Exception {
        databaseTester.onTearDown();
    }

    public IDataSet getDataSet() throws IOException, DataSetException {
        Set<String> filenames = getDataSets().getFilenames();

        IDataSet[] dataSets = new IDataSet[filenames.size()];
        Iterator<String> iterator = filenames.iterator();
        for(int i = 0; iterator.hasNext(); i++) {
            dataSets[i] = new FlatXmlDataSet(
                new FlatXmlProducer(
                    new InputSource(TEST_RESOURCES + iterator.next()), false, true
                )
            );
        }

        return new CompositeDataSet(dataSets);
    }

    public void setupHsqlDb() throws SQLException {
        Connection sqlConnection = DataSourceUtils.getConnection(dataSource);
        String databaseName = sqlConnection.getMetaData().getDatabaseProductName();
        sqlConnection.close();

        if("HSQL Database Engine".equals(databaseName)) {
            jdbcTemplate.update("SET DATABASE REFERENTIAL INTEGRITY FALSE;");

            // MD5
            jdbcTemplate.update("DROP FUNCTION MD5 IF EXISTS;");
            jdbcTemplate.update(
                "CREATE FUNCTION MD5(VARCHAR(226)) " +
                    "RETURNS VARCHAR(226) " +
                    "LANGUAGE JAVA " +
                    "DETERMINISTIC " +
                    "NO SQL " +
                    "EXTERNAL NAME 'CLASSPATH:org.apache.commons.codec.digest.DigestUtils.md5Hex';"
            );
        } else {
            jdbcTemplate.update("SET foreign_key_checks = 0;");
        }
    }

    protected abstract DataSet getDataSets();

    protected void flush() {
        sessionFactory.getCurrentSession().flush();
    }

    protected void clear() {
        sessionFactory.getCurrentSession().clear();
    }

    protected void setCurrentUser(User user) {
        if(user != null) {
            Authentication authentication = new UsernamePasswordAuthenticationToken(user,
                user, user.getAuthorities());

            SecurityContextHolder.getContext().setAuthentication(authentication);
        }
    }

    protected void setNoCurrentUser() {
        SecurityContextHolder.getContext().setAuthentication(null);
    }

    protected User setCurrentUser(long userId) {
        User user = userRepository.find(userId);

        if(user.getId() != userId) {
            throw new IllegalArgumentException("There is no user with id: " + userId);
        }

        setCurrentUser(user);

        return user;
    }

    protected User getCurrentUser() {
        return (User) SecurityContextHolder.getContext().getAuthentication().getPrincipal();
    }

}

以下是我的应用程序上下文中的相关bean:

<bean class="org.springframework.beans.factory.config.PropertyPlaceholderConfigurer">
    <property name="locations" value="classpath:applicationContext.properties"/>
</bean>

<bean id="dataSource" class="com.mchange.v2.c3p0.ComboPooledDataSource"
      destroy-method="close">
    <property name="driverClass" value="${database.driver}"/>
    <property name="jdbcUrl" value="${database.url}"/>
    <property name="user" value="${database.username}"/>
    <property name="password" value="${database.password}"/>
    <property name="initialPoolSize" value="10"/>
    <property name="minPoolSize" value="10"/>
    <property name="maxPoolSize" value="50"/>
    <property name="idleConnectionTestPeriod" value="100"/>
    <property name="acquireIncrement" value="2"/>
    <property name="maxStatements" value="0"/>
    <property name="maxIdleTime" value="1800"/>
    <property name="numHelperThreads" value="3"/>
    <property name="acquireRetryAttempts" value="2"/>
    <property name="acquireRetryDelay" value="1000"/>
    <property name="checkoutTimeout" value="5000"/>
</bean>

<bean id="sessionFactory"
      class="org.springframework.orm.hibernate4.LocalSessionFactoryBean">
    <property name="dataSource" ref="dataSource"/>
    <property name="mappingResources">
        <list>
            <value>...</value>
        </list>
    </property>
    <property name="namingStrategy">
        <bean class="org.hibernate.cfg.ImprovedNamingStrategy"/>
    </property>
    <property name="hibernateProperties">
        <props>
            <prop key="javax.persistence.validation.mode">none</prop>

            <prop key="hibernate.dialect">${hibernate.dialect}</prop>
            <prop key="hibernate.hbm2ddl.auto">${hibernate.hbm2ddl.auto}
            </prop>
            <prop key="hibernate.generate_statistics">false</prop>

            <prop key="hibernate.show_sql">false</prop>
            <prop key="hibernate.format_sql">true</prop>

            <prop key="hibernate.cache.use_second_level_cache">false</prop>
            <prop key="hibernate.cache.provider_class">

            </prop>
        </props>
    </property>
</bean>

<bean class="org.springframework.orm.hibernate4.HibernateExceptionTranslator"/>

<bean id="transactionManager"
      class="org.springframework.orm.hibernate4.HibernateTransactionManager">
    <property name="sessionFactory" ref="sessionFactory"/>
</bean>

为了尝试插入更少的数据,我允许每个测试类选择一个只加载所需数据的DataSet枚举。它是这样指定的:

public enum DataSet {
    NONE(create()),
    CORE(create("core.xml")),
    USERS(combine(create("users.xml"), CORE)),
    TAGS(combine(create("tags.xml"), USERS)),

这会导致它运行得更慢而不是更快吗?我的想法是,如果我只想要核心xml(语言,省份等),我只需要加载这些记录。我认为这会使测试套件更快,但它仍然太慢。

我可以通过创建专门为每个测试类设计的单独xml数据集来节省一些时间。这会删除一些插入语句。但即使我在一个xml数据集中有20个插入语句(因此,除了将数据集直接插入到java代码中之外的最小I / O损失),每个测试在初始化期间仍需要.1到.15秒数据库数据!我难以置信,将20条记录插入内存需要0.5秒。

在我使用Spring 3.0和Hibernate 3.x的其他项目中,在每次测试之前插入所有内容需要30毫秒,但实际上每次测试插入100行或更多行。对于只有20个插入物的测试,它们就像没有任何延迟一样飞行。这是我的预期。我开始认为问题在于Spring的注释 - 或者我在DatabaseTest类中设置它们的方式。这基本上是唯一不同的东西。

此外,我的存储库使用sessionFactory.getCurrentSession()而不是HibernateTemplate。这是我第一次开始使用Spring中基于注释的单元测试内容,因为不推荐使用Spring测试类。这可能是他们变慢的原因吗?

如果您需要了解其他任何信息以帮助解决问题,请告知我们。我有点难过。

编辑:我写了答案。问题是hsqldb 2.2.x.恢复到2.0.0可以解决问题。

2 个答案:

答案 0 :(得分:3)

看起来很快,恕我直言。我看到了更慢的集成测试。也就是说,有各种方法可以使您的测试更快:

  • 减少插入的数据量
  • 如果数据集相同且先前的测试是只读测试(通常是这种情况,特别是如果您在每次测试结束时回滚),则避免插入与先前测试相同的数据。

我想用DbUnit这样做应该是可能的。如果您已准备好使用其他框架,则可以使用我自己的DbSetup,它支持开箱即用。

答案 1 :(得分:3)

问题是Hsqldb 2.2.8。我恢复到2.0.0,我的性能提升了8-10倍,或者立刻提升了。它没有花费150-280毫秒,而是下降到7-15(有时是20)毫秒。

我的整个测试套件(490次测试)现在只用了18秒而不是80秒。

我想给每个人留言:避免使用hsqldb 2.2.x.我认为他们添加了多线程支持,这导致了这种用例的性能问题。