springboot配置反向代理

一. 单一反向代理配置

1. 配置文件

# 自定义代理相关配置
# 代理的本地路由
proxy.servlet_url: /nginx/*
# 代理的本地路由
proxy.servlet_url1: /nginx1/*
# 要代理的地址
proxt.target_url: http://localhost:8080/nginx

2. 自定义注解类

package com.config;


import com.google.common.collect.ImmutableMap;
import org.mitre.dsmiley.httpproxy.ProxyServlet;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.boot.web.servlet.ServletRegistrationBean;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;


import javax.servlet.Servlet;
import java.util.Map;

/**
 * @author 天真热
 * @create 2022-01-24 20:25
 * @desc
 **/
@Configuration
public class ProxyServletConfiguration {

    // 读取配置文件中路由设置
    @Value("${proxy.servlet_url}")
    private String servlet_url;
    // 读取配置文件中路由设置
    @Value("${proxy.servlet_url1}")
    private String servlet_url1;
    // 读取配置中代理目标地址
    @Value("${proxt.target_url}")
    private String target_url;


    @Bean
    public Servlet createProxyServlet() {
        // 创建新的ProxyServlet
        return new ProxyServlet();
    }

    @Bean
    public ServletRegistrationBean proxyServletRegistration() {
        ServletRegistrationBean registrationBean = new ServletRegistrationBean(createProxyServlet(), servlet_url);
        // 这个setName必须要设置，并且多个的时候，名字需要不一样
        registrationBean.setName("suit");
        //设置网址以及参数
        Map<String, String> params = ImmutableMap.of(
                "targetUri", target_url,
                "log", "true");
        registrationBean.setInitParameters(params);
        return registrationBean;
    }

    @Bean
    public ServletRegistrationBean proxyServletRegistration1() {
        ServletRegistrationBean registrationBean = new ServletRegistrationBean(createProxyServlet(), servlet_url1);
        // 这个setName必须要设置，并且多个的时候，名字需要不一样
        registrationBean.setName("suit1");

        //设置网址以及参数
        Map<String, String> params = ImmutableMap.of(
                "targetUri", target_url,
                "log", "true");
        registrationBean.setInitParameters(params);
        return registrationBean;
    }
}

可以看到这种配置反向代理地址，对于多个地址，需要手动注入多次bean，较为麻烦。但是一般项目够用

二. 动态反向代理配置

1. 配置文件
   用“|”将请求和转发地址分开，并且多个地址用“，”隔开

proxy:
  url: /nginx/*|http://localhost:8080/nginx,/nginx1/*|http://localhost:8080/nginx

2. 注解类

package com.config;

import org.springframework.beans.BeansException;
import org.springframework.beans.factory.config.ConfigurableListableBeanFactory;
import org.springframework.beans.factory.support.BeanDefinitionBuilder;
import org.springframework.beans.factory.support.BeanDefinitionRegistry;
import org.springframework.beans.factory.support.BeanDefinitionRegistryPostProcessor;
import org.springframework.boot.web.servlet.ServletRegistrationBean;
import org.springframework.context.EnvironmentAware;
import org.springframework.core.env.Environment;

import com.google.common.collect.ImmutableMap;
import org.mitre.dsmiley.httpproxy.ProxyServlet;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;


import javax.servlet.Servlet;
import java.util.HashSet;
import java.util.LinkedHashMap;
import java.util.Set;


@Configuration
public class ProxyServletConfiguration implements BeanDefinitionRegistryPostProcessor, EnvironmentAware {
    Environment env;
    @Bean
    public Servlet createProxyServlet() {
        // 创建新的ProxyServlet
        return new ProxyServlet();
    }
    @Override
    public void postProcessBeanFactory(ConfigurableListableBeanFactory beanFactory) throws BeansException {
    }
    @Override
    public void postProcessBeanDefinitionRegistry(BeanDefinitionRegistry registry) throws BeansException {

        //获取需要反向代理的路径字符串集合
        String proxyUrlStr = env.getProperty("proxy.url");
        //拆分
        String[] proxyUrls = proxyUrlStr.split(",");
        //遍历注入bean
        for (int i = 0; i < proxyUrls.length; i++) {
            String proxyUrl = proxyUrls[i];
            String[] urls = proxyUrl.split("\\|");
            //访问地址
            String servlet_url = urls[0];
            //转发地址
            String target_url = urls[1];
            //动态设置bean
            BeanDefinitionBuilder beanDefinitionBuilder = BeanDefinitionBuilder.rootBeanDefinition(ServletRegistrationBean.class);
            //设置参数
            beanDefinitionBuilder.addPropertyValue("name", "suit" + i);
            beanDefinitionBuilder.addPropertyValue("initParameters", new LinkedHashMap(ImmutableMap.of("targetUri", target_url, "log", "true")));
            Set<String> urlMappings = new HashSet<>();
            urlMappings.add(servlet_url);
            beanDefinitionBuilder.addPropertyValue("urlMappings", urlMappings);
            beanDefinitionBuilder.addPropertyValue("servlet", createProxyServlet());
            //注入
            registry.registerBeanDefinition("proxy" + i, beanDefinitionBuilder.getBeanDefinition());
        }
    }
    @Override
    public void setEnvironment(Environment environment) {
        this.env = environment;
    }
}

这种方式较之前比较机动性较强！