matrix.module.jdbc.config.DruidAutoConfiguration Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of matrix-module-jdbc-starter Show documentation

vwkit-matrix

There is a newer version: 1.1.1

Show newest version

package matrix.module.jdbc.config;

import com.alibaba.druid.support.http.StatViewServlet;
import com.alibaba.druid.support.http.WebStatFilter;
import matrix.module.jdbc.properties.JdbcProperties;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
import org.springframework.boot.context.properties.EnableConfigurationProperties;
import org.springframework.boot.web.servlet.FilterRegistrationBean;
import org.springframework.boot.web.servlet.ServletRegistrationBean;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

import javax.servlet.Filter;
import java.util.Collections;
import java.util.HashMap;
import java.util.Map;

/**
 * @author WangCheng
 */
@Configuration
@EnableConfigurationProperties(JdbcProperties.class)
@ConditionalOnProperty(value = {"jdbc.enabled", "jdbc.druid.enabled"})
public class DruidAutoConfiguration {

    @Autowired
    private JdbcProperties jdbcProperties;

    @Bean
    public ServletRegistrationBean registerDruidServlet() {
        JdbcProperties.DruidParam druidParam = jdbcProperties.getDruid();
        ServletRegistrationBean registration = new ServletRegistrationBean<>();
        registration.setServlet(new StatViewServlet());
        registration.setName("DruidStatView");
        registration.setEnabled(true);
        Map params = new HashMap<>();
        params.put("loginUsername", druidParam.getUsername());
        params.put("loginPassword", druidParam.getPassword());
        registration.setInitParameters(params);
        registration.setUrlMappings(Collections.singletonList(druidParam.getContextPath()));
        return registration;
    }

    @Bean
    public FilterRegistrationBean registerDruidFilter() {
        JdbcProperties.DruidParam druidParam = jdbcProperties.getDruid();
        FilterRegistrationBean registration = new FilterRegistrationBean<>();
        registration.setFilter(new WebStatFilter());
        registration.addUrlPatterns("/*");
        registration.setName("druidFilter");
        Map params = new HashMap<>();
        params.put("exclusions", "*.js,*.gif,*.jpg,*.bmp,*.png,*.css,*.ico," + druidParam.getContextPath());
        registration.setInitParameters(params);
        registration.setOrder(1);
        return registration;
    }
}