com.custardsource.dybdob.WarningRecordRepository Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of dybdob-core Show documentation
core classes used by other dybdob modules
The newest version!
package com.custardsource.dybdob;

import java.util.List;
import java.util.Properties;

import org.hibernate.SessionFactory;
import org.hibernate.criterion.DetachedCriteria;
import org.hibernate.criterion.Order;
import org.hibernate.criterion.Restrictions;
import org.springframework.jdbc.datasource.DriverManagerDataSource;
import org.springframework.orm.hibernate3.HibernateTemplate;
import org.springframework.orm.hibernate3.annotation.AnnotationSessionFactoryBean;

public class WarningRecordRepository {
    private HibernateTemplate hibernateTemplate;

    public WarningRecordRepository(String jdbcDriver, String jdbcConnection, String jdbcUser, String jdbcPassword, String hibernateDialect) {
        try {
            Class.forName(jdbcDriver);
        } catch (ClassNotFoundException e) {
            throw new RuntimeException("Cannot load specified JDBC driver: " + jdbcDriver, e);
        }
        DriverManagerDataSource dataSource=new DriverManagerDataSource(jdbcConnection, jdbcUser, jdbcPassword);

        AnnotationSessionFactoryBean sessionFactoryBean = new AnnotationSessionFactoryBean();
        sessionFactoryBean.setDataSource(dataSource);
        Properties config = new Properties();
        config.setProperty("hibernate.dialect", hibernateDialect);
        config.setProperty("hibernate.connection.autocommit", "true");
        config.setProperty("hibernate.hbm2ddl.auto", "update");
        sessionFactoryBean.setHibernateProperties(config);

        sessionFactoryBean.setAnnotatedClasses(new Class[]{WarningRecord.class});
        try {
            sessionFactoryBean.afterPropertiesSet();
        } catch (Exception e) {
            throw new RuntimeException("Could not set up database connection", e);
        }
        hibernateTemplate = new HibernateTemplate((SessionFactory) sessionFactoryBean.getObject());
    }

    public void recordWarningCount(WarningRecord record) {
        hibernateTemplate.save(record);
    }


    public WarningRecord lastWarningRecordFor(WarningRecord record) {
        DetachedCriteria c = DetachedCriteria.forClass(WarningRecord.class);
        ProjectVersion projectVersion = record.projectVersion();
        WarningSource source = record.source();
        c.add(Restrictions.eq("projectVersion.groupId", projectVersion.getGroupId()));
        c.add(Restrictions.eq("projectVersion.artifactId", projectVersion.getArtifactId()));
        c.add(Restrictions.eq("projectVersion.version", projectVersion.getVersion()));
        c.add(Restrictions.eq("source.source", source.getSource()));
        c.add(Restrictions.eq("source.metric", source.getMetric()));
        c.addOrder(Order.desc("dateLogged"));
        List matches = hibernateTemplate.findByCriteria(c, 0, 1);
        if (matches.isEmpty()) {
            return null;
        }
        return matches.get(0);
    }
}