Spring MVC根据运行参数添加filter代理（启动时动态添加filter和listener）

在工作中涉及到开发环境中需要跳过SSO，一开始在Maven中使用profiles在构建的时候加入SSO的filter，而开发环境中的web.xml不配置SSO。因为使用的时候需要在maven构建命令中加入参数，和测试和运营沟通的不便利，决定另寻他法。

这种方法通过实现spring mvc的WebApplicationInitializer接口，在这里面读取运行参数，利用这个参数决定是不是不要添加SSO filter（默认添加）。

实现目标：在运行参数中定义 -Dapp.env=dev，启动的时候能够 加载config/properties目录内的env_dev.properties文件，并根据文件内的配置项sso.pass=true跳过SSO的filter添加。

1、定义属性

/**
	 * 配置文件前缀
	 */
	private static final String SETTINGS_FILE_PREFIX = "env_";
	
	/**
	 * 运行环境参数key
	 */
	private static final String ENVIROMENT_KEY = "app.env";
	
	/**
	 * 是否跳过SSO设置key
	 */
	private static final String KEY_IS_PASS_SSO = "sso.pass"; 
	
	/**
	 * 是否跳过SSO
	 */
	private static final Boolean IS_PASS_SSO;

2、获取运行参数，在static块中运行

	
String env = System.getProperty(ENVIROMENT_KEY);
Properties prop = new Properties();
		if(env!=null && env.trim().length()>0){
			try {
				InputStream inputStream = ApplicationInitializer.class.getClassLoader().getResourceAsStream("config/properties/"+SETTINGS_FILE_PREFIX + env.trim()+".properties");
				prop.load(inputStream);
			} catch (IOException e) {
				logger.error("参数文件{}读取出错",SETTINGS_FILE_PREFIX+env+".properties");
			}  
		}
		//SSO
		String passSSOValue = prop.getProperty(KEY_IS_PASS_SSO);
		if(StringUtils.isEmpty(passSSOValue)){
			IS_PASS_SSO = false;
		}else{
			IS_PASS_SSO = Boolean.parseBoolean(passSSOValue);
		}

3、实现接口的onStartup方法

//单点登录设置
    	if(IS_PASS_SSO){
    		logger.info("设置为跳过SSO");
    	}else{
    		try {
    			//添加监听器
    			servletContext.addListener("org.springframework.security.web.session.HttpSessionEventPublisher");
    			
    			//添加filter
    			Dynamic addFilter = servletContext.addFilter("springSecurityFilterChain", new DelegatingFilterProxy());
    			
    			addFilter.addMappingForUrlPatterns(EnumSet.of(DispatcherType.REQUEST), false, "/*");
			} catch (Exception e) {
				logger.error("添加SSO filter出现错误");
				throw e;
			}
    	}

完毕。

以下是全部代码：

import java.io.IOException;
import java.io.InputStream;
import java.util.EnumSet;
import java.util.Properties;

import javax.servlet.DispatcherType;
import javax.servlet.FilterRegistration.Dynamic;
import javax.servlet.ServletContext;
import javax.servlet.ServletException;

import org.apache.commons.lang.StringUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.web.WebApplicationInitializer;
import org.springframework.web.filter.DelegatingFilterProxy;

public class ApplicationInitializer implements WebApplicationInitializer {
	
	private final static Logger logger = LoggerFactory.getLogger(ApplicationInitializer.class);
	
	/**
	 * 配置文件前缀
	 */
	private static final String SETTINGS_FILE_PREFIX = "env_";
	
	/**
	 * 运行环境参数key
	 */
	private static final String ENVIROMENT_KEY = "app.env";
	
	/**
	 * 是否跳过SSO设置key
	 */
	private static final String KEY_IS_PASS_SSO = "sso.pass"; 
	
	/**
	 * 是否跳过SSO
	 */
	private static final Boolean IS_PASS_SSO;
	
	
	/**
	 * 根据配置文件初始化参数
	 * 读取的配置文件为"env_"+环境参数+"."
	 */
	static{
		String env = System.getProperty(ENVIROMENT_KEY);
		Properties prop = new Properties();
		if(env!=null && env.trim().length()>0){
			try {
				InputStream inputStream = ApplicationInitializer.class.getClassLoader().getResourceAsStream("config/properties/"+SETTINGS_FILE_PREFIX + env.trim()+".properties");
				prop.load(inputStream);
			} catch (IOException e) {
				logger.error("参数文件{}读取出错",SETTINGS_FILE_PREFIX+env+".properties");
			}  
		}
		//SSO
		String passSSOValue = prop.getProperty(KEY_IS_PASS_SSO);
		if(StringUtils.isEmpty(passSSOValue)){
			IS_PASS_SSO = false;
		}else{
			IS_PASS_SSO = Boolean.parseBoolean(passSSOValue);
		}
	}
	
    @Override
    public void onStartup(ServletContext servletContext) throws ServletException {
    	
    	//单点登录设置
    	if(IS_PASS_SSO){
    		logger.info("设置为跳过SSO");
    	}else{
    		try {
    			//添加监听器
    			servletContext.addListener("org.springframework.security.web.session.HttpSessionEventPublisher");
    			
    			//添加filter
    			Dynamic addFilter = servletContext.addFilter("springSecurityFilterChain", new DelegatingFilterProxy());
    			
    			addFilter.addMappingForUrlPatterns(EnumSet.of(DispatcherType.REQUEST), false, "/*");
			} catch (Exception e) {
				logger.error("添加SSO filter出现错误");
				throw e;
			}
    	}
    }
}