nl._42.jarb.constraint.violation.resolver.ConfigurableViolationResolver Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of jarb Show documentation
Provides support for databases
There is a newer version: 7.0.0
package nl._42.jarb.constraint.violation.resolver;

import java.util.Arrays;
import java.util.Collection;
import java.util.Set;

import javax.sql.DataSource;

import nl._42.jarb.constraint.violation.resolver.vendor.H2ViolationResolver;
import nl._42.jarb.constraint.violation.resolver.vendor.HsqlViolationResolver;
import nl._42.jarb.constraint.violation.resolver.vendor.MysqlViolationResolver;
import nl._42.jarb.constraint.violation.resolver.vendor.OracleViolationResolver;
import nl._42.jarb.constraint.violation.resolver.vendor.PostgresViolationResolver;

import nl._42.jarb.constraint.violation.resolver.vendor.H2ViolationResolver;
import nl._42.jarb.constraint.violation.resolver.vendor.HsqlViolationResolver;
import nl._42.jarb.constraint.violation.resolver.vendor.MysqlViolationResolver;
import nl._42.jarb.constraint.violation.resolver.vendor.OracleViolationResolver;
import nl._42.jarb.constraint.violation.resolver.vendor.PostgresViolationResolver;
import nl._42.jarb.utils.ClassScanner;
import nl._42.jarb.utils.jdbc.DatabaseProduct;
import org.springframework.beans.BeanUtils;

/**
 * Default violation resolver that lazily registers all custom and default
 * resolvers in a chain. Because the resolver is lazy, the registration is
 * done on first usage rather than on initialization.
 * 
 * @author Jeroen van Schagen
 */
public class ConfigurableViolationResolver extends LazyInitViolationResolver {
	
    private final ViolationResolverChain resolvers = new ViolationResolverChain();

	private final DataSource dataSource;
	
    private final Collection basePackages;

	public ConfigurableViolationResolver(DataSource dataSource, String basePackage) {
        this(dataSource, Arrays.asList(basePackage));
	}
    
    public ConfigurableViolationResolver(DataSource dataSource, Collection basePackages) {
        this.dataSource = dataSource;
        this.basePackages = basePackages;
    }

	@Override
	protected DatabaseConstraintViolationResolver init() {
        final DatabaseProduct product = DatabaseProduct.fromDataSource(dataSource);

        registerCustomResolvers(product);
        registerDefaultResolvers(product);
        return resolvers;
	}

    public void registerResolver(DatabaseConstraintViolationResolver resolver) {
        resolvers.add(resolver);
    }

    private void registerCustomResolvers(DatabaseProduct databaseProduct) {
        for (String basePackage : basePackages) {
            Set> resolverClasses = ClassScanner.getAllOfType(basePackage, DatabaseConstraintViolationResolver.class);
            for (Class resolverClass : resolverClasses) {
                resolvers.addIfSupported(newResolver(resolverClass), databaseProduct);
            }
        }
    }

	private DatabaseConstraintViolationResolver newResolver(Class resolverClass) {
		return (DatabaseConstraintViolationResolver) BeanUtils.instantiateClass(resolverClass);
	}

    private void registerDefaultResolvers(DatabaseProduct product) {
        resolvers.addIfSupported(new H2ViolationResolver(), product);
        resolvers.addIfSupported(new HsqlViolationResolver(), product);
        resolvers.addIfSupported(new MysqlViolationResolver(), product);
        resolvers.addIfSupported(new OracleViolationResolver(), product);
        resolvers.addIfSupported(new PostgresViolationResolver(), product);
        resolvers.add(new HibernateViolationResolver());
	}
	
}