Springboot2.0配置請求服務代理

Springboot2.0配置請求服務代理

注：下圖內容源自博客https://blog.csdn.net/wohaqiyi/article/details/81327512該博主版本爲2.0以下

借鑑方法思路實現功能發現org.springframework.boot.bind這個包早已刪掉，
org.springframework.boot.bind.RelaxedPropertyResolver;
具體解決辦法如下:

package com.siyeto.cweb.common.configuration;

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

import javax.servlet.ServletException;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import org.springframework.boot.context.properties.bind.BindResult;
import org.springframework.boot.context.properties.bind.Binder;
import org.springframework.boot.context.properties.source.ConfigurationPropertySource;
import org.springframework.boot.context.properties.source.ConfigurationPropertySources;
import org.springframework.boot.web.servlet.ServletRegistrationBean;
import org.springframework.context.EnvironmentAware;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.core.env.Environment;

@SuppressWarnings({ "unchecked", "rawtypes" })
@Configuration
public class SolrProxyServletConfiguration implements EnvironmentAware  {
	
	private Properties properties;
	
	@Bean
    public ServletRegistrationBean servletRegistrationBean(HttpServletRequest servletRequest,HttpServletResponse servletResponse) throws ServletException, IOException{
		ServletRegistrationBean servletRegistrationBean = new ServletRegistrationBean(new MyProxyServlet(),properties.getProperty("dataServer-servlet-url"));
        servletRegistrationBean.setName("dataServer");
        servletRegistrationBean.addInitParameter("targetUri", properties.getProperty("dataServer-target-url"));
        servletRegistrationBean.addInitParameter(MyProxyServlet.P_LOG, properties.getProperty("logging_enabled", "false"));
        return servletRegistrationBean;
    }

    @Override
    public void setEnvironment(Environment environment) {
    	/**
         * springboot2.X 升級後改用ConfigurationPropertySource進行綁定
         */
        Iterable<ConfigurationPropertySource> sources = ConfigurationPropertySources.get(environment);
        Binder binder = new Binder(sources);
        BindResult<Properties> bindResult = binder.bind("proxy.solr", Properties.class);
        properties = bindResult.get();
    }
    
}

新建類繼承ProxyServlet重寫doExecute方法

package com.siyeto.cweb.common.configuration;

import java.io.IOException;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import org.apache.http.HttpRequest;
import org.apache.http.HttpResponse;
import org.mitre.dsmiley.httpproxy.ProxyServlet;

import com.siyeto.cweb.common.utill.HMACSHA256;
import com.siyeto.cweb.common.utill.SpringContextUtils;
import com.siyeto.cweb.modules.management.service.CompanyService;

/**
 * @Description: 自定義servlet
 * @author: lzh
 * @date: 2020年3月4日 上午9:44:29
 */
public class MyProxyServlet extends ProxyServlet  {
	private static final long serialVersionUID = 1L;
	private static final String ALLOWED_HEADERS = "Origin, Accept, Content-Type, x-requested-with, X-Forwarded-Prefix, token, lang, id, timestamp";
	
	/**
	 * proxyServlet.service(servletRequest, servletResponse);
	 * 需要重寫doExecute()方法
	 */
	@Override
    protected HttpResponse doExecute(HttpServletRequest servletRequest, HttpServletResponse servletResponse,
            HttpRequest proxyRequest) throws IOException {
		String companyId = servletRequest.getHeader("companyId");
		String currentTimeMillis = System.currentTimeMillis()+"";
		CompanyService companyService = (CompanyService) SpringContextUtils.getBean("companyService");
		String key = companyService.selectCompanyKey(Long.valueOf(companyId));
        HttpResponse response = super.doExecute(servletRequest, servletResponse, proxyRequest);
        response.setHeader("Access-Control-Allow-Origin","*");
        response.setHeader("Access-Control-Allow-Credentials","true");
        response.setHeader("Access-Control-Allow-Methods","*");
        response.setHeader("Access-Control-Allow-Headers",ALLOWED_HEADERS);
        response.setHeader("token", HMACSHA256.encrytSHA256(currentTimeMillis,key));
        response.setHeader("timestamp", currentTimeMillis);
        response.setHeader("id", companyId);
        response.setHeader("lang", servletRequest.getHeader("lang"));
        return response;
    }
}

因爲需求需要添加請求頭，所以仔細閱讀ProxyServlet源碼中的service方法解決問題